arXiv:2509.23268cs.LGcs.CY2025-09被引 1

用迁移学习等方法提升早期乳腺癌生存预测精度,尤其在数据不全时表现更优。

Transfer Learning and Machine Learning for Training Five Year Survival Prognostic Models in Early Breast Cancer

  • 通过迁移学习、随机生存森林和集成融合改进预测模型
  • 在数据缺失情况下仍能准确预测,校准误差降低至0.007以下
  • 适合临床数据不完整或存在数据分布差异的场景

生存预后信息对乳腺癌管理决策至关重要。尽管近年来侧重基因组预测工具,但临床病理学预测成本更低且更易获取。机器学习(ML)、迁移学习和集成融合为构建稳健的预后框架提供了可能。本研究通过比较从零训练的ML模型、基于预训练工具PREDICT v3的迁移学习以及集成融合,评估其在乳腺癌生存预后中的潜力。模型在MA.27试验数据上训练,外部验证分别在TEAM试验和SEER队列中进行。迁移学习通过微调PREDICT v3实现,从零训练的ML包括随机生存森林(RSF)和极端梯度提升(XGB),集成融合采用加权预测值求和。与预训练模型相比,迁移学习、从零训练的RSF及集成融合在MA.27中显著改善了校准性能(校准误差从PREDICT v3的0.042降至≤0.007),而判别能力保持相当(AUC从0.738提升至0.744–0.799)。由于信息缺失,PREDICT v3在MA.27中有23.8%–25.8%的个体无法生成预测结果,而所有ML模型和集成方法可无视缺失信息完成预测。各模型中,患者年龄、淋巴结状态、病理分级和肿瘤大小的SHAP值最高,表明其关键作用。外部验证显示,迁移学习、RSF和集成融合在SEER队列中表现良好,但在TEAM队列中未达显著优势。研究证明,在缺乏相关输入或存在数据分布偏移时,迁移学习、从零训练的RSF及集成融合可有效提升预后预测能力。

原文摘要 · Abstract (English)

Prognostic information is essential for decision-making in breast cancer management. Recently trials have predominantly focused on genomic prognostication tools, even though clinicopathological prognostication is less costly and more widely accessible. Machine learning (ML), transfer learning and ensemble integration offer opportunities to build robust prognostication frameworks. We evaluate this potential to improve survival prognostication in breast cancer by comparing de-novo ML, transfer learning from a pre-trained prognostic tool and ensemble integration. Data from the MA.27 trial was used for model training, with external validation on the TEAM trial and a SEER cohort. Transfer learning was applied by fine-tuning the pre-trained prognostic tool PREDICT v3, de-novo ML included Random Survival Forests and Extreme Gradient Boosting, and ensemble integration was realized through a weighted sum of model predictions. Transfer learning, de-novo RSF, and ensemble integration improved calibration in MA.27 over the pre-trained model (ICI reduced from 0.042 in PREDICT v3 to <=0.007) while discrimination remained comparable (AUC increased from 0.738 in PREDICT v3 to 0.744-0.799). Invalid PREDICT v3 predictions were observed in 23.8-25.8% of MA.27 individuals due to missing information. In contrast, ML models and ensemble integration could predict survival regardless of missing information. Across all models, patient age, nodal status, pathological grading and tumor size had the highest SHAP values, indicating their importance for survival prognostication. External validation in SEER, but not in TEAM, confirmed the benefits of transfer learning, RSF and ensemble integration. This study demonstrates that transfer learning, de-novo RSF, and ensemble integration can improve prognostication in situations where relevant information for PREDICT v3 is lacking or where a dataset shift is likely.

乳腺癌生存预测迁移学习机器学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。