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

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%.