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

Y Wang, G Yang, S Li, Y Li, L He, D Liu - Biomedical Signal Processing and …, 2023 - Elsevier
Y Wang, G Yang, S Li, Y Li, L He, D Liu
Biomedical Signal Processing and Control, 2023Elsevier
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 …
Abstract
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%.
Elsevier
以上显示的是最相近的搜索结果。 查看全部搜索结果