arXiv:2504.04262cs.AI2025-04被引 19

用优化算法提升肾病检测准确率,模型达98.75%准确率且可解释。

Improving Chronic Kidney Disease Detection Efficiency: Fine Tuned CatBoost and Nature-Inspired Algorithms with Explainable AI

  • 用模拟退火、 cuckoo 搜寻等自然启发算法优化特征与参数
  • 细调后的 CatBoost 模型准确率达 98.75%,AUC 达 0.9993
  • 通过 SHAP 解释关键指标,适合资源有限地区的临床应用

慢性肾病(CKD)是全球重大健康问题,死亡率持续上升。早期检测对延缓进展和改善预后至关重要,但传统诊断方法在资源匮乏地区存在局限。本研究评估了四种机器学习模型:随机森林(RF)、多层感知机(MLP)、逻辑回归(LR)和微调的 CatBoost 算法。其中,微调后的 CatBoost 表现最佳,准确率为 98.75%,AUC 为 0.9993,卡帕系数达 97.35%。该模型结合模拟退火筛选重要特征,Cuckoo 搜寻调整异常值,网格搜索优化超参数。通过 SHAP 可解释性技术揭示关键临床特征为:尿比重、血清肌酐、白蛋白、血红蛋白及糖尿病史。研究证明先进机器学习在低收入和中等收入医疗环境中具有显著潜力,可提供高精度、可解释且高效的诊断工具,助力早期干预与整体诊疗改善。

原文摘要 · Abstract (English)

Chronic Kidney Disease (CKD) is a major global health issue which is affecting million people around the world and with increasing rate of mortality. Mitigation of progression of CKD and better patient outcomes requires early detection. Nevertheless, limitations lie in traditional diagnostic methods, especially in resource constrained settings. This study proposes an advanced machine learning approach to enhance CKD detection by evaluating four models: Random Forest (RF), Multi-Layer Perceptron (MLP), Logistic Regression (LR), and a fine-tuned CatBoost algorithm. Specifically, among these, the fine-tuned CatBoost model demonstrated the best overall performance having an accuracy of 98.75%, an AUC of 0.9993 and a Kappa score of 97.35% of the studies. The proposed CatBoost model has used a nature inspired algorithm such as Simulated Annealing to select the most important features, Cuckoo Search to adjust outliers and grid search to fine tune its settings in such a way to achieve improved prediction accuracy. Features significance is explained by SHAP-a well-known XAI technique-for gaining transparency in the decision-making process of proposed model and bring up trust in diagnostic systems. Using SHAP, the significant clinical features were identified as specific gravity, serum creatinine, albumin, hemoglobin, and diabetes mellitus. The potential of advanced machine learning techniques in CKD detection is shown in this research, particularly for low income and middle-income healthcare settings where prompt and correct diagnoses are vital. This study seeks to provide a highly accurate, interpretable, and efficient diagnostic tool to add to efforts for early intervention and improved healthcare outcomes for all CKD patients.

肾病检测可解释AICatBoost特征优化

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