Abstract:Background and objective: Real-world data encompass population diversity, enabling insights into chronic disease mortality risk among the elderly. Deep learning excels on large datasets, offering promise for real-world data. However, current models focus on single diseases, neglecting comorbidities prevalent in patients. Moreover, mortality is infrequent compared to illness, causing extreme class imbalance that impedes reliable prediction. We aim to develop a deep learning framework that accurately forecasts mortality risk from real-world data by addressing comorbidities and class imbalance. Methods: We integrated multi-task and cost-sensitive learning, developing an enhanced deep neural network architecture that extends multi-task learning to predict mortality risk across multiple chronic diseases. Each patient cohort with a chronic disease was assigned to a separate task, with shared lower-level parameters capturing inter-disease complexities through distinct top-level networks. Cost-sensitive functions were incorporated to ensure learning of positive class characteristics for each task and achieve accurate prediction of the risk of death from multiple chronic diseases. Results: Our study covers 15 prevalent chronic diseases and is experimented with real-world data from 482,145 patients (including 9,516 deaths) in Shenzhen, China. The proposed model is compared with six models including three machine learning models: logistic regression, XGBoost, and CatBoost, and three state-of-the-art deep learning models: 1D-CNN, TabNet, and Saint. The experimental results show that, compared with the other compared algorithms, MTL-CSDNN has better prediction results on the test set (ACC=0.99, REC=0.99, PRAUC=0.97, MCC=0.98, G-means = 0.98). Conclusions: Our method provides valuable insights into leveraging real-world data for precise multi-disease mortality risk prediction, offering potential applications in optimizing chronic disease management, enhancing well-being, and reducing healthcare costs for the elderly population.

Multi-task Learning via Adaptation to Similar Tasks for Mortality Prediction of Diverse Rare Diseases

A Method for the Early Prediction of Chronic Diseases Based on Short Sequential Medical Data.

Integrating multi-task and cost-sensitive learning for predicting mortality risk of chronic diseases in the elderly using real-world data

Federated Learning of Electronic Health Records Improves Mortality Prediction in Patients Hospitalized with COVID-19.

Multi-task oriented diffusion model for mortality prediction in shock patients with incomplete data

M3T-LM: A multi-modal multi-task learning model for jointly predicting patient length of stay and mortality

Mortality Prediction with Adaptive Feature Importance Recalibration for Peritoneal Dialysis Patients: a deep-learning-based study on a real-world longitudinal follow-up dataset

Contrastive Learning-based Imputation-Prediction Networks for In-hospital Mortality Risk Modeling using EHRs

Multi-disease prediction using LSTM recurrent neural networks

Multi-Task Learning for Alzheimer's Disease Diagnosis and Mini-Mental State Examination Score Prediction

Automated Multi-Task Learning for Joint Disease Prediction on Electronic Health Records

Knowledge-Aware Deep Dual Networks for Text-Based Mortality Prediction

Monitoring ICU Mortality Risk with A Long Short-Term Memory Recurrent Neural Network

Time Associated Meta Learning for Clinical Prediction

Integrating Physiological Time Series and Clinical Notes with Deep Learning for Improved ICU Mortality Prediction

A Hemodialysis Mortality Prediction Model Based on Active Contrastive Learning.

Federated Learning of Electronic Health Records to Improve Mortality Prediction in Hospitalized Patients With COVID-19: Machine Learning Approach

Dual-Core Mutual Learning Between Scoring Systems and Clinical Features for ICU Mortality Prediction

Mortality Prediction for ICU Patients Combining Just-in-time Learning and Extreme Learning Machine

Adaptive multi-task positive-unlabeled learning for joint prediction of multiple chronic diseases using online shopping behaviors