Prediction of the Mortality Risk in Peritoneal Dialysis Patients using Machine Learning Models: A Nation-wide Prospective Cohort in Korea

Junhyug Noh, Kyung Don Yoo, Wonho Bae, Jong Soo Lee, Kangil Kim, Jang Hee Cho, Hajeong Lee, Dong Ki Kim, Chun Soo Lim, Shin Wook Kang, Yong Lim Kim, Yon Su Kim, Gunhee Kim, Jung Pyo Lee

Research output: Contribution to journalArticlepeer-review

1 Scopus citations

Abstract

Herein, we aim to assess mortality risk prediction in peritoneal dialysis patients using machine-learning algorithms for proper prognosis prediction. A total of 1,730 peritoneal dialysis patients in the CRC for ESRD prospective cohort from 2008 to 2014 were enrolled in this study. Classification algorithms were used for prediction of N-year mortality including neural network. The survival hazard ratio was presented by machine-learning algorithms using survival statistics and was compared to conventional algorithms. A survival-tree algorithm presented the most accurate prediction model and outperformed a conventional method such as Cox regression (concordance index 0.769 vs 0.745). Among various survival decision-tree models, the modified Charlson Comorbidity index (mCCI) was selected as the best predictor of mortality. If peritoneal dialysis patients with high mCCI (>4) were aged ≥70.5 years old, the survival hazard ratio was predicted as 4.61 compared to the overall study population. Among the various algorithm using longitudinal data, the AUC value of logistic regression was augmented at 0.804. In addition, the deep neural network significantly improved performance to 0.841. We propose machine learning-based final model, mCCI and age were interrelated as notable risk factors for mortality in Korean peritoneal dialysis patients.

Original languageEnglish
Article number7470
JournalScientific Reports
Volume10
Issue number1
DOIs
StatePublished - 1 Dec 2020

Fingerprint Dive into the research topics of 'Prediction of the Mortality Risk in Peritoneal Dialysis Patients using Machine Learning Models: A Nation-wide Prospective Cohort in Korea'. Together they form a unique fingerprint.

Cite this