计算机与现代化 ›› 2018, Vol. 0 ›› Issue (04): 117-.doi: 10.3969/j.issn.10062475.2018.04.022

• 应用与开发 • 上一篇    下一篇

基于蚁群路径优化决策树及逻辑回归的慢性肾病进展概率预测模型

  

  1.   (1.四川大学电子信息学院,四川成都610065;2.四川大学华西第二医院,四川成都610041)
  • 出版日期:2018-04-28 发布日期:2018-05-02
  • 作者简介:冯苗(1995),女,四川成都人,四川大学电子信息学院硕士研究生,研究方向:医疗数据处理; 綦小蓉(1973),女,四川成都人,四川大学华西第二医院副主任医师,副教授,博士,研究方向:临床医学; 李智(1975),男,四川成都人,副教授,硕士生导师,博士,研究方向:压缩感知,医疗大数据。

Progression Prediction Model of Chronic Kidney Disease Based on  Decision Tree Ant Path Optimization and Logistic Regression

  1.  (1. College of Electronics and Information Engineering, Sichuan University, Chengdu 610065, China;
    2. West China Second University Hospital, Sichuan University, Chengdu 610041, China) 
  • Online:2018-04-28 Published:2018-05-02

摘要: 慢性肾病(Chronic Kidney Disease, CKD)是一种进展性疾病,早期若不及时加以治疗会导致病情发展,甚至肾衰竭。为了研究CKD患者从早期发展到终末期的概率,本文提出一种CKD进展概率预测模型:结合蚁群路径优化决策树算法(Decision Tree Ant Path Optimization, DTAPO)和逻辑回归算法(Logistic Regression, LR),将CKD患者数据分为P(进展)和NP(非进展)2类,得到分类精确率和召回率,从而计算CKD患者由3期进展到4期或5期的概率。实验结果表明,当特征数目为13时,结合逻辑回归的蚁群路径优化决策树算法的预测效果最好,其分类精确率为98.84%,由该精确率预测得到的进展患者确实由3期进展到4期或5期的概率为0.9827。

关键词: 慢性肾病, 进展预测, 逻辑回归, 蚁群路径优化, 决策树

Abstract: Chronic kidney disease (CKD) is a progressive disease, it will lead to the development of the disease and even renal failure if not treated in a timely manner. To study the progression probability from earlystage to endstage of CKD patients, a prediction model of CKD progression probability is proposed. Combining decision tree ant path optimization (DTAPO) and logistic regression (LR) algorithm, this paper divides CKD patients’ data into two categories: progress (P) and non progress (NP), the classification accuracy rate and recall rate are obtained so as to calculate the probability from the stage 3 to the stage 4 or 5. It is demonstrated from the experimental results that when the number of features is 13, the prediction algorithm combining decision tree ant path optimization algorithm with logistic regression achieves the best performance, and the accuracy rate of classification is 98.84%. The probability of progression from the stage 3 to the stage 4 or 5 is 0.9827.

Key words: chronic kidney disease, progression forecast, logistic regression, ant path optimization, decision tree

中图分类号: