A BLSTM with Attention Network for Predicting Acute Myeloid Leukemia Patient's Prognosis using Comprehensive Clinical Parameters.

Abstract

The prognosis management is crucial for highrisk disease like Acute Myeloid Leukemia (AML) in order to support decisions of clinical treatment. However, the challenges of accurate and consistent forecasting lie in the high variability of the disease outcomes and the complexity of the multiple clinical measurements available over the course of the treatment. In order to capture the multi-dimensional and longitudinal aspect of these comprehensive clinical parameters, we utilize an attention-based bi-directional long shortterm memory (Att-BLSTM) network to predict AML patient's survival and relapse. Specifically, we gather a 10-year worth of real patient's clinical data including blood test, medication, HSCT status, and gene mutation information. Our proposed Att-BLSTM framework achieves 77.1% and 67.3% AUC in tasks of predicting the next 2-year mortality and disease relapse with these comprehensive clinical parameters, and our further analysis demonstrates that a next 0 to 3 months prediction performs equally well, i.e., 74.8% and 67% AUC for mortality and relapse respectively.