CSD-AFNet: Computationally Efficient Atrial Fibrillation Classification from ECGs using 2D Causal Strided Dilated Convolutions

Published: 19 Aug 2025, Last Modified: 12 Oct 2025BHI 2025EveryoneRevisionsBibTeXCC BY 4.0
Confirmation: I have read and agree with the IEEE BHI 2025 conference submission's policy on behalf of myself and my co-authors.
Keywords: Electrocardiography, Atrial Fibrillation, Deep Learning, Computational Efficiency, Compact Model, Causal Convolution, Strided Dilated Convolution, Arrhythmia Detection
TL;DR: CSD-AFNet is a lightweight deep learning model for atrial fibrillation classification from ECGs that matches state-of-the-art performance while drastically reducing parameters and FLOPs, enabling efficient deployment on resource-constrained devices.
Abstract: Automated analysis of electrocardiogram (ECG) signals using deep learning (DL) methods has shown substantial promise in atrial fibrillation (AFib) classification, particularly for detecting subtle indicators during normal sinus rhythm and for predicting new-onset AFib. However, many existing state-of-the-art models exhibit high computational demands, characterised by large parameter and floating-point operations (FLOPs) counts. This presents a high barrier to entry for training in budget-limited institutes and hinders the models’ deployment on medical edge devices. This paper introduces CSD-AFNet, a computationally efficient DL model specifically designed for AFib-related ECG classification tasks. CSD-AFNet achieves substantial reductions in both parameter and FLOPs counts by replacing expensive temporal convolutions with novel Feature-Preserving Pooled Convolutions (FPP-Convs). FPP-Convs enable the combination of striding and dilation without input feature loss, preserving temporal coverage while reducing the computational cost. The model further incorporates two-dimensional causal padding to prevent temporal leakage in downstream representations. Evaluation on the public CODE-15% and PTB-XL datasets demonstrates that CSD-AFNet matches the classification performance of leading benchmark models while reducing parameter count by a factor of 71 and FLOPs by a factor of 122 compared to the ResNet-10 inspired baseline. These findings support the suitability of CSD-AFNet for practical clinical scenarios, enabling training under resource constraints and efficient inference on medical edge devices, thereby facilitating scalable and cost-effective ECG-based AFib screening and monitoring.
Track: 4. Clinical Informatics
Registration Id: S5NNZ4WX5HP
Submission Number: 88
Loading