计算机科学 ›› 2021, Vol. 48 ›› Issue (10): 121-126.doi: 10.11896/jsjkx.200700103
杨春德1,2, 贾竹1, 李欣蔚2
YANG Chun-de1,2, JIA Zhu1, LI Xin-wei2
摘要: 探索高效、快速、精准的心电信号识别分类算法是心电诊断的难点。基于心电片段的识别分类更贴合临床应用。基于此,文中将改进的深度卷积生成对抗网络(Deep Convolutional Generative Adversarial Networks,DCGAN)用于数据扩充,将优化的一维U-Net++用于心律不齐的片段信号识别。连续截取MIT-BIH数据库中1 200个采样点的心电片段作为实验数据集,以每条片段记录中心拍标签次数出现最多的类型作为整段记录的标签。再将优化的一维U-Net++作为DCGAN结构的生成器实现部分数据扩充,以解决数据不平衡的问题。在原始心电信号未经过任何预处理以及生成的扩充数据用于完成小波阈值去噪的情况下,优化的一维U-Net++模型对于正常、室性早搏、左束支阻滞、右束支阻滞4类不同的心电类型训练集的准确率能够达到98.10%,且对于测试集的精准率、召回率和F1值等指标均有较好的结果。在相同实验数据集下,优化的一维U-Net++模型比U-Net模型的准确率提高了1.05%;在相同实验参数的条件下,与欠采样数据对比,经DCGAN数据扩充后的数据集实验模型的准确率提高了0.85%。
中图分类号:
[1]MA J W,LIU S P.Overview of ECG Signal Recognition andClassification Algorithms[J].Journal of Chongqing University of Technology (Natural Science),2018,32(12):122-128. [2]JIN L P,DONG J.Study on Classification Algorithm of Clinical Electrocardiogram based on Integrated Learning[J].Journal of Biomedical Engineering,2016,33(5):825-833. [3]MA R L,LIU X,ZHANG Y,et al.Deep learning based on ECG Signal anomaly Recognition Method[J].Transducer andMicro-system Technologies,2020,39(1):29-32. [4]CHEN M,WANG R F.Arrhythmia Classification Based onTwo-Dimensional Image and Tranfer Convolutional Neural Network[J].Computer Engineering,2020,46(10):315-320. [5]FENG Y R,CHEN W,CAI G Y.Biometric Extraction and Re-cognition based on ECG Signals[J].Computer & Digital Engineering,2016,46(6):1099-1103. [6]VENKATESAN C,KARTHIGAIKUMAR P,VARATHARAJAN R.A novel LMS Algorithm for ECG Signal Preprocessing and KNN Classifier-based Abnormality Detection[J].Multi-media Tools and Applications,2018,77(8):10365-10374. [7]ZHANG K,LI X,XIE X J,et al.Study on Arrhythmia Detection Algorithm based on Deep Learning[J].Chinese Medical Equipment Journal,2008,39(12):6-9,31. [8]LI D,ZHANG H X,LIU Z Q,et al.Recognition of Arrhythmia in ECG based on Deep Residual Convolution Neural Network[J].Journal of Biomedical Engineering,2019,36(2):189-198. [9]HESAR H D,MOHEBBI M.An Adaptive Kalman Filter Bank for ECG Denoising[J].IEEE Journal of Biomedical and Health Informatics,2021,25(1):13-21. [10]GAO N H,WANG H,FENG X H.Classification Method ofElectrocardiogram Signals Based on Dynamic Fuzzy Decision Tree[J].Computer Engineering,2020,46(1):80-86. [11]RONNEBERGER O,FISCHER P,BROX T.U-Net:ConvolutionalNetworks for Biomedical Image Segmentation[C]//Internatio-nal Conference on Medical Image Computing & Computer-assisted Intervention.Cham:Springer,2015:234-241. [12]LONG J,SHELHAMER E,DARRELL T.Fully Convolutional Networks for Semantic Segmentation[C]//IEEE Conference on Computer Vision and Pattern Recognition.2015:3431-3440. [13]ZHOU Z,SIDDIQUEE M M R,TAJBAKHSH N,et al. U-Net++:A Nested U-Net Architecture for Medical Image Segmentation[C]//Deep Learning in Medical Image Analysis and Multimodal Learning for Clinical Decision Support.Cham:Springer,2018:3-11. [14]HUANG G,LIU Z,MAATEN L V D,et al.Densely Connected Convolutional Networks[C]//IEEE Conference on Computer Vision and Pattern Recognition.2017:2261-2269. [15]LIN M,CHEN Q,YAN S.Network in network[J].arXiv:1312.4400,2013. [16]KRIZHEVSKY A,SUTSKEVER I,HINTON G E.Image-net Classification with Deep Convolutional Neural Networks[C]//Advances in Neural Information Processing Systems.2012:1097-1105. [17]LIANG J J,WEI J J,JIANG Z F.Overview of GAN Generation Adversarial Network[J].Journal of Frontiers of Computer Scien-ce and Technology,2020,14(1):1-17. [18]GOODFELLOW I,POUGET-ABADIE J,MIRZA M,et al.Gene-rative Adversarial Nets[C]//Advances in Neural Information Processing Systems.2014:2672-2680. [19]CHEN G D,ZENG Y L,LI Z.Adaptive Double Threshold ECG Signal Detection Algorithm Research[J].Journal of Jinan University (Natural Science & Medicine Edition),2008,39(3):262-268. [20]LUO B.Research on the solution of interference signal removal in electrocardiograph [J].Quality & Market,2020(14):88-90. [21]LI F,XIE S H.Abnormal Diagnosis of Mobile ECG based on Unsupervised Learning[J].Computer Science,2017,44(S2):68-71,109. [22]FAN M,WANG X F,MENG X F.Research on Wearable ECG Adaptive Classification Algorithm [J].Computer Science,2019,46(12):292-297. [23]GAO H J,QIU T S,CHOU Y T,et al.Fundus Image Vascular Segmentation based on Improved U-Net Network[J].Chinese Journal of Biomedical Engineering,2019,38(1):1-8. [24]OH S L,NG E Y K,TAN R S,et al.Automated Beat-wiseArrhythmia Diagnosis using Modified U-Net on Extended Electrocardiographic Recordings with Heterogeneous Arrhythmia Types[J].Computers in Biology and Medicine,2019,105:92-101. |
[1] | 张佳, 董守斌. 基于评论方面级用户偏好迁移的跨领域推荐算法 Cross-domain Recommendation Based on Review Aspect-level User Preference Transfer 计算机科学, 2022, 49(9): 41-47. https://doi.org/10.11896/jsjkx.220200131 |
[2] | 孙奇, 吉根林, 张杰. 基于非局部注意力生成对抗网络的视频异常事件检测方法 Non-local Attention Based Generative Adversarial Network for Video Abnormal Event Detection 计算机科学, 2022, 49(8): 172-177. https://doi.org/10.11896/jsjkx.210600061 |
[3] | 戴朝霞, 李锦欣, 张向东, 徐旭, 梅林, 张亮. 基于DNGAN的磁共振图像超分辨率重建算法 Super-resolution Reconstruction of MRI Based on DNGAN 计算机科学, 2022, 49(7): 113-119. https://doi.org/10.11896/jsjkx.210600105 |
[4] | 尹文兵, 高戈, 曾邦, 王霄, 陈怡. 基于时频域生成对抗网络的语音增强算法 Speech Enhancement Based on Time-Frequency Domain GAN 计算机科学, 2022, 49(6): 187-192. https://doi.org/10.11896/jsjkx.210500114 |
[5] | 徐辉, 康金梦, 张加万. 基于特征感知的数字壁画复原方法 Digital Mural Inpainting Method Based on Feature Perception 计算机科学, 2022, 49(6): 217-223. https://doi.org/10.11896/jsjkx.210500105 |
[6] | 赵人行, 徐频捷, 刘瑶. 基于深度卷积残差网络的心电单导联房颤检测方法 ECG-based Atrial Fibrillation Detection Based on Deep Convolutional Residual Neural Network 计算机科学, 2022, 49(5): 186-193. https://doi.org/10.11896/jsjkx.220200002 |
[7] | 高志宇, 王天荆, 汪悦, 沈航, 白光伟. 基于生成对抗网络的5G网络流量预测方法 Traffic Prediction Method for 5G Network Based on Generative Adversarial Network 计算机科学, 2022, 49(4): 321-328. https://doi.org/10.11896/jsjkx.210300240 |
[8] | 黎思泉, 万永菁, 蒋翠玲. 基于生成对抗网络去影像的多基频估计算法 Multiple Fundamental Frequency Estimation Algorithm Based on Generative Adversarial Networks for Image Removal 计算机科学, 2022, 49(3): 179-184. https://doi.org/10.11896/jsjkx.201200081 |
[9] | 李建, 郭延明, 于天元, 武与伦, 王翔汉, 老松杨. 基于生成对抗网络的多目标类别对抗样本生成算法 Multi-target Category Adversarial Example Generating Algorithm Based on GAN 计算机科学, 2022, 49(2): 83-91. https://doi.org/10.11896/jsjkx.210800130 |
[10] | 谈馨悦, 何小海, 王正勇, 罗晓东, 卿粼波. 基于Transformer交叉注意力的文本生成图像技术 Text-to-Image Generation Technology Based on Transformer Cross Attention 计算机科学, 2022, 49(2): 107-115. https://doi.org/10.11896/jsjkx.210600085 |
[11] | 陈贵强, 何军. 自然场景下遥感图像超分辨率重建算法研究 Study on Super-resolution Reconstruction Algorithm of Remote Sensing Images in Natural Scene 计算机科学, 2022, 49(2): 116-122. https://doi.org/10.11896/jsjkx.210700095 |
[12] | 石达, 芦天亮, 杜彦辉, 张建岭, 暴雨轩. 基于改进CycleGAN的人脸性别伪造图像生成模型 Generation Model of Gender-forged Face Image Based on Improved CycleGAN 计算机科学, 2022, 49(2): 31-39. https://doi.org/10.11896/jsjkx.210600012 |
[13] | 唐雨潇, 王斌君. 基于深度生成模型的人脸编辑研究进展 Research Progress of Face Editing Based on Deep Generative Model 计算机科学, 2022, 49(2): 51-61. https://doi.org/10.11896/jsjkx.210400108 |
[14] | 张玮琪, 汤轶丰, 李林燕, 胡伏原. 基于场景图的段落生成序列图像方法 Image Stream From Paragraph Method Based on Scene Graph 计算机科学, 2022, 49(1): 233-240. https://doi.org/10.11896/jsjkx.201100207 |
[15] | 蒋宗礼, 樊珂, 张津丽. 基于生成对抗网络和元路径的异质网络表示学习 Generative Adversarial Network and Meta-path Based Heterogeneous Network Representation Learning 计算机科学, 2022, 49(1): 133-139. https://doi.org/10.11896/jsjkx.201000179 |
|