The Dual-Branch FT-Transformer framework achieved an F1-score of 0.43, recall of 0.59, and AUPRC of 0.38, reducing false negatives by >50% compared to LightGBM for cardiovascular risk prediction.
Does a precision-aware Dual-Branch FT-Transformer framework improve cardiovascular risk prediction performance compared to baseline models?
A novel dual-branch FT-Transformer framework provides a robust approach for imbalanced tabular prediction in population-scale cardiovascular risk assessment, improving recall without excessive false positives.
Effect estimate: F1-score 0.43, recall 0.59, AUPRC 0.38
Heart disease remains one of the leading causes of mortality worldwide, creating a strong need for reliable population-scale risk prediction models for large-scale screening and preventive monitoring. However, existing machine learning and deep learning approaches often struggle under severe class imbalance, data leakage risks, and unstable precision–recall trade-offs, limiting reliability in population-scale health-monitoring settings. To address these challenges, this study proposes a precision-aware Dual-Branch FT-Transformer framework for cardiovascular risk prediction using the BRFSS-2024 dataset. The proposed architecture separates recall-oriented detection and precision-oriented verification through two specialized prediction heads and integrates them using a lightweight gating mechanism trained strictly within training folds to prevent information leakage and enable controlled error arbitration. Under a strict leakage-safe 5-fold cross-validation protocol, the proposed model achieves an F1-score of 0.43, recall of 0.59, and AUPRC of 0.38 at a fixed threshold of 0.50 while reducing false negatives by more than 50% compared to LightGBM without excessive false positives. Although some baseline models achieve higher AUROC values, the proposed framework demonstrates more balanced and clinically meaningful precision–recall behaviour at operational screening thresholds. Additional evaluation on an independent NHANES cohort under the same leakage-safe re-training protocol further suggests robustness across heterogeneous population-health settings. Overall, the proposed dual-objective learning framework provides a practical and robust approach for imbalanced tabular prediction in population-scale cardiovascular risk assessment.
Islam et al. (2026) studied Heart disease. Dual-Branch FT-Transformer framework vs. LightGBM and other baseline models was evaluated on Cardiovascular risk prediction performance (F1-score, recall, AUPRC) (F1-score 0.43, recall 0.59, AUPRC 0.38). The Dual-Branch FT-Transformer framework achieved an F1-score of 0.43, recall of 0.59, and AUPRC of 0.38, reducing false negatives by >50% compared to LightGBM for cardiovascular risk prediction.