Integrating deep learning with ECG, heart rate variability and demographic data for improved detection of atrial fibrillation.

Abstract

Background: Atrial fibrillation (AF) is a common but often undiagnosed condition, increasing the risk of stroke and heart failure. Early detection is crucial, yet traditional methods struggle with AF's transient nature. This study investigates how augmenting ECG data with heart rate variability (HRV) and demographic data (age and sex) can improve AF detection.

Methods: We analysed 35 634 12-lead ECG recordings from three public databases (China Physiological Signal Challenge-Extra, PTB-XL and Georgia), each with physician-validated AF labels. A range of convolutional neural network models, including AlexNet, VGG-16, ResNet and transformers, were tested for AF prediction, enriched with HRV and demographic data to explore the effectiveness of the multimodal approach. Each data modality (ECG, HRV and demographic) was assessed for its contribution to model performance using fivefold cross-validation. Performance improvements were evaluated across key metrics, and saliency maps were generated to provide further insights into model behaviour and identify critical features in AF detection.

Results: Integrating HRV and demographic data with ECG substantially improved performance. AlexNet and VGG-16 outperformed more complex models, achieving AUROC of 0.9617 (95% CI 0.95 to 0.97) and 0.9668 (95% CI 0.96 to 0.97), respectively. Adding HRV data showed the most significant improvement in sensitivity, with AlexNet increasing from 0.9117 to 0.9225 and VGG-16 from 0.9216 to 0.9225. Combining both HRV and demographic data led to further improvements, with AlexNet achieving a sensitivity of 0.9225 (up from 0.9192 with HRV) and VGG-16 reaching 0.9113 (up from 0.9097 with HRV). The combination of HRV and demographic data resulted in the highest gains in sensitivity and area under the receiver operating characteristic curve. Saliency maps confirmed the models identified key AF features, such as the absence of the P-wave, validating the multimodal approach.

Conclusions: AlexNet and VGG-16 excelled in AF detection, with HRV data improving sensitivity, and demographic data providing additional benefits. These results highlight the potential of multimodal approaches, pending further clinical validation.