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

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.