基于多头注意力机制的房颤检测方法

近年来,随着人工智能的发展,深度学习模型已在ECG数据分析(尤其是房颤的检测)中得到广泛应用.本文提出了一种基于多头注意力机制的算法来实现房颤的分类,并通过PhysioNet 2017年挑战赛的公开数据集对其进行训练和验证.该算法首先采用深度残差网络提取心电信号的局部特征,随后采用双向长短期记忆网络在此基础上提取全局特...

Atrial Fibrillation Detection Using Multi-Head Attention Mechanism

In recent years, driven by the progress in artificial intelligence, deep learning models have been widely applied to ECG data analysis (especially the detection of atrial fibrillation). This study proposes an algorithm based on the multi-head attention mechanism to classify atrial fibrillation, which is trained and validated through the public data set of the PhysioNet 2017 Challenge. Firstly, the local features of the ECG signal are extracted through the deep residual network. Then, the bidirectional long short-term memory network is built to extract the global features on this basis. Finally, the multi-head attention mechanism layer is used to extract the key features, and cascade modules greatly improve the performance of the overall model. The experimental results show that the proposed heads-8 model can achieve precision of 0.861, recall of 0.862, F1 score of 0.861, and accuracy of 0.860, which is better than the latest methods based on ECG signals for classifying atrial fibrillation.