计算机科学 ›› 2021, Vol. 48 ›› Issue (10): 121-126.doi: 10.11896/jsjkx.200700103

• 人工智能* 上一篇    下一篇

基于U-Net++的心电信号识别分类研究

杨春德1,2, 贾竹1, 李欣蔚2   

  1. 1 重庆邮电大学计算机科学与技术学院 重庆400065
    2 重庆邮电大学生物信息学院 重庆400065
  • 收稿日期:2020-07-16 修回日期:2020-10-20 出版日期:2021-10-15 发布日期:2021-10-18
  • 通讯作者: 李欣蔚(lixinwei@cqupt.edu.cn)
  • 作者简介:yangchunde64@126.com
  • 基金资助:
    重庆市教育委员会科学技术研究项目(KJQN201800622)

Study on ECG Signal Recognition and Classification Based on U-Net++

YANG Chun-de1,2, JIA Zhu1, LI Xin-wei2   

  1. 1 School of Computer Science and Technology,Chongqing University of Posts and Telecommunications,Chongqing 400065,China
    2 School of Biological Information,Chongqing University of Posts and Telecommunications,Chongqing 400065,China
  • Received:2020-07-16 Revised:2020-10-20 Online:2021-10-15 Published:2021-10-18
  • About author:YANG Chun-de,born in 1964,master,professor.His main research interests include digital image processing,information and computing theory.
    LI Xin-wei,born in 1990,Ph.D,lectu-rer.Her main research interests include biomedical image processing and so on.
  • Supported by:
    Science and Technology Research Project of Chongqing Education Commission(KJQN201800622).

摘要: 探索高效、快速、精准的心电信号识别分类算法是心电诊断的难点。基于心电片段的识别分类更贴合临床应用。基于此,文中将改进的深度卷积生成对抗网络(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%。

关键词: MIT-BIH, U-Net++, 生成对抗网络, 识别分类, 心电信号

Abstract: It is difficult to explore efficient,fast and accurate ECG signal recognition and classification algorithm.The classification of ECG fragments is more suitable for clinical application.Based on this,the improved generation countermeasure network (DCGAN) is used for data expansion,and the optimized one-dimensional U-Net++ is used for fragment signal recognition of arrhythmia.ECG fragments from 1 200 sampling points in MIT-BIH database are continuously intercepted as the experimental data set,and the type that appears the most times of beats in each fragment recording center is used as the label of the whole record.Then the DCGAN,which uses optimized one-dimensional U-Net++ as generator,is used to realize partial data expansion to solve the problem of data imbalance.Under the condition that the original ECG signals are not preprocessed and the generated extended data are used to complete the wavelet threshold denoising,the accuracy of the optimized one-dimensional U-Net++ model for normal,ventricular premature beat,left bundle branch block,right bundle branch block four kind of different type can reach 98.10% for the training sets.The precision ratio,recall ratio and F1 score of the test set have good results.Under the same experimental data set,the accuracy of U-Net++ model is 1.05% higher than that of U-Net model.Under the same experimental parameters,compared with under sampling data,the accuracy of the experimental model of the data set expanded by DCGAN is improved by 0.85%.

Key words: DCGAN, ECG, MIT-BIH, Recognition and classification, U-Net++

中图分类号: 

  • TN911.7
[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
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!