Robust and Generalizable Atrial Fibrillation Detection from ECG Using Time-Frequency Fusion and Supervised Contrastive Learning

Abstract

Atrial fibrillation (AF) is a common cardiac arrhythmia that significantly increases the risk of stroke and heart failure, necessitating reliable and generalizable detection methods from electrocardiogram (ECG) recordings. Although deep learning has advanced automated AF diagnosis, existing approaches often struggle to exploit complementary time frequency information effectively, limiting both robustness under intra-dataset and generalization across diverse clinical datasets. To address these challenges, we propose a crossmodal deep learning framework comprising two key components: a Bidirectional Gating Module (BGM) and a Cross modal Supervised Contrastive Learning (CSCL) strategy. The BGM facilitates dynamic, reciprocal refinement between time and frequency domain features, enhancing model robustness to signal variations within a dataset. Meanwhile, CSCL explicitly structures the joint embedding space by pulling together label consistent samples and pushing apart different ones, thereby improving interclass separability and enabling strong cross dataset generalization. We evaluate our method using five fold crossvalidation on the AFDB and CPSC2021 datasets. Furthermore, to assess cross dataset generalization, we conduct bidirectional cross dataset experiments across AFDB, CPSC2021, LTAF, and SHDBAF by training on one dataset and testing on another. Results show consistent improvements over state of the art methods across multiple metrics, demonstrating that our approach achieves both high intra dataset robustness and excellent crossdataset generalization. We further demonstrate that our method achieves high computational efficiency and anti interference capability, making it suitable for edge deployment.