OA-MedSQL: Order-Aware Medical Sequence Learning for Clinical Outcome Prediction

Tong Wu,Yue Wang,Yunlong Wang,Emily Zhao,Gao Wang
DOI: https://doi.org/10.1109/bibm52615.2021.9669367
2021-01-01
Abstract:Leveraging large amounts of longitudinal electronic health records (EHR) data has shown great potentials for predicting progression of various diseases. In this work, we proposed a deep learning algorithm based on a customized Long Short-Term Memory (LSTM) model with built-in temporal sequence learning mechanism, named as Order-Aware Medical SeQuence Learning (OA-MedSQL). The model can predict various clinical outcomes from EHR sequences with competitive accuracies, meanwhile automatically distills patient medical journeys into clinically meaningful and relevant temporal sequences. The extraction of milestone temporal sequences is mainly facilitated by enforcing an order to the neurons of the LSTM model when updating their cell states at each time step. We have evaluated the proposed approach on a real-world EHR dataset that focuses on the task of predicting first-line treatment initiation for Waldenström Macroglobulinemia (WM) patients. OA-MedSQL can achieve comparable or better performance than several competitive baseline methods. Furthermore, OA-MedSQL offers a native way to extract milestone temporal sequences that are both explanatory of patient journeys and predictive of clinical outcomes.
What problem does this paper attempt to address?