Abstract:

Acute kidney injury is a clinical disease with a high morbidity rate, and early identification of potential patients can facilitate medical interventions to reduce morbidity and mortality. In recent years, electronic health records have been widely used to predict an individual’s potential risk. Most of the existing acute kidney injury prediction models tackle the issue of sparsity and irregularity in the physiological variables data by aggregating data or imputing the missing value, but ignore the patient’s health status implied by the missing information. Moreover, they do not consider the characteristics of and correlation between the various modalities. To solve the above issues, we present a multi-modal disease prediction model for acute kidney injury. The proposed model considers a variety of modal data, including physiological variables, disease, and demographic data. A new mask and time span based long short term memory (LSTM) network is designed to learn the time span and missing information of individual Physiological variables, and furthermore, to capture their numerical changes and frequency changes. The multi-head self-attention mechanism is introduced to promote interaction learning of each modality representation. Experiments on the real-world application of acute kidney injury risk prediction and mortality risk prediction demonstrate the effectiveness and rationality of the proposed model.

Key words: acute kidney injury, electronic health records, multimodal, time series data classification