IDEAS home Printed from https://ideas.repec.org/a/plo/pone00/0246306.html
   My bibliography  Save this article

Predicting mortality of patients with acute kidney injury in the ICU using XGBoost model

Author

Listed:
  • Jialin Liu
  • Jinfa Wu
  • Siru Liu
  • Mengdie Li
  • Kunchang Hu
  • Ke Li

Abstract

Purpose: The goal of this study is to construct a mortality prediction model using the XGBoot (eXtreme Gradient Boosting) decision tree model for AKI (acute kidney injury) patients in the ICU (intensive care unit), and to compare its performance with that of three other machine learning models. Methods: We used the eICU Collaborative Research Database (eICU-CRD) for model development and performance comparison. The prediction performance of the XGBoot model was compared with the other three machine learning models. These models included LR (logistic regression), SVM (support vector machines), and RF (random forest). In the model comparison, the AUROC (area under receiver operating curve), accuracy, precision, recall, and F1 score were used to evaluate the predictive performance of each model. Results: A total of 7548 AKI patients were analyzed in this study. The overall in-hospital mortality of AKI patients was 16.35%. The best performing algorithm in this study was XGBoost with the highest AUROC (0.796, p

Suggested Citation

  • Jialin Liu & Jinfa Wu & Siru Liu & Mengdie Li & Kunchang Hu & Ke Li, 2021. "Predicting mortality of patients with acute kidney injury in the ICU using XGBoost model," PLOS ONE, Public Library of Science, vol. 16(2), pages 1-11, February.
    • Handle: RePEc:plo:pone00:0246306
    DOI: 10.1371/journal.pone.0246306
    as

    Download full text from publisher

    File URL: https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0246306
    Download Restriction: no
    File URL: https://journals.plos.org/plosone/article/file?id=10.1371/journal.pone.0246306&type=printable
    Download Restriction: no
    File URL: https://libkey.io/10.1371/journal.pone.0246306?utm_source=ideas
    LibKey link: if access is restricted and if your library uses this service, LibKey will redirect you to where you can use your library subscription to access this item
    ---><---

    References listed on IDEAS

    as
    1. Huiting Zheng & Jiabin Yuan & Long Chen, 2017. "Short-Term Load Forecasting Using EMD-LSTM Neural Networks with a Xgboost Algorithm for Feature Importance Evaluation," Energies, MDPI, vol. 10(8), pages 1-20, August.
    Full references (including those not matched with items on IDEAS)

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. Bingjie Jin & Guihua Zeng & Zhilin Lu & Hongqiao Peng & Shuxin Luo & Xinhe Yang & Haojun Zhu & Mingbo Liu, 2022. "Hybrid LSTM–BPNN-to-BPNN Model Considering Multi-Source Information for Forecasting Medium- and Long-Term Electricity Peak Load," Energies, MDPI, vol. 15(20), pages 1-20, October.
    2. Jiaan Zhang & Chenyu Liu & Leijiao Ge, 2022. "Short-Term Load Forecasting Model of Electric Vehicle Charging Load Based on MCCNN-TCN," Energies, MDPI, vol. 15(7), pages 1-25, April.
    3. Ijaz Ul Haq & Amin Ullah & Samee Ullah Khan & Noman Khan & Mi Young Lee & Seungmin Rho & Sung Wook Baik, 2021. "Sequential Learning-Based Energy Consumption Prediction Model for Residential and Commercial Sectors," Mathematics, MDPI, vol. 9(6), pages 1-17, March.
    4. Umut Ugurlu & Ilkay Oksuz & Oktay Tas, 2018. "Electricity Price Forecasting Using Recurrent Neural Networks," Energies, MDPI, vol. 11(5), pages 1-23, May.
    5. Mohamed Massaoudi & Ines Chihi & Lilia Sidhom & Mohamed Trabelsi & Shady S. Refaat & Fakhreddine S. Oueslati, 2021. "Enhanced Random Forest Model for Robust Short-Term Photovoltaic Power Forecasting Using Weather Measurements," Energies, MDPI, vol. 14(13), pages 1-20, July.
    6. Oreshkin, Boris N. & Dudek, Grzegorz & Pełka, Paweł & Turkina, Ekaterina, 2021. "N-BEATS neural network for mid-term electricity load forecasting," Applied Energy, Elsevier, vol. 293(C).
    7. Salah Bouktif & Ali Fiaz & Ali Ouni & Mohamed Adel Serhani, 2018. "Optimal Deep Learning LSTM Model for Electric Load Forecasting using Feature Selection and Genetic Algorithm: Comparison with Machine Learning Approaches †," Energies, MDPI, vol. 11(7), pages 1-20, June.
    8. Wei, Nan & Li, Changjun & Peng, Xiaolong & Li, Yang & Zeng, Fanhua, 2019. "Daily natural gas consumption forecasting via the application of a novel hybrid model," Applied Energy, Elsevier, vol. 250(C), pages 358-368.
    9. Chen Zhao & Xia Zhao & Zhao Li & Qiong Zhang, 2022. "XGBoost-DNN Mixed Model for Predicting Driver’s Estimation on the Relative Motion States during Lane-Changing Decisions: A Real Driving Study on the Highway," Sustainability, MDPI, vol. 14(11), pages 1-23, June.
    10. Shouxiang Wang & Pengfei Dong & Yingjie Tian, 2017. "A Novel Method of Statistical Line Loss Estimation for Distribution Feeders Based on Feeder Cluster and Modified XGBoost," Energies, MDPI, vol. 10(12), pages 1-17, December.
    11. Gong, Mingju & Zhao, Yin & Sun, Jiawang & Han, Cuitian & Sun, Guannan & Yan, Bo, 2022. "Load forecasting of district heating system based on Informer," Energy, Elsevier, vol. 253(C).
    12. Abdulelah Alkesaiberi & Fouzi Harrou & Ying Sun, 2022. "Efficient Wind Power Prediction Using Machine Learning Methods: A Comparative Study," Energies, MDPI, vol. 15(7), pages 1-24, March.
    13. Alexandros Menelaos Tzortzis & Sotiris Pelekis & Evangelos Spiliotis & Evangelos Karakolis & Spiros Mouzakitis & John Psarras & Dimitris Askounis, 2023. "Transfer Learning for Day-Ahead Load Forecasting: A Case Study on European National Electricity Demand Time Series," Mathematics, MDPI, vol. 12(1), pages 1-24, December.
    14. Ivana Kiprijanovska & Simon Stankoski & Igor Ilievski & Slobodan Jovanovski & Matjaž Gams & Hristijan Gjoreski, 2020. "HousEEC: Day-Ahead Household Electrical Energy Consumption Forecasting Using Deep Learning," Energies, MDPI, vol. 13(10), pages 1-29, May.
    15. Zhao, Yin & Gong, Mingju & Sun, Jiawang & Han, Cuitian & Jing, Lei & Li, Bo & Zhao, Zhixuan, 2023. "A new hybrid optimization prediction strategy based on SH-Informer for district heating system," Energy, Elsevier, vol. 282(C).
    16. Luyu Ding & Yang Lv & Ruixiang Jiang & Wenjie Zhao & Qifeng Li & Baozhu Yang & Ligen Yu & Weihong Ma & Ronghua Gao & Qinyang Yu, 2022. "Predicting the Feed Intake of Cattle Based on Jaw Movement Using a Triaxial Accelerometer," Agriculture, MDPI, vol. 12(7), pages 1-18, June.
    17. Zheng Wan & Hui Li, 2023. "Short-Term Power Load Forecasting Based on Feature Filtering and Error Compensation under Imbalanced Samples," Energies, MDPI, vol. 16(10), pages 1-22, May.
    18. Xiaolu Wei & Yubo Tian & Na Li & Huanxin Peng, 2024. "Evaluating ensemble learning techniques for stock index trend prediction: a case of China," Portuguese Economic Journal, Springer;Instituto Superior de Economia e Gestao, vol. 23(3), pages 505-530, September.
    19. Xiaosheng Peng & Kai Cheng & Jianxun Lang & Zuowei Zhang & Tao Cai & Shanxu Duan, 2021. "Short-Term Wind Power Prediction for Wind Farm Clusters Based on SFFS Feature Selection and BLSTM Deep Learning," Energies, MDPI, vol. 14(7), pages 1-18, March.
    20. Yu, Binbin & Li, Jianjing & Liu, Che & Sun, Bo, 2022. "A novel short-term electrical load forecasting framework with intelligent feature engineering," Applied Energy, Elsevier, vol. 327(C).

    More about this item

    Statistics

    Access and download statistics

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:plo:pone00:0246306. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: plosone (email available below). General contact details of provider: https://journals.plos.org/plosone/ .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.