计算机工程 ›› 2020, Vol. 46 ›› Issue (10): 315-320.doi: 10.19678/j.issn.1000-3428.0055932

• 开发研究与工程应用 • 上一篇    

基于二维图像与迁移卷积神经网络的心律失常分类

陈敏, 王娆芬   

  1. 上海工程技术大学 电子电气工程学院, 上海 201620
  • 收稿日期:2019-09-05 修回日期:2019-10-09 发布日期:2019-11-04
  • 作者简介:陈敏(1994-),女,硕士研究生,主研方向为生理信号处理、模式识别;王娆芬(通信作者),副教授、博士。
  • 基金项目:
    国家自然科学基金(71701124);国家自然科学基金青年科学基金(61803255);上海市自然科学基金(18ZR1416700)。

Arrhythmia Classification Based on Two-Dimensional Image and Tranfer Convolutional Neural Network

CHEN Min, WANG Raofen   

  1. School of Electronic and Electrical Engineering, Shanghai University of Engineering Science, Shanghai 201620, China
  • Received:2019-09-05 Revised:2019-10-09 Published:2019-11-04

摘要: 心律失常的自动分类对心血管疾病的诊断和预防具有重要意义。传统分类方法需要对心电信号进行人工特征提取,这对分类准确度有很大的影响。针对该问题,提出一种基于二维图像与迁移卷积神经网络(TCNN)的分类方法。通过对心电信号进行格拉姆角场变换将其转换为二维图像,在保证心电图像完整性的同时,保留原始信号的时间依赖性。在此基础上,结合迁移学习的思想,设计结构简单且参数量较少的TCNN模型对心电图像进行分类。实验结果表明,该方法网络训练用时较少,并且分类总准确率达到99.82%,可实现对心律失常的有效分类。

关键词: 心电信号, 格拉姆角场, 二维图像, 迁移学习, 迁移卷积神经网络

Abstract: The automatic classification of arrhythmia is very important for the diagnosis and prevention of cardiovascular diseases.Traditional classification methods require artificial feature extraction of ECG signals,which has a great impact on the accuracy of classification.To solve this problem,a classification method based on two dimensional image and Transfer Convolutional Neural Network(TCNN) is proposed.The ECG signal is transformed into a two-dimensional image by means of Gramian Angular Field.The integrity of the ECG image is guaranteed while the time dependence of the original signal is retained.On this basis,combined with the idea of transfer learning,a TCNN model with simple structure and fewer parameters is designed to classify the ECG images.The experimental results show that the network training time of this method is less,and the classification accuracy reaches 99.82%,which can realize the effective classification of arrhythmia.

Key words: ECG signal, Gramian Angular Field(GAF), two-dimensional image, transfer learning, Transfer Convolutional Neural Network(TCNN)

中图分类号: