Diagnosis of Arrhythmia Based on Multi-scale Feature Fusion and Imbalanced Data

被引:2
作者
Cheng, Z. [1 ]
Liu, Zx [1 ]
Yang, Gl [2 ]
机构
[1] Wuhan Univ Sci & Technol, Sch Informat Sci & Engn, Wuhan 430081, Peoples R China
[2] Wuhan Univ Sci & Technol, Hosp WUST, Wuhan 430081, Peoples R China
来源
PROCEEDINGS OF 2022 7TH INTERNATIONAL CONFERENCE ON MACHINE LEARNING TECHNOLOGIES, ICMLT 2022 | 2022年
关键词
Deep learning; ADASYN; Electrocardiogram; Wavelet denoising; Multi-scale feature fusion;
D O I
10.1145/3529399.3529415
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Evidence suggests that Electrocardiogram (ECG) analysis plays an important role in the diagnosis of arrhythmia and the prevention of cardiovascular diseases. Extracting disease-related signals from ECG to improve the diagnostic efficiency of arrhythmia is still a challenging problem at present. In this paper, we propose a network framework based on multi-scale feature fusion and imbalanced data to classify and diagnose arrhythmias. The original ECG signals are denoised by improved wavelet threshold and the data issue is optimized by adaptive oversampling algorithm ADASYN, which can effectively improve the learning ability of the model for different types of samples. At the same time, in order to improve the classification efficiency of the model, we construct a deep learning framework based on the fusion of artificial features, convolutional depth features and temporal features. In the MIT-BIH database, the best accuracy of this method is 99.47%, which is higher than previous advanced methods. The average accuracy of 99.40% was obtained by cross-validation for the generalization performance of the model. The results show that the proposed method has excellent performance in ECG feature extraction and diagnosis of arrhythmias.
引用
收藏
页码:92 / 98
页数:7
相关论文
共 17 条
[1]   Remaining Useful Life Estimation for Prognostics of Lithium-Ion Batteries Based on Recurrent Neural Network [J].
Catelani, Marcantonio ;
Ciani, Lorenzo ;
Fantacci, Romano ;
Patrizi, Gabriele ;
Picano, Benedetta .
IEEE TRANSACTIONS ON INSTRUMENTATION AND MEASUREMENT, 2021, 70
[2]   Nonstationary Signal Denoising Using an Envelope-Tracking Filter [J].
Chen, Shiqian ;
Dong, Xingjian ;
Xiong, Yuyong ;
Peng, Zhike ;
Zhang, Wenming .
IEEE-ASME TRANSACTIONS ON MECHATRONICS, 2018, 23 (04) :2004-2015
[3]   DE-NOISING BY SOFT-THRESHOLDING [J].
DONOHO, DL .
IEEE TRANSACTIONS ON INFORMATION THEORY, 1995, 41 (03) :613-627
[4]   Cardiologist-level arrhythmia detection and classification in ambulatory electrocardiograms using a deep neural network [J].
Hannun, Awni Y. ;
Rajpurkar, Pranav ;
Haghpanahi, Masoumeh ;
Tison, Geoffrey H. ;
Bourn, Codie ;
Turakhia, Mintu P. ;
Ng, Andrew Y. .
NATURE MEDICINE, 2019, 25 (01) :65-+
[5]   ADASYN: Adaptive Synthetic Sampling Approach for Imbalanced Learning [J].
He, Haibo ;
Bai, Yang ;
Garcia, Edwardo A. ;
Li, Shutao .
2008 IEEE INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS, VOLS 1-8, 2008, :1322-1328
[6]   Automatic Cardiac Arrhythmia Classification Using Combination of Deep Residual Network and Bidirectional LSTM [J].
He, Runnan ;
Liu, Yang ;
Wang, Kuanquan ;
Zhao, Na ;
Yuan, Yongfeng ;
Li, Qince ;
Zhang, Henggui .
IEEE ACCESS, 2019, 7 :102119-102135
[7]   LSTM-Based Auto-Encoder Model for ECG Arrhythmias Classification [J].
Hou, Borui ;
Yang, Jianyong ;
Wang, Pu ;
Yan, Ruqiang .
IEEE TRANSACTIONS ON INSTRUMENTATION AND MEASUREMENT, 2020, 69 (04) :1232-1240
[8]   An efficient ECG arrhythmia classification method based on Manta ray foraging optimization [J].
Houssein, Essam H. ;
Ibrahim, Ibrahim E. ;
Neggaz, Nabil ;
Hassaballah, M. ;
Wazery, Yaser M. .
EXPERT SYSTEMS WITH APPLICATIONS, 2021, 181
[9]   Cardiac arrhythmia classification using tunable Q-wavelet transform based features and support vector machine classifier [J].
Jha, Chandan Kumar ;
Kolekar, Maheshkumar H. .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2020, 59
[10]   Heartbeat classification using deep residual convolutional neural network from 2-lead electrocardiogram [J].
Li, Zhi ;
Zhou, Dengshi ;
Wan, Li ;
Li, Jian ;
Mou, Wenfeng .
JOURNAL OF ELECTROCARDIOLOGY, 2020, 58 :105-112