机器学习模型预测乳腺导管内癌组织学分级:基于X线征象及临床信息

摘要

目的 探讨基于X线征象及临床信息构建机器学习模型预测乳腺导管内癌组织学分级的可行性。方法 回顾性分析239例经病理证实为乳腺导管内癌患者的乳腺X线征象及临床信息，根据病理结果分为非高级别（n=109）及高级别（n=130）。把采集的10个临床信息及15个乳腺X线征象进行统计学分析，筛选出差异有统计学意义的特征构建极端梯度提升、逻辑回归、多项式朴素贝叶斯3种机器学习模型，以ROC曲线下面积（AUC）为主要指标择出最优模型。结果 极端梯度提升、逻辑回归、多项式朴素贝叶斯训练集的AUC值分别为0.790、0.794、0.802，测试集的AUC值分别为0.760、0.758、0.774，准确度分别为0.760、0.759、0.774，敏感度分别为0.725、0.825、0.800，特异度分别为0.625、0.434、0.625。结论 基于机器学习构建的乳腺导管内癌组织学分级模型预测效能较好，多项式朴素贝叶斯的预测效能最优。

关键词

乳腺癌 / 乳腺导管内癌 / 组织学分级 / 乳腺X线 / 机器学习