基于TPE-SVM的乳腺癌诊断可解释人工智能方法
Explainable AI Approach for Breast Cancer Diagnosis via TPE-Optimized Support Vector Machine
DOI: 10.12677/mos.2025.1410600, PDF,    科研立项经费支持
作者: 宋柯蕾, 王秋阳, 谢彤嫣:上海理工大学管理学院,上海;傅文翰*:上海理工大学管理学院,上海;上海理工大学智慧应急管理学院,上海
关键词: 可解释人工智能乳腺癌诊断支持向量机贝叶斯优化可解释性分析Explainable Artificial Intelligence Breast Cancer Diagnosis Support Vector Machine Bayesian Optimization Interpretability Analysis
摘要: 乳腺癌是女性中最常见的恶性肿瘤之一,早期精准诊断对提高患者生存率至关重要。针对传统乳腺癌诊断方法存在的主观性强和误诊率高,以及现有人工智能模型可解释性较差等问题。本文设计一个基于贝叶斯优化支持向量机(TPE-SVM)的乳腺癌智能诊断模型,并结合LIME解释方法提高诊断过程中的可解释性。方法上,构建基于径向基核函数的支持向量机模型,并利用TPE算法对关键超参数进行优化,最后引入LIME方法实现诊断结果的特征可视化与解释。采用UCI数据库中的威斯康星乳腺癌数据集进行仿真验证,结果显示,该方法在各项指标上表现优异,进一步的LIME可解释性分析也表明,模型判别依据与临床医学知识高度一致。本文所构建的诊断框架为AI在医疗场景中的可用性与可信性提供了新思路。
Abstract: Breast cancer is one of the most common malignant tumors among women, and accurate early diagnosis is crucial for improving patient survival rates. To address the high subjectivity and misdiagnosis rates of traditional diagnostic methods, as well as the limited interpretability of current artificial intelligence models, this study proposes an intelligent diagnostic model for breast cancer based on a Bayesian-Optimized Support Vector Machine (TPE-SVM), integrated with the LIME explanation method to enhance interpretability during the diagnostic process. Methodologically, a Support Vector Machine with a radial basis function kernel is constructed, and its key hyperparameters are optimized using the TPE algorithm. The LIME method is subsequently employed to visualize and interpret the model’s diagnostic decisions. The model is validated on the Wisconsin Breast Cancer dataset from the UCI repository. Experimental results demonstrate excellent performance across multiple evaluation metrics, and further LIME-based interpretability analysis confirms that the model’s decision-making criteria are highly consistent with established clinical knowledge. The diagnostic framework proposed in this study offers new insights into the usability and trustworthiness of AI applications in medical scenarios.
文章引用:宋柯蕾, 傅文翰, 王秋阳, 谢彤嫣. 基于TPE-SVM的乳腺癌诊断可解释人工智能方法[J]. 建模与仿真, 2025, 14(10): 1-11. https://doi.org/10.12677/mos.2025.1410600

参考文献

[1] 徐兵河. 早筛可提高乳腺癌患者生存率[J]. 科学新生活, 2024, 27(8): 15.
[2] 张姝艳, 皮婷婷. 医疗领域中人工智能应用的可解释性困境与治理[J]. 医学与哲学, 2023, 44(3): 25-29, 35.
[3] 刘子华, 郑汉东, 刘卫勇. 基于改进动态集成选择算法的乳腺肿块辅助诊断模型[J]. 计算机应用研究, 2023, 40(1): 147-154.
[4] Guidotti, R., Monreale, A., Ruggieri, S., Turini, F., Giannotti, F. and Pedreschi, D. (2018) A Survey of Methods for Explaining Black Box Models. ACM Computing Surveys, 51, 1-42. [Google Scholar] [CrossRef
[5] Gunning, D. and Aha, D.W. (2019) Darpa’s Explainable Artificial Intelligence Program. AI Magazine, 40, 44-58. [Google Scholar] [CrossRef
[6] Ribeiro, M.T., Singh, S. and Guestrin, C. (2016) “Why Should I Trust You?” Explaining the Predictions of Any Classifier. Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, San Francisco, 13-17 August 2016, 1135-1144. [Google Scholar] [CrossRef
[7] Barr Kumarakulasinghe, N., Blomberg, T., Liu, J., Saraiva Leao, A. and Papapetrou, P. (2020) Evaluating Local Interpretable Model-Agnostic Explanations on Clinical Machine Learning Classification Models. 2020 IEEE 33rd International Symposium on Computer-Based Medical Systems (CBMS), Rochester, 28-30 July 2020, 7-12. [Google Scholar] [CrossRef
[8] Jalali, A., Schindler, A., Haslhofer, B. and Rauber, A. (2020) Machine Learning Interpretability Techniques for Outage Prediction: A Comparative Study. PHM Society European Conference, 5, 10. [Google Scholar] [CrossRef
[9] Onchis, D.M. and Gillich, G. (2021) Stable and Explainable Deep Learning Damage Prediction for Prismatic Cantilever Steel Beam. Computers in Industry, 125, Article ID: 103359. [Google Scholar] [CrossRef
[10] Zhu, X., Zhang, K., Li, X., Su, F. and Tian, J. (2024) An Interpretable Machine Learning Method for Risk Stratification of Patients with Acute Coronary Syndrome. Heliyon, 10, e36815. [Google Scholar] [CrossRef] [PubMed]
[11] 杨萌宇, 张雷, 曾悦. 改进粒子群算法优化的SVM在恶性肿瘤诊断中的应用[J]. 现代电子技术, 2020, 43(15): 110-114, 118.
[12] Kim, K.H. and Sohn, S.Y. (2020) Hybrid Neural Network with Cost-Sensitive Support Vector Machine for Class-Imbalanced Multimodal Data. Neural Networks, 130, 176-184. [Google Scholar] [CrossRef] [PubMed]
[13] Lotter, W., Diab, A.R., Haslam, B., Kim, J.G., Grisot, G., Wu, E., et al. (2021) Robust Breast Cancer Detection in Mammography and Digital Breast Tomosynthesis Using an Annotation-Efficient Deep Learning Approach. Nature Medicine, 27, 244-249. [Google Scholar] [CrossRef] [PubMed]
[14] Lu, M.Y., Williamson, D.F.K., Chen, T.Y., Chen, R.J., Barbieri, M. and Mahmood, F. (2021) Data-Efficient and Weakly Supervised Computational Pathology on Whole-Slide Images. Nature Biomedical Engineering, 5, 555-570. [Google Scholar] [CrossRef] [PubMed]
[15] He, K., Zhang, X., Ren, S. and Sun, J. (2016) Deep Residual Learning for Image Recognition. 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, 27-30 June 2016, 770-778. [Google Scholar] [CrossRef
[16] Vaswani, A., Shazeer, N., Parmar, N., et al. (2017) Attention Is All You Need. arXiv: 1706.03762.
[17] Witowski, J., Heacock, L., Reig, B., Kang, S.K., Lewin, A., Pysarenko, K., et al. (2022) Improving Breast Cancer Diagnostics with Deep Learning for MRI. Science Translational Medicine, 14, eabo4802. [Google Scholar] [CrossRef] [PubMed]
[18] Shafi, I., Din, S., Khan, A., Díez, I.D.L.T., Casanova, R.d.J.P., Pifarre, K.T., et al. (2022) An Effective Method for Lung Cancer Diagnosis from CT Scan Using Deep Learning-Based Support Vector Network. Cancers, 14, Article 5457. [Google Scholar] [CrossRef] [PubMed]
[19] 龚安, 吕秀明. 基于卷积神经网络的乳腺癌病理图像分类方法[J]. 计算机应用与软件, 2023, 40(6): 133-139, 198.
[20] 徐坤财, 张宁, 廖益龙, 等. 基于CNN和Transformer的两阶段乳腺癌病理图像分类方法研究[J]. 医疗卫生装备, 2024, 45(12): 1-8.
[21] 孟霖宜, 刘屿鸿. 基于PSO优化SVM算法的癌症诊断方法研究[J]. 计算机仿真, 2023, 40(8): 279-283.