Abstract:Deep learning methods have achieved success in disease prediction using electronic health records (EHR) data. Most of the existing methods have some limitations. First, most of the methods adopt a homogeneous decay way to deal with the effect of time interval on patient's previous visits information. However, the effect of the time interval between patient's visits is not always negative. For example, although the time interval between visits for patients with chronic diseases is relatively long, the importance of the previous visit to the next visit is high, and we may not be able to consider the effect of the time interval as negative at this point. That is, the effect of the time interval on previous visits is exerted in a nonmonotonic manner, and it is either positive, negative, or neutral. In addition, the effect of text information on prediction results is not taken into account in most of methods. The text in EHR contains a description of the patient's past medical history and current symptoms of the disease, which is important for prediction results. In order to solve these issues, we propose a Time Interval Uncertainty-Aware and Text-Enhanced Based Disease Prediction Model, which utilizes the uncertain effects of time intervals and patient's text information for disease prediction. Firstly, we apply a cross-attention mechanism to generate a global representation of the patient using the patient's disease and text information from the EHR. Then, we use the key-query attention mechanism to obtain the two importance weights of the two visit sequences with and without time intervals, respectively. Furthermore, we achieve disease prediction by making slight adjustments to the encode part of the Transformer, a deep learning model based on a self-attention mechanism. We compare with various state-of-the-art models on two publicly available datasets, MIMIC-III and MIMIC-IV, and select the top 10 diseases with the highest frequency in the dataset as the target diseases. On the MIMIC-III dataset, our model is up to three percent higher than the optimal baseline in terms of evaluation metrics.

A Knowledge-Guided Method for Disease Prediction Based on Attention Mechanism

Deep Knowledge Reasoning Guided Disease Prediction.

A Knowledge-based and Data-driven Approach for Predicting Acute Kidney Injury in Patients with Heart Failure.

Interpretable Disease Prediction based on Reinforcement Path Reasoning over Knowledge Graphs

Disease Prediction Model Based on BiLSTM and Attention Mechanism

Time-aware Heterogeneous Graph Transformer with Adaptive Attention Merging for Health Event Prediction

Interpretable Disease Progression Prediction Based on Reinforcement Reasoning over a Knowledge Graph

EHR coding with hybrid attention and features propagation on disease knowledge graph

Stage-Aware Hierarchical Attentive Relational Network for Diagnosis Prediction

Time interval uncertainty-aware and text-enhanced based disease prediction

Predicting disease genes based on multi-head attention fusion

Mdpg: a novel multi-disease diagnosis prediction method based on patient knowledge graphs

Learning by Reasoning: an Explainable Hierarchical Association Regularized Deep Learning Method for Disease Prediction.

Context-Aware Health Event Prediction via Transition Functions on Dynamic Disease Graphs

Predictive Modeling of Clinical Events with Mutual Enhancement Between Longitudinal Patient Records and Medical Knowledge Graph

Knowledge enhanced attention aggregation network for medicine recommendation

A knowledge graph-based disease-gene prediction system using multi-relational graph convolution networks

Optimizing Disease Prediction with Artificial Intelligence Driven Feature Selection and Attention Networks

RDKG: A Reinforcement Learning Framework for Disease Diagnosis on Knowledge Graph

Detecting Mental and Physical Disorders Using Multi-Task Learning Equipped with Knowledge Graph Attention Network

Explainable Prediction of Medical Codes With Knowledge Graphs