PulseExploreJournal ClubDebatesTrendingResearchersJournals
Instagram
HomeExploreJournal ClubTrending
Synapse
⌘+K
Synapse
November 5, 2025Frontiers in Physiology4 citationsOpen Access

Heart disease prediction using hybrid TabNet architecture with stacked ensemble learning

RYRizwana YasmeenLKLal KhanACAhyoung Choi

Structured PICO

Does a stacked ensemble framework integrating TabNet and XGBoost improve cardiovascular disease risk prediction compared to baseline models?

P
Population
Kaggle and UCI cardiovascular disease (CVD) datasets
I
Intervention
Stacked ensemble framework integrating TabNet and XGBoost with Logistic Regression (LR) or Support Vector Machine (SVM) as a meta learner
C
Comparator
Baseline machine learning models
O
Outcome
Prediction performance metrics (accuracy, F1-score, precision, recall, ROC-AUC, PR-AUC, and Matthews correlation coefficient)

A hybrid machine learning model combining TabNet and XGBoost improves the accuracy of cardiovascular disease risk prediction compared to baseline models.

Abstract

Cardiovascular diseases (CVDs) remain the leading cause of death worldwide, and early detection is critical for timely intervention and improved patient outcomes. However, current prediction tools are often limited by noisy, heterogeneous patient data and modest accuracy. To address this challenge, we propose a stacked ensemble framework that integrates: TabNet, a deep learning model that can identify the most relevant clinical features, and XGBoost, a powerful tree-based method known for its robustness. Their outputs are integrated using a Logistic Regression (LR) or Support Vector Machine (SVM) as meta learner, creating a system that balances accuracy and interpretability. Testing on Kaggle and UCI CVD datasets demonstrate that our ensemble consistently outperforms baseline models across accuracy, F1-score, precision, recall, ROC-AUC, PR-AUC, and matthews correlation coefficient (MCC). These results suggest that combining deep learning with tree-based models offers a practical way to improve risk prediction, supporting clinicians in making more reliable decisions for early CVD detection.

Ask AI
Helpful
Bookmark
Share
View Full Paper

Cite This Study

Yasmeen et al. (2025) studied this question.

synapsesocial.com/papers/6a1bd49c26cb5670aa9cf953https://doi.org/10.3389/fphys.2025.1665128
Ask AI
Helpful
Bookmark
Share
View Full Paper