StackTrans–Multimodal Heart Disease Detection Using Stacked Transformer Fusion Framework
Cardiac disease diagnosis demands precise interpretation of complex physiological signals, yet existing systems often rely on unimodal data and lack adaptive fusion strategies. Most conventional frameworks fall short in capturing intermodal dependencies and adjusting to performance variations across heterogeneous inputs. This study introduces StackTrans, a transformer-based multimodal classification framework designed to improve diagnostic accuracy through ECG and PCG signal integration. The architecture comprises modality-specific transformer encoders, a bidirectional cross-modal fusion transformer that facilitates latent-level attention between modalities, and a stacked ensemble mechanism governed by a meta-learner. Residual learning modules enhance prediction refinement, while entropy-guided adaptive voting improves confidence-weighted decision reliability. The PCG and ECG modules are independently trained using the PhysioNet 2016 and MIT-BIH Arrhythmia datasets, respectively, and integrated through joint inference. Evaluations using TensorFlow on an NVIDIA RTX GPU demonstrate that StackTrans attains a precision of 98.6%, an F1 score of 98.4%, and an AUC of 0.99—outperforming unimodal ECG and PCG models by 2.5% and 7.5%, respectively.
Authors
- Yang Yi (ORCID: https://orcid.org/0000-0003-0926-4523)
- Md Niaz Imtiaz (ORCID: https://orcid.org/0000-0002-3056-6756)
- Muhammad Adnan
- Enci Wang (ORCID: https://orcid.org/0000-0002-3646-5729)
Institutions
- Yangzhou University (CN)
Publication Details
- Journal
- IEEE Access
- Published
- 2025-01-01
- DOI
- https://doi.org/10.1109/access.2025.3574310
- Citations
- 3
- Primary Topic
- Brain Tumor Detection and Classification
- Type
- article
- Field-Weighted Citation Impact
- 2.53