A comparative study of machine learning algorithms for predicting acute kidney injury after liver cancer resection

PeerJ

article

Lei Lei¹  Ying Wang¹  Qiong Xue¹  Jianhua Tong¹  Cheng-Mao Zhou¹  Jian-Jun Yang¹ 

[1] Department of Anesthesiology, Pain and Perioperative Medicine, The First Affiliated Hospital of Zhengzhou University

关键词: Machine learning;    AKI;    Hepatectomy;    Postoperative;    Secondary analysis;   

DOI  :  10.7717/peerj.8583

学科分类：社会科学、人文和艺术（综合）

来源: Inra

【 摘 要 】

ObjectiveMachine learning methods may have better or comparable predictive ability than traditional analysis. We explore machine learning methods to predict the likelihood of acute kidney injury after liver cancer resection.MethodsThis is a secondary analysis cohort study. We reviewed data from patients who had undergone resection of primary hepatocellular carcinoma between January 2008 and October 2015.ResultsThe analysis included 1,173 hepatectomy patients, 77 (6.6%) of whom had AKI and 1,096 (93.4%) who did not. The importance matrix for the Gbdt algorithm model shows that age, cholesterol, tumor size, surgery duration and PLT were the five most important parameters. Figure 1 shows that Age, tumor size and surgery duration had weak positive correlations with AKI. Cholesterol and PLT also had weak negative correlations with AKI. The models constructed by the four machine learning algorithms in the training group were compared. Among the four machine learning algorithms, random forest and gbm had the highest accuracy, 0.989 and 0.970 respectively. The precision of four of the five algorithms was 1, random forest being the exception. Among the test group, gbm had the highest accuracy (0.932). Random forest and gbm had the highest precision, both being 0.333. The AUC values for the four algorithms were: Gbdt (0.772), gbm (0.725), forest (0.662) and DecisionTree (0.628).ConclusionsMachine learning technology can predict acute kidney injury after hepatectomy. Age, cholesterol, tumor size, surgery duration and PLT influence the likelihood and development of postoperative acute kidney injury.

【 授权许可】

CC BY