Journal of Peking University(Health Sciences) ›› 2018, Vol. 50 ›› Issue (2): 239-244. doi: 10.3969/j.issn.1671-167X.2018.02.007

• Article • Previous Articles     Next Articles

Application of support vector machine in predicting in-hospital mortality risk of patients with acute kidney injury in ICU

LIN Ke1,2, XIE Jun-qing1,2, HU Yong-hua1,2, KONG Gui-lan2△   

  1. (1. Department of Epidemiology and Biostatistics, Peking University School of Public Health, Beijing 100191, China; 2. Medical Informatics Center, Peking University, Beijing 100191, China)
  • Online:2018-04-18 Published:2018-04-18
  • Contact: KONG Gui-lan E-mail:guilan.kong@hsc.pku.edu.cn
  • Supported by:
    Supported by the National Natural Science Foundation of China(81771938,81301296),  Peking University Clinical Medicine + X Project (PKU2017LCX05) and Peking University Seed Fund for MedicineInformation Interdisciplinary Research Project (BMU20160592)

Abstract: Objective: To construct an in-hospital mortality prediction model for patients with acute kidney injury (AKI) in intensive care unit (ICU) by using support vector machine (SVM), and compare it with the simplified acute physiology score Ⅱ (SAPS-Ⅱ) which is commonly used in the ICU. Methods: We used Medical Information Mart for Intensive Care Ⅲ (MIMIC-Ⅲ) database as data source. The AKI patients in the MIMIC-Ⅲ database were selected according to the 2012 Kidney Disease: Improving Global Outcomes (KDIGO) definition of AKI. We employed the same predictor variable set as used in SAPS-Ⅱ to construct an SVM model. Meanwhile, we also developed a customized SAPS-Ⅱ mo-del using MIMIC-Ⅲ database, and compared performances between the SVM model and the customized SAPS-Ⅱ model. The performance of each model was evaluated via area under the receiver operation characteristic curve (AUROC), root mean squared error (RMSE), sensitivity, specificity, Youden’s index and accuracy based on 5-fold cross-validation. The agreement of the results between the SVM mo-del and the customized SAPS-Ⅱ model was illustrated using Bland-Altman plots. Results: A total number of 19 044 patients with AKI were included. The observed in-hospital mortality of the AKI patients was 13.58% in MIMIC-Ⅲ. The results based on the 5-fold cross validation showed that the average AUROC of the SVM model and the customized SAPS-Ⅱ model was 0.86 and 0.81, respectively (The difference between the two models was statistically significant with t=13.0, P<0.001). The average RMSE of the SVM model and the customized SAPS-Ⅱ model was 0.29 and 0.31, respectively (The difference was statistically significant with t=-9.6, P<0.001). The SVM model also outperformed the customized SAPS-Ⅱ model in terms of sensitivity and Youden’s index with significant statistical differences (P = 0.002 and <0.001, respectively).The Bland-Altman plot showed that the SVM model and the custo-mized SAPS-Ⅱ model had similar mortality prediction results when the mortality of a patient was certain, but the consistency between the mortality prediction results of the two models was poor when the mortality of a patient was with high uncertainty. Conclusion: Compared with the SAPS-Ⅱ model, the SVM model has a better performance, especially when the mortality of a patient is with high uncertainty. The SVM model is more suitable for predicting the mortality of patients with AKI in ICU and early intervention in patients with AKI in ICU. The SVM model can effectively help ICU clinicians improve the quality of me-dical treatment, which has high clinical value.

Key words: Intensive care unit, Acute kidney injury, Mortality prediction, Support vector machine

CLC Number: 

  • R181.2
[1] Yu-xin WANG,Yu-han DENG,Yin-liang TAN,Bao-hua LIU. Predictive value of stress-induced hyperglycemia on 28 d risk of all-cause death in intensive care patients [J]. Journal of Peking University (Health Sciences), 2023, 55(3): 442-449.
[2] WU Jing-yi,LIN Yu,LIN Ke,HU Yong-hua,KONG Gui-lan. Predicting prolonged length of intensive care unit stay via machine learning [J]. Journal of Peking University (Health Sciences), 2021, 53(6): 1163-1170.
[3] LIN Yu,WU Jing-yi,LIN Ke,HU Yong-hua,KONG Gui-lan. Prediction of intensive care unit readmission for critically ill patients based on ensemble learning [J]. Journal of Peking University (Health Sciences), 2021, 53(3): 566-572.
[4] MENG Zhao-ting,MU Dong-liang. Impact of oliguria during lung surgery on postoperative acute kidney injury [J]. Journal of Peking University (Health Sciences), 2021, 53(1): 188-194.
[5] Si-yu ZHANG,Xi-hui LI,Feng XIAO. Clinical features and prognosis of infective endocarditis patients with acute kidney injury [J]. Journal of Peking University(Health Sciences), 2019, 51(4): 737-741.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!