An improved Bi-LSTM method based on heterogeneous features fusion and attention mechanism for ECG recognition

Abstract

Electrocardiogram (ECG) plays a critical role in early prevention and diagnosis of cardiovascular diseases. However, extracting powerful deep features from ECG signal for recognition is still a challenging problem today due to the variable abnormal rhythms and noise distribution. This work proposes a Bi-LSTM algorithm based on heterogeneous features fusion and attention mechanism (HFFAM + Bi-LSTM). Combining the empirical features and the features learned by the deep learning network, HFFAM + Bi-LSTM can comprehensively extract the temporal frequency information and spatial structure information of the ECG signal. Meanwhile, a novel attention mechanism based on improved DTW (AM-DTW) is designed to analyze and control the fusion process of features. The role of AM-DTW in HFFAM + Bi-LSTM is twofold, one is to measure the feature similarity between ECG signal sets with different labels using the improved DTW, and the other is to distinguish the features into isomorphic and heterogeneous features as well as adaptive weighting of the features. It is worth mentioning that overly similar isomorphic features are filtered out to further optimize the algorithm. Thus, HFFAM + Bi-LSTM has the advantage of strengthening the heterogeneous information in the feature subspace while accounting for the isomorphic features. The accuracy of HFFAM + Bi-LSTM reaches up to 98.1 % and 97.1 % on the simulated and real datasets, respectively. Compared to the all benchmark models, the classification accuracy of HFFAM + Bi-LSTM is 1.3 % higher than the best. The experiments also demonstrate that HFFAM + Bi-LSTM has better performance compared with existing methods, which provides a new scheme for automatic detection of ECG signal.

Keywords: Attention mechanism; Bi-LSTM; Electrocardiogram (ECG); Filter perceptron; Heterogeneous features.