DWT-CNNTRN: a Convolutional Transformer for ECG Classification with Discrete Wavelet Transform

Annu Int Conf IEEE Eng Med Biol Soc. 2023 Jul:2023:1-6. doi: 10.1109/EMBC40787.2023.10340561.

Abstract

Cardiovascular diseases are the leading cause of death worldwide. The diagnoses of cardiovascular diseases are usually carried out by cardiologists utilizing Electrocardiograms (ECGs). To assist these physicians in making an accurate diagnosis, there is a growing need for reliable and automatic ECG classifiers.In this study, a new method is proposed to classify 12-lead ECG recordings. The proposed model is composed of four components: the CNN(Convolutional Neural Network) module, the transformer module, the global hybrid pooling layer, and a classification layer. To improve the classification performance, the model takes the discrete wavelet transform of ECG signals as the model inputs and utilizes a hybrid pooling layer to condense the most important features over each period.The proposed model is evaluated using the test set of the China Physiological Signal Challenge 2018 dataset with 12-lead ECGs. It performs with an average accuracy of 0.86 and an average F1-scores of 0.83. The scores are particularly good for the block conditions (LBBB, RBBB, I-AVB). The main advantage of the proposed model is that, it obtains good results with a significantly smaller number of parameters compared to other individual and ensemble models.Clinical relevance- This work establishes a new ECG classifier model with high performance and low model size. It can make automatic ECG analysis more accessible, efficient, and accurate, especially in remote or underserved areas.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Cardiovascular Diseases*
  • Electrocardiography / methods
  • Humans
  • Neural Networks, Computer
  • Signal Processing, Computer-Assisted
  • Wavelet Analysis*