基于CT影像的可解释机器学习模型,精准区分子宫内膜癌良恶性。
CT Radiomics-Based Explainable Machine Learning Model for Accurate Differentiation of Malignant and Benign Endometrial Tumors: A Two-Center Study
- 从术前CT图像提取1132个影像特征,用随机森林算法建模诊断。
- 测试集准确率达96%(AUROC=0.96),训练集完美区分(AUROC=1.00)。
- 通过SHAP和可视化展示关键特征,帮助医生理解模型决策逻辑。
本研究旨在开发并验证一种基于CT影像组学的可解释机器学习模型,用于精准区分子宫内膜癌(EC)患者的良恶性肿瘤。纳入两中心共83例患者(恶性46例,良性37例),数据分为训练集(n=59)与测试集(n=24)。对术前CT扫描进行手动勾画病灶区域,使用Pyradiomics提取1132个影像组学特征。分别应用六种可解释机器学习算法,筛选最优影像组学分析流程。通过敏感性、特异性、准确率、精确率、F1分数、AUROC和AUPRC评估诊断性能。为提升临床可读性,采用SHAP分析与特征映射可视化,并评估校准曲线与决策曲线。结果显示,随机森林模型表现最佳,训练集AUROC达1.00,测试集仍达0.96。SHAP分析揭示所有入选特征均与子宫内膜癌显著相关(P<0.05)。特征图提供临床可行的辅助评估工具。决策曲线分析显示,该模型净获益高于“全选”与“不选”策略,有助于识别高风险病例,减少不必要的干预。结论:该模型具备优异诊断性能,可作为子宫内膜癌智能辅助诊断工具。
原文摘要 · Abstract (English)
Aimed to develop and validate a CT radiomics-based explainable machine learning model for precise diagnosing malignancy and benignity specifically in endometrial cancer (EC) patients. A total of 83 EC patients from two centers, including 46 with malignant and 37 with benign conditions, were included, with data split into a training set (n=59) and a testing set (n=24). The regions of interest (ROIs) were manually segmented from pre-surgical CT scans, and 1132 radiomic features were extracted from the pre-surgical CT scans using Pyradiomics. Six explainable machine learning (ML) modeling algorithms were implemented respectively, for determining the optimal radiomics pipeline. The diagnostic performance of the radiomic model was evaluated by using sensitivity, specificity, accuracy, precision, F1 score, AUROC, and AUPRC. To enhance clinical understanding and usability, we separately implemented SHAP analysis and feature mapping visualization, and evaluated the calibration curve and decision curve. By comparing six modeling strategies, the Random Forest model emerged as the optimal choice for diagnosing EC, with a training AUROC of 1.00 and a testing AUROC of 0.96. SHAP identified the most important radiomic features, revealing that all selected features were significantly associated with EC (P < 0.05). Radiomics feature maps also provide a feasible assessment tool for clinical applications. Decision Curve Analysis (DCA) indicated a higher net benefit for our model compared to the "All" and "None" strategies, suggesting its clinical utility in identifying high-risk cases and reducing unnecessary interventions. In conclusion, the CT radiomics-based explainable ML model achieved high diagnostic performance, which could be used as an intelligent auxiliary tool for the diagnosis of endometrial cancer.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。