CLINet: A novel deep learning network for ECG signal classification

Abstract

Machine learning is poised to revolutionize medicine with algorithms that spot cardiac arrhythmia. An automated diagnostic approach can boost the efficacy of diagnosing life-threatening arrhythmia disorders in routine medical procedures. In this paper, we propose a deep learning network CLINet for ECG signal classification. Our network uses convolution, LSTM and involution layers to bring their unique advantages together. For both convolution and involution layers, we use multiple, large size kernels for multi-scale representation learning. CLINet does not require complicated pre-processing and can handle electrocardiograms of any length. Our network achieves 99.90% accuracy on the ICCAD dataset and 99.94% accuracy on the MIT-BIH dataset. With only 297 K parameters, our model can be easily embedded in smart wearable devices. The source code of CLINet is available at https://github.com/CandleLabAI/CLINet-ECG-Classification-2024.

Keywords: ECG signal classification; LSTM; Machine learning; convolution; involution.