J Shanghai Jiaotong Univ Sci ›› 2022, Vol. 27 ›› Issue (1): 55-69.doi: 10.1007/s12204-021-2371-8

• Robotics & AI in Interdisciplinary Medicine and Engineering • Previous Articles     Next Articles

Multi-Lead ECG Classification via an Information-Based Attention Convolutional Neural Network

TUNG Hao1 (董昊), ZHENG Chao2 (郑超), MAO Xinsheng2 (毛新生), QIAN Dahong3∗ (钱大宏)   

  1. (1. School of Biomedical Engineering, Shanghai Jiao Tong University, Shanghai 200240, China; 2. Shukun (Beijing) Network Technology Co., Ltd., Beijing 100190, China; 3. Institute of Medical Robotics, Shanghai Jiao Tong University, Shanghai 200240, China)
  • Received:2020-03-30 Online:2022-01-28 Published:2022-01-14

Abstract: A novel structure based on channel-wise attention mechanism is presented in this paper. With the proposed structure embedded, an efficient classification model that accepts multi-lead electrocardiogram (ECG) as input is constructed. One-dimensional convolutional neural networks (CNNs) have proven to be effective in pervasive classification tasks, enabling the automatic extraction of features while classifying targets. We implement the residual connection and design a structure which can learn the weights from the information contained in different channels in the input feature map during the training process. An indicator named mean square deviation is introduced to monitor the performance of a particular model segment in the classification task on the two out of five ECG classes. The data in the MIT-BIH arrhythmia database is used and a series of control experiments is conducted. Utilizing both leads of the ECG signals as input to the neural network classifier can achieve better classification results than those from using single channel inputs in different application scenarios. Models embedded with the channel-wise attention structure always achieve better scores on sensitivity and precision than the plain Resnet models. The proposed model exceeds most of the state-of-the-art models in ventricular ectopic beats (VEB) classification performance and achieves competitive scores for supraventricular ectopic beats (SVEB). Adopting more lead ECG signals as input can increase the dimensions of the input feature maps, helping to improve both the performance and generalization of the network model. Due to its end-to-end characteristics, and the extensible intrinsic for multi-lead heart diseases diagnosing, the proposed model can be used for the realtime ECG tracking of ECG waveforms for Holter or wearable devices.

Key words: attention model, convolutional neural network (CNN), ECG classification, arrhythmia

CLC Number: