Pruned Lightweight Neural Networks for Arrhythmia Classification with Clinical 12-Lead ECGs

Yunqing Liu,Jinlei Liu,Yuanyuan Tian,Yanrui Jin,Zhiyuan Li,Liqun Zhao,Chengliang Liu
DOI: https://doi.org/10.1016/j.asoc.2024.111340
IF: 8.7
2024-01-01
Applied Soft Computing
Abstract:Real-time electrocardiogram (ECG) monitoring through portable or wearable devices is critical for detecting lethal arrhythmias. Despite the accuracy of 12-lead ECGs in clinical image analysis, their integration into portable devices poses challenges. This paper introduces a novel method for nonmalignant arrhythmia classification, optimized for wearable and portable devices. We utilize ECG records from Shanghai First People’s Hospital, proposing a lightweight neural network strategy involving benchmark network selection, model pruning, and learning rate decay-based finetuning. The proposed Random Horizontal Flip (RHF)-based classification method demonstrated superior performance, achieving a 94.8 % accuracy on a 12-lead clinical ECG test dataset. Utilizing the modified pruning method, the classification accuracy for five-class ECGs improved by 7.2 % over the benchmark network. The model size was reduced by 51.26 %, with parameters and FLOPs decreasing by 47.6 % and 49.1 %, respectively, compared to the benchmark, all under identical hardware conditions. Further experiments were conducted on the Hold-out Test Set (HTS), designed to include ECGs that present slight variations to the original conditions, yielding a slightly lower accuracy of 92.4 %, reflecting the dataset's complexity and clinical variability. Moreover, benchmarking tests using the Massachusetts Institute of Technology-Beth Israel Hospital (MIT-BIH) arrhythmia database validated the method’s effectiveness, achieving a 99.24 % accuracy and maintaining a lightweight model of 4366KB. Comparative analysis with existing methods confirmed the proposed method’s superiority in accuracy and real-world applicability. This research presents a significant advancement in ECG analysis, offering a viable solution for efficient arrhythmia monitoring in portable healthcare devices.
What problem does this paper attempt to address?