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

Objective Machine 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. Methods This is a secondary analysis cohort study. We reviewed data from p...

Full description

Bibliographic Details
Main Authors: Lei Lei, Ying Wang, Qiong Xue, Jianhua Tong, Cheng-Mao Zhou, Jian-Jun Yang
Format: Article
Language:English
Published: PeerJ Inc. 2020-02-01
Series:PeerJ
Subjects:
Online Access:https://peerj.com/articles/8583.pdf
_version_ 1797419843882844160
author Lei Lei
Ying Wang
Qiong Xue
Jianhua Tong
Cheng-Mao Zhou
Jian-Jun Yang
author_facet Lei Lei
Ying Wang
Qiong Xue
Jianhua Tong
Cheng-Mao Zhou
Jian-Jun Yang
author_sort Lei Lei
collection DOAJ
description Objective Machine 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. Methods This 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. Results The 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). Conclusions Machine 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.
first_indexed 2024-03-09T06:54:05Z
format Article
id doaj.art-5807fbd1b159469e9c8a27153880b77e
institution Directory Open Access Journal
issn 2167-8359
language English
last_indexed 2024-03-09T06:54:05Z
publishDate 2020-02-01
publisher PeerJ Inc.
record_format Article
series PeerJ
spelling doaj.art-5807fbd1b159469e9c8a27153880b77e2023-12-03T10:14:09ZengPeerJ Inc.PeerJ2167-83592020-02-018e858310.7717/peerj.8583A comparative study of machine learning algorithms for predicting acute kidney injury after liver cancer resectionLei Lei0Ying Wang1Qiong Xue2Jianhua Tong3Cheng-Mao Zhou4Jian-Jun Yang5Department of Anesthesiology, Pain and Perioperative Medicine, The First Affiliated Hospital of Zhengzhou University, Zhengzhou, ChinaDepartment of Anesthesiology, Pain and Perioperative Medicine, The First Affiliated Hospital of Zhengzhou University, Zhengzhou, ChinaDepartment of Anesthesiology, Pain and Perioperative Medicine, The First Affiliated Hospital of Zhengzhou University, Zhengzhou, ChinaDepartment of Anesthesiology, Pain and Perioperative Medicine, The First Affiliated Hospital of Zhengzhou University, Zhengzhou, ChinaDepartment of Anesthesiology, Pain and Perioperative Medicine, The First Affiliated Hospital of Zhengzhou University, Zhengzhou, ChinaDepartment of Anesthesiology, Pain and Perioperative Medicine, The First Affiliated Hospital of Zhengzhou University, Zhengzhou, ChinaObjective Machine 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. Methods This 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. Results The 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). Conclusions Machine 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.https://peerj.com/articles/8583.pdfMachine learningAKIHepatectomyPostoperativeSecondary analysis
spellingShingle Lei Lei
Ying Wang
Qiong Xue
Jianhua Tong
Cheng-Mao Zhou
Jian-Jun Yang
A comparative study of machine learning algorithms for predicting acute kidney injury after liver cancer resection
PeerJ
Machine learning
AKI
Hepatectomy
Postoperative
Secondary analysis
title A comparative study of machine learning algorithms for predicting acute kidney injury after liver cancer resection
title_full A comparative study of machine learning algorithms for predicting acute kidney injury after liver cancer resection
title_fullStr A comparative study of machine learning algorithms for predicting acute kidney injury after liver cancer resection
title_full_unstemmed A comparative study of machine learning algorithms for predicting acute kidney injury after liver cancer resection
title_short A comparative study of machine learning algorithms for predicting acute kidney injury after liver cancer resection
title_sort comparative study of machine learning algorithms for predicting acute kidney injury after liver cancer resection
topic Machine learning
AKI
Hepatectomy
Postoperative
Secondary analysis
url https://peerj.com/articles/8583.pdf
work_keys_str_mv AT leilei acomparativestudyofmachinelearningalgorithmsforpredictingacutekidneyinjuryafterlivercancerresection
AT yingwang acomparativestudyofmachinelearningalgorithmsforpredictingacutekidneyinjuryafterlivercancerresection
AT qiongxue acomparativestudyofmachinelearningalgorithmsforpredictingacutekidneyinjuryafterlivercancerresection
AT jianhuatong acomparativestudyofmachinelearningalgorithmsforpredictingacutekidneyinjuryafterlivercancerresection
AT chengmaozhou acomparativestudyofmachinelearningalgorithmsforpredictingacutekidneyinjuryafterlivercancerresection
AT jianjunyang acomparativestudyofmachinelearningalgorithmsforpredictingacutekidneyinjuryafterlivercancerresection
AT leilei comparativestudyofmachinelearningalgorithmsforpredictingacutekidneyinjuryafterlivercancerresection
AT yingwang comparativestudyofmachinelearningalgorithmsforpredictingacutekidneyinjuryafterlivercancerresection
AT qiongxue comparativestudyofmachinelearningalgorithmsforpredictingacutekidneyinjuryafterlivercancerresection
AT jianhuatong comparativestudyofmachinelearningalgorithmsforpredictingacutekidneyinjuryafterlivercancerresection
AT chengmaozhou comparativestudyofmachinelearningalgorithmsforpredictingacutekidneyinjuryafterlivercancerresection
AT jianjunyang comparativestudyofmachinelearningalgorithmsforpredictingacutekidneyinjuryafterlivercancerresection