重庆理工大学学报2024，Vol.38Issue(1)：142-149,8.DOI:10.3969/j.issn.1674-8425(z).2024.01.016
一种改进U-Net网络的心电图分类算法研究
Study on ECG classification algorithm based on improved U-Net network
摘要
Abstract
Cardiovascular disease,with the highest mortality rate across the globe,kills over ten million people every year.Thanks to the continuous development of artificial intelligence,patients'heart conditions can now be quickly and accurately diagnosed with the assistance of automatic electrocardiogram anomaly classification technology.This paper proposes an electrocardiogram classification algorithm based on CPSC-2018 twelve lead data,which combines U-Net network and attention mechanism.First,the data are processed for equal length and normalization to address their varied lengths.Then,the preprocessed data with longer lengths are reprocessed by the skip layer connection and encoding and decoding methods in the U-Net network.An attention mechanism is added to the last layer of U-Net network decoding to combat noise and improve the effective information attention and accuracy of the model.Finally,CPSC-2018 dataset is employed to verify the model.Our experimental results show the proposed model delivers fairly satisfying classification performance,recording its accuracy,recall,and F1 values of over 90％in identifying atrial fibrillation(AF)and right bundle branch block(RBBB)arrhythmias,and an average F1 value of 82.5％.关键词
心律失常/心电图/U-Net网络/注意力机制Key words
arrhythmia/electrocardiogram/U-Net network/attention mechanism分类
医药卫生引用本文复制引用
王建荣,尉向前,辛彬彬,高睿丰,李国翚..一种改进U-Net网络的心电图分类算法研究[J].重庆理工大学学报,2024,38(1):142-149,8.基金项目
国家重点研发计划(2018YFC2000701) （2018YFC2000701）
中国博士后科学基金项目(2021M692400) （2021M692400）
山西省基础研究计划项目(202203021221017) （202203021221017）
