Arrhythmia classification algorithm based on multi-head self-attention mechanism

•ECG signal preprocessing method based on wavelet transform to reduce noise.•Linear projection layer is designed to acquire semantic features of ECG signal.•Novel position encoding is proposed to obtain time and voltage series information.•Novel arrhythmia classification algorithm based on multihead...

Celý popis

Uloženo v:
Podrobná bibliografie
Vydáno v:Biomedical signal processing and control Ročník 79; s. 104206
Hlavní autoři: Wang, Yue, Yang, Guanci, Li, Shaobo, Li, Yang, He, Ling, Liu, Dan
Médium: Journal Article
Jazyk:angličtina
Vydáno: Elsevier Ltd 01.01.2023
Témata:
Arrhythmia classification
Attention mechanism
Electrocardiogram (ECG)
Feature extraction
Attention mechanism
Arrhythmia classification
Feature extraction
Electrocardiogram (ECG)
ISSN:1746-8094, 1746-8108
On-line přístup:Získat plný text
Tagy: Přidat tag
Žádné tagy, Buďte první, kdo vytvoří štítek k tomuto záznamu!
Popis
Shrnutí:•ECG signal preprocessing method based on wavelet transform to reduce noise.•Linear projection layer is designed to acquire semantic features of ECG signal.•Novel position encoding is proposed to obtain time and voltage series information.•Novel arrhythmia classification algorithm based on multihead self-attention mechanism.•The proposed ACA-MA outperforms other state-of-the-art methods. Cardiovascular disease is a major illness that causes human death, especially in the elderly. Timely and accurate diagnosis of arrhythmia types is the key to early prevention and diagnosis of cardiovascular diseases. This paper proposed an arrhythmia classification algorithm based on multi-head self-attention mechanism (ACA-MA). First, an ECG signal preprocessing algorithm based on wavelet transform is put forward and implemented using db6 wavelet transform to focus on improving the data quality of ECG signals and reduce the noise of ECG signals. Second, a linear projection layer for acquiring semantic features of ECG signals is designed using the matching relationship between ECG tag and segmented ECG signals. Third, a position encoding-based spatiotemporal characterization method of ECG signal sequences is designed to integrate time series information into a matrix operation. Fourth, a multi-head self-attentive mechanism capable of capturing global contextual information is proposed to extract relationships and semantic features between ECG segments and achieve semantic association and information stitching of nonadjacent ECG signals. Finally, experimental results on the arrhythmia dataset MIT/BIH show that ACA-MA outperforms other state-of-the-art methods with an overall classification accuracy of 99.4%, a specific rate of 99.41%, and a sensitivity of 97.36%.
ISSN:1746-8094
1746-8108
DOI:10.1016/j.bspc.2022.104206