基于机器学习的胆结石疾病预测效能研究OA
Research on the Prediction Efficiency of Gallstone Disease Based on Machine Learning
目的 探讨不同机器学习模型对胆结石疾病预测的效能差异,识别关键预测指标,为早期筛查、风险预警及临床决策提供数据驱动的智能化解决方案.方法 基于UCI胆结石数据库中的319例受试者的38项特征变量与1项二分类目标变量,采用分层抽样法将数据划分为训练集与测试集,构建随机森林(Random Forest)、支持向量机(SVM)和极端梯度提升(XGBoost)三种预测模型,通过准确率、召回率、精确率及F1值等指标综合评估模型性能,并结合特征重要性分析识别关键风险因素.结果 研究显示三种模型均可有效预测胆结石,其中XGBoost模型性能最优,准确率90.77%、召回率93.94%、精确率88.57%、F1值91.18%,达到漏诊与误诊风险最优平衡;随机森林模型表现稳健,可作为备选;SVM模型适应性较弱,不推荐直接用于临床筛查.特征分析证实C反应蛋白、维生素D及肥胖度为胆结石预测的核心指标.结论 XGBoost算法在高维临床数据处理及复杂特征交互捕捉方面优势显著,适用于胆结石疾病预测任务.本研究发现炎症反应与代谢异常对胆结石形成具有关键作用,可为疾病早期筛查提供辅助工具,有助于推动预防关口前移,并为机器学习临床转化提供方法学参考.
Objective To explore the performance differences of different machine learning models in the prediction of gallstone diseases,identify key predictive indicators,and provide data-driven intelligent solutions for early screening,early risk warning and clinical decision-making.Methods Based on 38 feature variables and one dichotomous target variable of 319 subjects in UCI gallstone database,the data were divided into training set and test set by stratified sampling method,and three prediction models of random forest,support vector machine(SVM)and extreme gradient boosting(XGBoost)were constructed.The performance of the models was comprehensively evaluated by accuracy,recall rate,precision rate and F1 value,and the key risk factors were identified by combining feature importance analysis.Results The study showed that all three models could effectively predict gallstones.Among them,the XGBoost model had the best performance,with an accuracy rate of 90.77%,a recall rate of 93.94%,a precision rate of 88.57%,and a F1 value of 91.18%,achieving the optimal balance between missed diagnosis and misdiagnosis risk.The random forest model demonstrated robust performance and could serve as an alternative,while the SVM model showed weaker adaptability and was not recommended for direct clinical screening.Feature analysis confirmed that C-reactive protein,vitamin D,and obesity degree were core indicators for gallstone prediction.Conclusion The XGBoost algorithm demonstrates significant advantages in processing high-dimensional clinical data and capturing complex feature interactions,making it suitable for gallstone disease prediction tasks.This study identifies the critical roles of inflammatory response and metabolic abnormalities in gallstone formation,providing a tool for early disease screening,facilitating the advancement of prevention strategies,and offering a methodological reference for the clinical translation of machine learning.
韩丽娟
对外经济贸易大学统计学院/拓尔思信息技术股份有限公司解决方案中心,北京 100029
医药卫生
胆结石机器学习XGBoost模型
GallstoneMachine learningXGBoost model
《医学信息》 2026 (18)
29-34,6
评论