Multi-Period Attention for Automatic ECG Classification
Konferenz: ICMLCA 2021 - 2nd International Conference on Machine Learning and Computer Application
17.12.2021 - 19.12.2021 in Shenyang, China
Tagungsband: ICMLCA 2021
Seiten: 4Sprache: EnglischTyp: PDF
Persönliche VDE-Mitglieder erhalten auf diesen Artikel 10% Rabatt
Autoren:
Zhang, Xulu; Zhou, Kai (Department of Computer Science, Sichuan University, Chengdu, Sichuan, China)
Inhalt:
Electrocardiogram (ECG) is of great importance to the diagnoses of heart diseases, and has been increasingly studied in the field of deep learning. Many sophisticated deep learning methods have been transferred to the classification of ECG due to the success in image classification. However, these methods have not perfectly adapted to the periodic characteristic of ECG which is an essential aspect. In this paper, we propose to address this issue with an efficient attention block called Multi-Period Attention, in which periodic masks have been generated based on periodic function. Besides, the masks have been mutually independent and adaptively adjusted according to different channels to avoid overfitting. We have applied the blocks into ResNet18 and conducted experiments on clinical 12-lead ECG dataset PTB-XL. The results have showed that our method has comparative ability over the other models with much more parameters.