首页 | 本学科首页   官方微博 | 高级检索  
     

基于并行架构和时空注意力机制的心电分类方法
引用本文:彭向东,潘从成,柯泽浚,朱华强,周肖.基于并行架构和时空注意力机制的心电分类方法[J].浙江大学学报(自然科学版 ),2022,56(10):1912-1923.
作者姓名:彭向东  潘从成  柯泽浚  朱华强  周肖
作者单位:江西财经大学 软件与物联网工程学院,江西 南昌 330032
基金项目:江西省自然科学基金资助项目(20192BAB207003);江西省教育厅科学技术研究资助项目(GJJ180263)
摘    要:为了有效提取心电信号 (ECG) 的时空特征和提高分类准确性,提出基于深度学习的并行架构心电分类模型. 该模型采用基于GCA Block和GTSA Block模块实现多路特征融合的时空注意力机制. 使用双向长短时记忆网络和卷积神经网络作为基特征提取器,分别捕捉心电信号序列数据的前后依赖关系和不同尺度上的局部相关特征,实现对5种不同类型的心电信号的自动分类. 在MIT-BIH数据集上验证的结果表明,该方法对5种不同心电信号的总体分类准确率、特异性、敏感度、精确度和Macro-F1分别为99.50%、99.61%、96.20%、98.02%和97.08%. 相较于其他心电分类模型,该模型不仅能够有效地缩短网络模型深度,防止模型过拟合,而且能够更准确地提取心电信号的时空特征,获得更好的分类性能.

关 键 词:心电分类  数据不平衡  深度学习  并行架构  时空注意力机制  

Classification method for electrocardiograph signals based on parallel architecture model and spatial-temporal attention mechanism
Xiang-dong PENG,Cong-cheng PAN,Ze-jun KE,Hua-qiang ZHU,Xiao ZHOU.Classification method for electrocardiograph signals based on parallel architecture model and spatial-temporal attention mechanism[J].Journal of Zhejiang University(Engineering Science),2022,56(10):1912-1923.
Authors:Xiang-dong PENG  Cong-cheng PAN  Ze-jun KE  Hua-qiang ZHU  Xiao ZHOU
Abstract:A parallel architecture electrocardiograph (ECG) classification model based on deep learning was proposed in order to effectively extract the spatiotemporal characteristics of ECG signals and improve the classification accuracy. A spatiotemporal attention mechanism based on gate channel attention block (GCA block) and gate time step attention (GTSA block) module was adopted in order to achieve multi-channel feature fusion. The bidirectional long-short time memory network and the convolutional neural network were used as the base feature extractor. The before-after dependence of the ECG signal sequence data and the local correlation features at different scales were captured respectively, and the automatic classification of five different types of ECG signals was realized. Results verified on the MIT-BIH dataset showed that the accuracy, specificity, sensitivity, accuracy and Macro-F1 of the total classification of five different ECG signals by the method were 99.50%, 99.61%, 96.20%, 98.02% and 97.08%, respectively. The model can not only effectively shorten the depth of the network model and prevent the model from overfitting, but also more accurately extract the spatiotemporal characteristics of the ECG signal and obtain better classification performance compared with other ECG classification models.
Keywords:electrocardiograph classification  data imbalanced  deep learning  parallel architecture  spatiotemporal attention mechanism  
点击此处可从《浙江大学学报(自然科学版 )》浏览原始摘要信息
点击此处可从《浙江大学学报(自然科学版 )》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号