南方医科大学学报2025，Vol.45Issue(3)：650-660,11.DOI:10.12122/j.issn.1673-4254.2025.03.23
基于深度可分离卷积与注意力机制的单导联心房颤动轻量级分类网络
A lightweight classification network for single-lead atrial fibrillation based on depthwise separable convolution and attention mechanism
摘要
Abstract
Objective To design a deep learning model that balances model complexity and performance to enable its integration into wearable ECG monitoring devices for automated diagnosis of atrial fibrillation.Methods This study was performed based on data from 84 patients with atrial fibrillation,25 patients with atrial fibrillation,and 18 subjects without obvious arrhythmia collected from the publicly available datasets LTAFDB,AFDB,and NSRDB,respectively.A lightweight attention network based on depthwise separable convolution and fusion of channel-spatial information,namely DSC-AttNet,was proposed.Depthwise separable convolution was introduced to replace standard convolution and reduce model parameters and computational complexity to realize high efficiency and light weight of the model.The multilayer hybrid attention mechanism was embedded to compute the attentional weights of the channels and spatial information at different scales to improve the feature expression ability of the model.Ten-fold cross-validation was performed on LTAFDB,and external independent testing was conducted on AFDB and NSRDB datasets.Results DSC-AttNet achieved a ten-fold average accuracy of 97.33%and a precision of 97.30%on the test set,both of which outperformed the other 4 comparison models as well as the 3 classical models.The accuracy of the model on the external test set reached 92.78%,better than those of the 3 classical models.The number of parameters of DSC-AttNet was 1.01M,and the computational volume was 27.19G,both smaller than the 3 classical models.Conclusion This proposed method has a smaller complexity,achieves better classification performance,and has a better generalization ability for atrial fibrillation classification.关键词
心电图/心房颤动/卷积块注意模块/MobileNet/轻量级卷积神经网络Key words
electrocardiogram/atrial fibrillation/convolutional block attention module/MobileNet/lightweight convolutional neural network引用本文复制引用
洪永,张鑫,林铭俊,吴秋岑,陈超敏..基于深度可分离卷积与注意力机制的单导联心房颤动轻量级分类网络[J].南方医科大学学报,2025,45(3):650-660,11.基金项目
国家重点研发计划(2023YFC2414500,2023YFC2414502) （2023YFC2414500,2023YFC2414502）
