基于机器学习算法的恶性黑色素瘤生存预测模型
林冲
董斌
代智慧
张璐璐
毋磊
郑州市第一人民医院整形外科,河南郑州 450004
摘要：目的 基于不同机器学习算法建立预后预测模型,以帮助筛选恶性黑色素瘤高风险患者,提高生存率.方法 自2024 年 1-6 月,郑州市第一人民医院整形外科收集 SEER 数据库(Surveillance,Epidemiology,and End Results,SEER)中 2004-2013年确诊为皮肤恶性黑色素瘤的患者资料,采用机器学习中的逻辑回归(logistic regression,LR)、决策树(decision tree,DT)、朴素贝叶斯(naïve Bayes,NB)、K近邻(K-nearest neighbor,KNN)和随机森林(random forest,RF)算法,建立预测皮肤恶性黑色素瘤5年生存状态的模型,使用10折交叉验证法对模型效能进行验证,评估不同模型的预测效能.结果 共纳入研究对象92 328例,其中包括训练集(n=64 630,70％)和测试集(n=27 698,30％);所有患者的5年生存率为92.5％;在本研究中,逻辑回归、决策树、朴素贝叶斯、K近邻和随机森林模型的受试者工作特征曲线下面积分别为0.908、0.858、0.907、0.724和0.903.结论 本研究基于机器学习算法对恶性黑色素瘤患者5年后的生存状态进行预测,其中逻辑回归、朴素贝叶斯和随机森林模型对于恶性黑色素瘤生存预测效果较好,提供了一种与传统统计学方法不同的预测工具.辅助临床医师筛查高危患者,为恶性黑色素瘤患者提供精准的个体化治疗以及个性化的预后管理方式.
关键词：黑色素瘤预测模型生存分析机器学习
论文发表日期：2025-04-15
在线出版日期：2025-08-15（本平台首次上网日期，不代表文献的发表时间）
页数：5( 211-215 )
英文信息