Predicting acute kidney injury in cancer patients using heterogeneous and irregular data.

Abstract

How can we predict the occurrence of acute kidney injury (AKI) in cancer patients based on machine learning with serum creatinine data? Given irregular and heterogeneous clinical data, how can we make the most of it for accurate AKI prediction? AKI is a common and significant complication in cancer patients, and correlates with substantial morbidity and mortality. Since no effective treatment for AKI still exists, it is important to take timely preventive measures. While several approaches have been proposed for predicting AKI, their scope and applicability are limited as they either assume regular data measured over a short hospital stay, or do not fully utilize heterogeneous data. In this paper, we provide an AKI prediction model with a greater applicability, which relaxes the constraints of existing approaches, and fully utilizes irregular and heterogeneous data for learning the model. In a cohort of 21,022 cancer patients who were registered into Korea Central Cancer Registry (KCCR) in Seoul National University Hospital between January 1, 2004 and December 31, 2013, our method achieves 0.7892 precision, 0.7506 recall, and 0.7576 F-measure in predicting whether a patient will develop AKI during the next 14 days.