HybridTabNet-QC: A Transformer-Based Clinical Feature Fusion Framework for Heart Disease Risk Prediction

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.

Authors

Institutions

Publication Details

Journal
IEEE Open Journal of the Computer Society
Published
2025-11-26
DOI
https://doi.org/10.1109/ojcs.2025.3637308
Primary Topic
Machine Learning in Healthcare
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

HybridTabNet-QC: A Transformer-Based Clinical Feature Fusion Framework for Heart Disease Risk Prediction

Nabila Rahman, Md. Jakir Hossen, M. F. Mridha, Arifa Akter Eva et al.
IEEE Open Journal of the Computer Society
Machine Learning in Healthcare
article

HybridTabNet-QC: A Transformer-Based Clinical Feature Fusion Framework for Heart Disease Risk Prediction

Nabila Rahman, Md. Jakir Hossen, M. F. Mridha, Arifa Akter Eva, Md Shujan Shak, Fuad Mahmud, Ashim Chandra Das
article en

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.

IEEE Open Journal of the Computer SocietyVol. 7
American International University-Bangladesh (BD), Multimedia University (MY), University of the Potomac (US), University of Detroit Mercy (US), Gannon University (US)
Openalex Percentile: Top 80%
Machine Learning in Healthcare
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.