A novel multi-module neural network system for imbalanced heartbeats classification

被引：27
作者：
Jiang J. [1 ]
Zhang H. [1 ]
Pi D. [1 ]
Dai C. [1 ]
机构：
[1] College of Computer Science and Technology, Nanjing University of Aeronautics and Astronautics, Nanjing
来源：
Expert Systems with Applications: X | 2019年 / 1卷
基金：
中国国家自然科学基金;
关键词：
Convolutional neural network; Heartbeats classification; Imbalance problem; Multi-module;
D O I：
10.1016/j.eswax.2019.100003
中图分类号：
学科分类号：
摘要：
In this paper, a novel multi-module neural network system named MMNNS is proposed to solve the imbalance problem in electrocardiogram (ECG) heartbeats classification. Four submodules are designed to construct the system: preprocessing, imbalance problem processing, feature extraction and classification. Imbalance problem processing module mainly introduces three methods: BLSM, CTFM and 2PT, which are proposed from three aspects of resampling, data feature and algorithm respectively. BLSM is used to synthesize virtual samples linearly around the minority samples. CTFM consists of DAE-based feature extraction part and QRS-based feature selection part, in which selected features and complete features are applied to determine the heartbeat class simultaneously. The processed data are fed into a convolutional neural network (CNN) by applying 2PT to train and fine-tune. MMNNS is trained on MIT-BIH Arrhythmia Database following AAMI standard, using intra-patient and inter-patient scheme, especially the latter which is strongly recommended. The comparisons with several state-of-the-art methods using standard criteria on three datasets demonstrate the superiority of MMNNS for improving detection of heartbeats and addressing imbalance in ECG heartbeats classification. © 2019 The Authors
引用
收藏
相关论文
共 66 条
[31]  
Lecun Y., Bengio Y., Hinton G., Deep learning, Nature, 521, pp. 436-444, (2015)
[32]  
Lee H.J., Cho S., The novelty detection approach for different degrees of class imbalance, 13th International Conference on Neural Information Processing, pp. 21-30, (2006)
[33]  
Li D., Zhang J., Zhang Q., Wei X., Classification of ECG signals based on 1D convolution neural network, IEEE International Conference on E-Health Networking, Applications and Services, pp. 1-6, (2017)
[34]  
Li Q., Rajagopalan C., Clifford G.D., Ventricular fibrillation and tachycardia classification using a machine learning approach, IEEE Transactions on Bio-medical Engineering, 61, pp. 1607-1613, (2014)
[35]  
Li T., Zhou M., ECG classification using wavelet packet entropy and random forests, Entropy, 18, (2016)
[36]  
Li Y., Guo H., Liu X., Li Y., Li J., Adapted ensemble classification algorithm based on multiple classifier system and feature selection for classifying multi-class imbalanced data, Knowledge-Based Systems, 94, pp. 88-104, (2016)
[37]  
Liu X.Y., Wu J., Zhou Z.H., Exploratory undersampling for class-imbalance learning, IEEE Transactions on Systems Man & Cybernetics Part B, 39, pp. 539-550, (2009)
[38]  
Maldonado S., Lopez J., Imbalanced data classification using second-order cone programming support vector machines, Pattern Recognition, 47, pp. 2070-2079, (2014)
[39]  
Maldonado S., Lopez J., Vairetti C., An alternative SMOTE oversampling strategy for high-dimensional datasets, Applied Soft Computing, 76, pp. 380-389, (2019)
[40]  
Maratea A., Petrosino A., Manzo M., Adjusted F-measure and kernel scaling for imbalanced data learning, Information Sciences, 257, pp. 331-341, (2014)
1234567