Citation

Mahmud, Fuad and Das, Ashim Chandra and Shak, Md Shujan and Rahman, Nabila and Eva, Arifa Akter and Mridha, M. F. and Hossen, Md. Jakir (2025) HybridTabNet-QC: A Transformer-Based Clinical Feature Fusion Framework for Heart Disease Risk Prediction. IEEE Open Journal of the Computer Society. pp. 1-12. ISSN 2644-1268

Text HybridTabNet-QC_ A Transformer-Based Clinical Feature Fusion Framework for Heart Disease Risk Prediction.pdf - Published Version Restricted to Repository staff only Download (1MB)

Abstract

Heart disease remains the leading cause of mortality worldwide, highlighting the need for accurate and interpretable risk prediction models. This study proposes Hybrid Tabular Network with Clinical meta-features and Quality-Control Mechanisms (HybridTabNet-QC), a hybrid deep learning framework designed to enhance clinical decision support systems. By combining a transformer-based tabular encoder with medically engineered meta-features, the model delivers interpretable, generalizable, and noise-robust predictions for heart disease risk. Unlike conventional deep learning models that treat features as raw inputs, HybridTabNet-QC fuses attention-based learning with domain-specific indicators, such as body mass index (BMI), pulse pressure, and cholesterol-glucose interaction scores that ensure predictions are both statistically sound and clinically grounded. The model was evaluated on two publicly available datasets, the Heart Disease dataset and the Cardiovascular Disease dataset from Kaggle. On the combined dataset, HybridTabNet-QC achieved 90.1% accuracy, 90.0% F1-score, and 93.6% AUC-ROC, outperforming traditional machine learning and standard deep learning baselines. The framework demonstrated strong robustness under feature noise and consistent generalization on external validation subsets. Interpretability analyses using LIME and SHAP confirmed that the model prioritizes clinically relevant features, supporting its suitability for real-world clinical decision support systems. These findings demonstrate the model’s suitability for deployment in real-world healthcare environments where interpretability, robustness, and scalability are critical.

Item Type:	Article
Uncontrolled Keywords:	Deep learning, heart disease prediction, medical AI
Subjects:	Q Science > Q Science (General) > Q300-390 Cybernetics R Medicine > RC Internal medicine
Divisions:	Faculty of Engineering and Technology (FET)
Depositing User:	Ms Rosnani Abd Wahab
Date Deposited:	22 Dec 2025 05:43
Last Modified:	22 Dec 2025 05:43
URII:	http://shdl.mmu.edu.my/id/eprint/15106

Downloads

Edit (login required)