基于注意力机制与卷积神经网络的不均衡多标签心电图分类方法研究  

Attention-Based Convolutional Neural Network for Imbalanced Multi-Label Electrocardiogram

作　　者：张诗雨 孙占全 顾浩丞 陈骏立 Shiyu Zhang;Zhanquan Sun;Haocheng Gu;Junli Chen(College of Science,University of Shanghai for Science and Technology,Shanghai;School of Optical-Electrical and Computer Engineering,University of Shanghai for Science and Technology,Shanghai)

机构地区：[1]上海理工大学理学院,上海 [2]上海理工大学光电信息与计算机工程学院,上海

出　　处：《建模与仿真》2024年第4期4317-4334,共18页Modeling and Simulation

基　　金：国防基础研究计划项目(JCKY2019413D001);上海理工大学医学工程交叉项目(10-21-302-413);上海市自然科学基金项目(19ZR1436000)部分资助。

摘　　要：心血管疾病是全球主要死因之一,及早发现心脏病的风险因素对于预防猝死至关重要,基于心电图的心律失常自动检测是心脏病筛查的重要手段,对降低心脏病死亡率至关重要。临床上同一患者可能同时出现多个心电异常类型,因此,解决心电图数据中多标签问题且考虑标签关系显得尤为重要。本文提出了一个多标签心电图分类模型,包括卷积神经网络、Transformer和注意力机制等模块,其中卷积神经网络用于提取局部特征,Transformer用于提取全局特征,采用非对称损失函数平衡正负标签。对两个心电数据集进行实验,实验结果表明,本文提出的模型相较于现有心电图分类方法,表现更优异。Cardiovascular disease is one of the leading causes of death worldwide.Early detection of risk factors for heart disease is essential to prevent sudden death.The automatic detection of arrhythmia based on electrocardiogram is an important means of heart disease screening,which is crucial to reduce the mortality of heart disease.In clinical settings,patients may exhibit multiple abnormal categories rather than just one.Therefore,it becomes necessary to address the issue of multiple labels in electrocardiogram data and consider the relationships between these labels.We propose a multi-label classification model composed of convolutional neural networks,Transformers and attention mechanisms.The convolutional neural network is used to extract local features,Transfor-mer is used to extract global features,and an asymmetric loss function is used to balance positive and negative labels.Experimental results on two electrocardiogram dataset show that the proposed model has better performance than the existing electrocardiogram classification methods.

关 键 词：心电信号分类 多标签 注意力机制 TRANSFORMER 深度学习 

分 类 号：R54[医药卫生—心血管疾病]