计算机与现代化 ›› 2023, Vol. 0 ›› Issue (01): 81-87.

• 算法设计与分析 • 上一篇    下一篇

基于对抗域适应的心电信号深度学习分类算法

  

  1. (1.浙江大学公共卫生学院,浙江 杭州 310058; 2.浙江大学睿医人工智能研究中心,浙江 杭州 310000)
  • 出版日期:2023-03-02 发布日期:2023-03-02
  • 作者简介:蒋思清(1997—),男,浙江杭州人,硕士研究生,研究方向:深度学习,数据库,E-mail: jsq522748392@126.com; 陈潇俊(1992—),男,江苏南京人,工程师,硕士,研究方向:云心电,数据库,E-mail: chenxiaojun@zju.edu.cn; 高豪俊(1996—),男,浙江温州人,硕士研究生,研究方向:NLP,深度学习,E-mail: joagh@outlook.com; 何佳晋(1997—),男,浙江义乌人,硕士研究生,研究方向:联邦学习,E-mail: hejiajin@zju.edu.cn; 通信作者:吴健(1975—),男,浙江杭州人, 教授,博士,研究方向:医学人工智能,E-mail: wujian2000@zju.edu.cn。
  • 基金资助:
    国家自然科学基金资助项目(62176231); 浙江省公益技术研究项目(LGF20F020013)

Deep Learning Classification Algorithm for Electrocardiogram Signal Based on Adversarial Domain Adaptation

  1. (1. School of Public Health, Zhejiang University, Hangzhou 310058, China;
    2. Real-doctor Artificial Intelligence Research Center, Zhejiang University, Hangzhou 310000, China)
  • Online:2023-03-02 Published:2023-03-02

摘要: 心血管疾病已成为威胁人类生命健康的主要疾病之一。心电图是临床上常见的诊断心律失常的重要方法并被广泛用于监测心脏病患者的健康状况。由于现有的医疗资源匮乏,使用人工智能的方法来分析和诊断从而克服这些局限性的需求愈加迫切,在临床中使用自动检测和分类方法,可以帮助医生对疾病做出准确、快速的诊断。本文对8种常见的心律失常类型进行分类,提出一种基于对抗域自适应的心电信号深度学习分类的方法,解决并改善训练样本标注不足和个体差异导致的数据分布差异现象等问题。该方法包括3个模块:多尺度特征提取A模块、域识别B模块和多分类器C模块。A模块由2组不同的并行卷积块组成,增加了特征提取的宽度。B模块由3个卷积块和1个全连接层组成,用于充分提取浅层特征。在C模块中,将时间特征和深度学习提取特征串联在全连接层上,增强特征多样性。实验结果表明,该方法在准确率、敏感性和阳性预测值上可达到98.8%、97.9%和98.1%,所提出的模型可以帮助医生在常规心电图中准确地检测不同类别的心律失常。

关键词: 心电信号分类, 深度学习, 多尺度, 时间特征, 对抗域自适应

Abstract:  Cardiovascular disease has become one of the major diseases threatening human life and health. Electrocardiogram (ECG)is a common clinical diagnosis of the important methods for arrhythmia and is widely used in health monitoring of patients with heart disease. As a result of the existing medical resources, the use of artificial intelligence method to analysis and diagnosis in order to overcome these limitations of increasingly urgent demand, the use of automatic detection and classification methods in clinical practice may help doctors make accurate and rapid diagnosis of diseases. In this paper, eight common arrhythmia types are classified, and a deep learning classification method of ECG signals based on adaptive antagonism domain is proposed, which solves and improves the problems of insufficient training sample labeling and data distribution differences caused by individual differences. This method consists of three modules: Multi-scale feature extraction module A, domain recognition module B and multi-classifier module C. Module A is composed of two groups of different parallel convolution blocks, which increases the width of feature extraction. Module B is composed of three convolution blocks and a fully connected layer to fully extract shallow features. In module C, time features and deep learning extracted features are connected in series on the fully connected layer to enhance feature diversity. The experimental results show that the accuracy, sensitivity and positive predictive value of this method can reach 98.8%, 97.9% and 98.1%, and the proposed model can help doctors accurately detect different types of arrhythmias in routine electrocardiogram.

Key words: ECG signal classification, deep learning, multi-scale, time characteristics, adversarial domain adaptation