ECG arrhythmia classification using a 2-D convolutional neural network

Abstract

In this paper, we propose an effective electrocardiogram (ECG) arrhythmia classification method using a deep two-dimensional convolutional neural network (CNN) which recently shows outstanding performance in the field of pattern recognition. Every ECG beat was transformed into a two-dimensional grayscale image as an input data for the CNN classifier. Optimization of the proposed CNN classifier includes various deep learning techniques such as batch normalization, data augmentation, Xavier initialization, and dropout. In addition, we compared our proposed classifier with two well-known CNN models; AlexNet and VGGNet. ECG recordings from the MIT-BIH arrhythmia database were used for the evaluation of the classifier. As a result, our classifier achieved 99.05% average accuracy with 97.85% average sensitivity. To precisely validate our CNN classifier, 10-fold cross-validation was performed at the evaluation which involves every ECG recording as a test data. Our experimental results have successfully validated that the proposed CNN classifier with the transformed ECG images can achieve excellent classification accuracy without any manual pre-processing of the ECG signals such as noise filtering, feature extraction, and feature reduction.

Submitted to journal

References in corpus (8)

• Adam: A Method for Stochastic Optimization
• Very Deep Convolutional Networks for Large-Scale Image Recognition
• TensorFlow: Large-Scale Machine Learning on Heterogeneous Distributed Systems
• ADADELTA: An Adaptive Learning Rate Method
• Fast and Accurate Deep Network Learning by Exponential Linear Units (ELUs)
• cuDNN: Efficient Primitives for Deep Learning
• Cardiologist-Level Arrhythmia Detection with Convolutional Neural Networks
• Stochastic Pooling for Regularization of Deep Convolutional Neural Networks

Cited by in corpus (5)

• Topological Data Analysis for Arrhythmia Detection through Modular Neural Networks
• Atrial Fibrillation Detection and ECG Classification based on CNN-BiLSTM
• Personalized Deep Learning for Ventricular Arrhythmias Detection on Medical IoT Systems
• A PMU-Based Machine Learning Application for Fast Detection of Forced Oscillations from Wind Farms
• Strengthening the Training of Convolutional Neural Networks By Using Walsh Matrix