Predicting acute kidney injury via interpretable ensemble learning and attention weighted convoutional-recurrent neural networks

Yu Chung Peng, Niharika Shimona D'Souza, Brian Bush, Charles Brown, Archana Venkataraman

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

Acute Kidney Injury (AKI) is one of the most frequent postoperative complications and is associated with both short- and long-term mortality. Improved prediction of AKI is crucial and may help clinicians prevent and mitigate its adverse effects. In this paper, we explore the use of machine learning methods to predict postoperative AKI. Our analysis centers on the ensemble-based random forest (RF) classifier, which operates on static clinical variables, and a novel deep learning architecture that incorporates intraoperative time series data along with the static variables. The architecture uses a dual-attention mechanism to select both features and time intervals relevant for AKI prediction. We evaluate our models on the publicly available VitalDB database of 3, 640 patients who underwent non-cardiac surgery. The RF outperformed existing machine learning classifiers in the AKI literature (AUROC: 0.86, AUPRC: 0.54). In addition, the RF identified a robust set of preoperative variables that can be screened in a simple blood test. While the deep learning model achieved slightly lower performance (AUROC: 0.84, AUPRC: 0.44), the attention weights provide important intraoperative information, which can be monitored by clinicians during surgery. Taken together, our results highlight the promise of machine learning for AKI prediction and take the first steps towards developing clinically translatable models.

Original languageEnglish (US)
Title of host publication2021 55th Annual Conference on Information Sciences and Systems, CISS 2021
PublisherInstitute of Electrical and Electronics Engineers Inc.
ISBN (Electronic)9781665412681
DOIs
StatePublished - Mar 24 2021
Event55th Annual Conference on Information Sciences and Systems, CISS 2021 - Baltimore, United States
Duration: Mar 24 2021Mar 26 2021

Publication series

Name2021 55th Annual Conference on Information Sciences and Systems, CISS 2021

Conference

Conference55th Annual Conference on Information Sciences and Systems, CISS 2021
Country/TerritoryUnited States
CityBaltimore
Period3/24/213/26/21

Keywords

  • Acute Kidney Injury
  • Convolutional Neural Network
  • Random Forest
  • Recurrent Neural Network

ASJC Scopus subject areas

  • Artificial Intelligence
  • Computer Networks and Communications
  • Computer Science Applications
  • Information Systems
  • Information Systems and Management

Fingerprint

Dive into the research topics of 'Predicting acute kidney injury via interpretable ensemble learning and attention weighted convoutional-recurrent neural networks'. Together they form a unique fingerprint.

Cite this