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.

Kullback-Leibler-Based Discrete Failure Time Models for Integration of Published Prediction Models with New Time-To-Event Dataset

Learning from Local to Global - an Efficient Distributed Algorithm for Modeling Time-to-event Data

Assessing the Effect of Data Integration on Predictive Ability of Cancer Survival Models

A latent variable model for survival time prediction with censoring and diverse covariates

Semiparametric predictive inference for failure data using first-hitting-time threshold regression

Learning the Joint Representation of Heterogeneous Temporal Events for Clinical Endpoint Prediction

Prediction of Future Failures for Heterogeneous Reliability Field Data

Deep State-Space Generative Model For Correlated Time-to-Event Predictions

Efficient Risk Assessment of Time‐to‐Event Targets With Adaptive Information Transfer

Improving Event Time Prediction by Learning to Partition the Event Time Space

Dynamic Prediction of Competing Risk Events using Landmark Sub-distribution Hazard Model with Multiple Longitudinal Biomarker

Dynamic Prediction for Multiple Repeated Measures and Event Time Data: An Application to Parkinson's Disease

Leveraging Patient Similarity and Time Series Data in Healthcare Predictive Models

Scalable Modeling of Multivariate Longitudinal Data for Prediction of Chronic Kidney Disease Progression

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

Discrimination, calibration, and point estimate accuracy of GRU-D-Weibull architecture for real-time individualized endpoint prediction

Nonparametric failure time: Time‐to‐event machine learning with heteroskedastic bayesian additive regression trees and low information omnibus dirichlet process mixtures

predicTTE: An accessible and optimal tool for time-to-event prediction in neurological diseases

Adaptive Prediction Timing for Electronic Health Records

Risk Projection for Time-to-Event Outcome Leveraging Summary Statistics With Source Individual-Level Data

Discrete-time Competing-Risks Regression with or without Penalization