Abstract
Acute Kidney Injury (AKI) is common in the intensive care units (ICUs) and is associated with an increased risk of hospital mortality and additional healthcare-related costs. The prediction of AKI can help clinicians to identify patients’ risk of deterioration and intervene in the early stage of disease progression to prevent further damage. The development of EHR systems and the capabilities of data mining and machine learning methods have inspired us to explore novel tools for analyzing disease trajectories and provide us an opportunity for identifying early AKI onset more accurately. In this study, we applied the Attention LSTM model (ALSTM) and LightGBM models for AKI prediction at 4 prediction windows, where data imputation approaches were evaluated accordingly. The ALSTM models with Gaussian Process (GP) imputation showed higher performance in longer prediction windows, achieving AUC of 0.901, 0.890 and 0.877, and accuracy of 0.818, 0.806 and 0.797 in the 6-, 12- and 24-h model. While in the 2-h prediction model, LightGBM with linear imputation achieved the best AUC of 0.908. The deep learning models showed high potential in identifying patients at high risk of AKI earlier, which could provide information to guide personalized treatment. Early prediction of AKI could provide clinicians with decision support for early intervention and focus on the identified risk factors.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Hoste, E.A.J., Bagshaw, S.M., Bellomo, R., et al.: Epidemiology of acute kidney injury in critically ill patients: the multinational AKI-EPI study. Intensive Care Med. 41(8), 1411–1423 (2015)
Acute Kidney Injury, Mortality, Length of Stay, and Costs in Hospitalized Patients. Metcalfe, W., Simpson, M., Khan, I.H., et al. Acute renal failure requiring renal replacement therapy: incidence and outcome. QJM 95(9), 579–583 (2002)
Yamout, H., Levin, M.L., Rosa, R.M., et al.: Physician prevention of acute kidney injury. Am. J. Med. 128(9), 1001–1006 (2015)
Acute Kidney Injury Work Group: KDIGO clinical practice guideline for acute kidney injury. Kidney Inter Suppl. 2, 1–138 (2012)
Waikar, S.S., Bonventre, J.V.: Creatinine kinetics and the definition of acute kidney injury. J. Am. Soc. Nephrol. 20, 672–679 (2009). https://doi.org/10.1681/ASN.2008070669
Lehner, G.F., Forni, L.G., Joannidis, M.: Oliguria and biomarkers of acute kidney injury: star struck lovers or strangers in the night? Nephron 134, 183–190 (2016). https://doi.org/10.1159/000447979
Kashani, K., Cheungpasitporn, W., Ronco, C.: Biomarkers of acute kidney injury: the pathway from discovery to clinical adoption. Clin. Chem. Lab. Med. 55(8), 1074–1089 (2017). https://doi.org/10.1515/cclm-2016-0973
Darmon, M., Ostermann, M., Joannidis, M.: Predictions are Difficult…especially about AKI. Intensive Care Med. 43(6), 932–934 (2017). https://doi.org/10.1007/s00134-017-4720-5
Tang, Z., et al.: Embracing disease progression with a learning system for real world evidence discovery. In: Huang, D.-S., Jo, K.-H. (eds.) ICIC 2020. LNCS, vol. 12464, pp. 524–534. Springer, Cham (2020). https://doi.org/10.1007/978-3-030-60802-6_46
Flechet, M., et al.: AKIpredictor, an Online Prognostic Calculator for Acute Kidney Injury in Adult Critically Ill Patients: Development, Validation and Comparison to Serum Neutrophil Gelatinase-Associated Lipocalin. Intensive Care Med. 43(6), 764–773 (2017). https://doi.org/10.1007/s00134-017-4678-3
Mohamadlou, H., et al.: Prediction of acute kidney injury with a machine learning algorithm using electronic health record data. Can. J. Kidney Health Dis. 5, 205435811877632 (2018)
Koyner, J.L., Adhikari, R., Edelson, D.P., Churpek, M.M.: Development of a multicenter ward–based AKI prediction model. Clin. J. Am. Soc. Nephrology 11(11), 1935–1943 (2016). https://doi.org/10.2215/CJN.00280116
Tomašev, N., Glorot, X., Rae, J.W., et al.: A clinically applicable approach to continuous prediction of future acute kidney injury. Nature 572(7767), 116–119 (2019)
Su, X., et al.: Attention-based knowledge graph representation learning for predicting drug-drug interactions. Brief. Bioinf. 23(3), bbac140 (2022)
Peng, C., Zheng, Y., Huang, D.-S.: Capsule network based modeling of multi-omics data for discovery of breast cancer-related genes. IEEE/ACM Trans. Comput. Biol. Bioinf. 17(5), 1605–1612 (2020). https://doi.org/10.1109/TCBB.2019.2909905
Xiaorui, Su., Lun, Hu., You, Z., Pengwei, Hu., Wang, L., Zhao, B.: A deep learning method for repurposing antiviral drugs against new viruses via multi-view nonnegative matrix factorization and its application to SARS-CoV-2. Brief. Bioinf. 23(1), bbab526 (2022). https://doi.org/10.1093/bib/bbab526
Hu, P.-W., Chan, K.C.C., You, Z.-H.: Large-scale prediction of drug-target interactions from deep representations. In: 2016 International Joint Conference on Neural Networks (IJCNN). IEEE (2016)
Huang, Y.-a, Hu, P., Chan, K.C.C., You, Z.-H.: Graph convolution for predicting associations between miRNA and drug resistance. Bioinformatics 36(3), 851–858 (2020). https://doi.org/10.1093/bioinformatics/btz621
Luo, X., Wang, L., Hu, P., Hu, L.: Predicting protein-protein interactions using sequence and network information via variational graph autoencoder. IEEE/ACM Trans. Comput. Biol. Bioinf. (2023). https://doi.org/10.1109/TCBB.2023.3273567
He, F., Zhu, G., Wang, Y.Y., et al.: PCID: a novel approach for predicting disease comorbidity by integrating multi-scale data. IEEE/ACM Trans. Comput. Biol. Bioinf. 14(3), 678–686 (2017)
Siew, E.D., Ikizler, T.A.: Continuous prediction of future acute kidney injury: a step forward. Kidney Int. 97(6), 1094–1096 (2020)
Karim, F., Majumdar, S., Darabi, H., Harford, S.: Multivariate LSTM-FCNs for time series classification, 1–9 (2018). http://arxiv.org/abs/1801.04503
Vaswani, A., et al.: Attention is all you need. In: The 31st Conference on Neural Information Processing Systems (NeurIPS), pp. 5998–6008 (2017)
Feng, G., Quirk, J.G., Djuric, P.M.: Recovery of missing samples in fetal heart rate recordings with gaussian processes. In: 25th European Signal Processing Conference, EUSIPCO 2017 2017–January, pp. 261–265 (2017)
Johnson, A.E.W., et al.: MIMIC-III, a freely accessible critical care database. Scientific Data 3, 1–9 (2016). https://doi.org/10.1038/sdata.2016.35
Rasmussen, C.E., Williams, C.K.I.: Gaussian Processes for Machine Learning. MIT Press, Cambridge, Massachusetts (2006)
Hu, J., Shen, L., Albanie, S., Sun, G., Wu, E.: Squeeze-and-excitation networks. IEEE Trans. Pattern Anal. Mach. Intell., 1–1 (2019)
Lehman, L.-W., Feng, M., Yang, Y., Mark, R.G.: Blood pressure and the risk of acute kidney injury in the ICU: case-control versus case-crossover designs. In: Secondary Analysis of Electronic Health Records, pp. 369–375. Springer, Cham (2016). https://doi.org/10.1007/978-3-319-43742-2_25
Macedo, E., Malhotra, R., Bouchard, J., Wynn, S.K., Mehta, R.L.: Oliguria is an early predictor of higher mortality in critically ill patients. Kidney Int. 80, 760–767 (2011)
Acknowledgements
This work was supported by the Xinjiang Tianchi Talents Program (E33B9401).
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2023 The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd.
About this paper
Cite this paper
Zhang, Y. et al. (2023). A Deep Learning Approach Incorporating Data Missing Mechanism in Predicting Acute Kidney Injury in ICU. In: Huang, DS., Premaratne, P., Jin, B., Qu, B., Jo, KH., Hussain, A. (eds) Advanced Intelligent Computing Technology and Applications. ICIC 2023. Lecture Notes in Computer Science, vol 14088. Springer, Singapore. https://doi.org/10.1007/978-981-99-4749-2_29
Download citation
DOI: https://doi.org/10.1007/978-981-99-4749-2_29
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-99-4748-5
Online ISBN: 978-981-99-4749-2
eBook Packages: Computer ScienceComputer Science (R0)