Heart disease prediction using hybrid TabNet architecture with stacked ensemble learning
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.
Authors
- Ahyoung Choi (ORCID: https://orcid.org/0000-0001-7676-9869)
- Rizwana Yasmeen (ORCID: https://orcid.org/0000-0002-8590-7773)
- Lal Khan (ORCID: https://orcid.org/0000-0002-1624-7580)
Institutions
- Gachon University (KR)
- National University of Modern Languages (PK)
Publication Details
- Journal
- Frontiers in Physiology
- Published
- 2025-11-05
- DOI
- https://doi.org/10.3389/fphys.2025.1665128
- Citations
- 1
- Primary Topic
- Artificial Intelligence in Healthcare
- Type
- article
- Field-Weighted Citation Impact
- 2.94
Funders
- Korea Health Industry Development Institute
- National Institutes of Health