Related Experiment Videos
Cardiovascular disease prediction using variational recurrent autoencoders with uncertainty estimation
Ashim Chandra Das1, Md Shujan Shak1, Nabila Rahman2
1University of the Potomac, Washington, VA, 22043, USA.
Insights
This study introduces a deep learning Variational Recurrent Autoencoder (VRAE) for cardiovascular disease (CVD) classification from static clinical data. The VRAE model achieves high accuracy and provides reliable uncertainty estimates for improved diagnostic performance.
Area of Science:
- Cardiovascular research
- Machine learning in healthcare
- Biomedical informatics
Background:
- Cardiovascular diseases (CVDs) are a major global cause of mortality.
- Accurate classification and reliable probability estimates are crucial for CVD benchmark studies.
- Existing methods may lack robustness in handling noisy or incomplete clinical data.
Purpose of the Study:
- To propose a deep learning framework for cardiovascular disease classification using static tabular clinical records.
- To develop a model that provides calibrated probability estimates and uncertainty quantification.
- To evaluate the model's performance and robustness against various baselines.
Main Methods:
- A Variational Recurrent Autoencoder (VRAE) framework was developed, integrating Gated Recurrent Unit (GRU) encoding, variational latent sampling, and Monte Carlo (MC) dropout.
- Static tabular datasets were transformed into synthetic pseudo-sequences to simulate noise and missingness for denoising latent representation learning.
- The model was evaluated on the Heart Failure Prediction and Cardiovascular Disease datasets.
Main Results:
- The VRAE model achieved high accuracy (95.8% and 96.1%) and F1-scores (95.7% and 96.0%) on both datasets.
- It outperformed traditional classifiers and other deep learning models, including LSTM and GRU-Attention.
- The model demonstrated robustness under synthetic noise and achieved the lowest Brier scores, indicating improved calibration.
Conclusions:
- Uncertainty-aware representation learning using the VRAE framework is effective for benchmark-level static tabular CVD classification.
- The model offers reliable performance and improved calibration compared to existing methods.
- Further validation on real-world longitudinal electronic health record cohorts is recommended for clinical deployment.
Abstract:
Cardiovascular diseases (CVDs) remain a leading cause of mortality worldwide, motivating reproducible benchmark studies on accurate classification methods with calibrated probability estimates. In this paper, we propose a deep learning framework based on a Variational Recurrent Autoencoder (VRAE) with uncertainty estimation for CVD classification from static tabular clinical records represented as synthetic noise-augmented pseudo-sequences. The two datasets used in this study are static tabular datasets rather than real longitudinal clinical time-series. Therefore, the constructed pseudo-sequence dimension is used for denoising latent representation learning under simulated measurement perturbation and missingness, not for modeling observed clinical temporal dependencies, patient trajectories, disease progression, or treatment dynamics. The model integrates Gated Recurrent Unit (GRU)-based encoding, variational latent sampling, and Monte Carlo (MC) dropout to learn robust latent representations and quantify predictive uncertainty. We evaluate the model on two publicly available datasets: the Heart Failure Prediction dataset (918 samples) and the Cardiovascular Disease dataset (70,000 samples). The model achieves an accuracy of 95.8% and F1-score of 95.7% on the heart failure dataset, and an accuracy of 96.1% with F1-score of 96.0% on the larger cardiovascular dataset, outperforming traditional classifiers, calibrated tabular baselines, and deep learning baselines including regularized logistic regression, calibrated XGBoost, calibrated LightGBM, calibrated CatBoost, Long Short-Term Memory (LSTM), and GRU-Attention. The proposed VRAE model also demonstrates robustness under synthetic noise and provides improved calibration, as indicated by the lowest Brier scores (0.061 and 0.059) across both datasets. Additional calibration, selective prediction, and threshold-based utility analyses support the reliability of the benchmark results. These findings support uncertainty-aware representation learning for benchmark-level static tabular CVD classification, while independent validation on real-world longitudinal electronic health record cohorts is required before claims about prospective risk prediction, early intervention, clinical time-series prediction, or clinical deployment can be made.
Related Concept Videos
Prediction Intervals
However, the point estimate is most likely not the exact value of the population parameter, but close to it. After calculating point estimates, we construct interval estimates, called confidence intervals or prediction intervals. This prediction interval comprises a range of values unlike the point estimate and is a better predictor of the observed sample value, y.
The...
Propagation of Uncertainty from Random Error
Propagation of Uncertainty from Systematic Error
Uncertainty: Overview
Uncertainty: Confidence Intervals