Back

KIT-LSTM: Knowledge-guided Time-aware LSTM for Continuous Clinical Risk Prediction

Liu, L. J.; Ortiz-Soriano, V.; Neyra, J. A.; Chen, J.

2022-11-18 health informatics
10.1101/2022.11.14.22282332 medRxiv
Show abstract

Rapid accumulation of temporal Electronic Health Record (EHR) data and recent advances in deep learning have shown high potential in precisely and timely predicting patients risks using AI. However, most existing risk prediction approaches ignore the complex asynchronous and irregular problems in real-world EHR data. This paper proposes a novel approach called Knowledge-guIded Time-aware LSTM (KIT-LSTM) for continuous mortality predictions using EHR. KIT-LSTM extends LSTM with two time-aware gates and a knowledge-aware gate to better model EHR and interprets results. Experiments on real-world data for patients with acute kidney injury with dialysis (AKI-D) demonstrate that KIT-LSTM performs better than the state-of-the-art methods for predicting patients risk trajectories and model interpretation. KIT-LSTM can better support timely decision-making for clinicians.

Matching journals

The top 4 journals account for 50% of the predicted probability mass.

50% of probability mass above

"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.