A Novel Multi-Scale Convolutional Neural Network for Arrhythmia Classification on Reduced-Lead ECGs
Pan Xia,Zhengling He,Yusi Zhu,Zhongrui Bai,Xianya Yu,Yuqi Wang,Fanglin Geng,Lidong Du,Xianxiang Chen,Peng Wang,Zhen Fang
DOI: https://doi.org/10.22489/CinC.2021.098
2021-01-01
Abstract:The PhysioNet/Computing in Cardiology Challenge 2021 focused on exploring the utility of reduced-lead ECGs for arrhythmia classification. Our team, AIRCAS_MEL1, proposed a novel multi-scale convolutional neural network that can classify 30 scored arrhythmias from 12- lead, 6-lead, 4-lead, 3-lead, and 2-lead ECGs. The proposed network was achieved by multiple branch networks to effectively extract the pathological information from the "detail" scale to the "approximation" scale via a series of 2-D convolution kernels (filters). The backbone of each branch network was a carefully designed 2-D convolutional neural network (CNN) with residual connection and attention mechanism, and it can adapt to multi-lead ECGs as input. The first 10 seconds of records from corresponding leads were extracted and preprocessed as inputs for end-to-end training, and the prediction probabilities of 30 categories were outputs. Finally, the proposed algorithms were evaluated on the hidden test set, and our classifiers received scores of 0.38, 0.33, 0.37, 0.43, and 0.38 (ranked 21th, 23th, 20th, 16th, and 20th out of 39 teams) for the 12-lead, 6-lead, 4-lead, 3-lead, and 2-lead versions of the hidden test set with the Challenge evaluation metric.