A Spatial Pyramid Pooling-Based Deep Convolutional Neural Network for the Classification of Electrocardiogram Beats

Research output: Journal Publications and Reviews (RGC: 21, 22, 62)21_Publication in refereed journal

3 Scopus Citations
View graph of relations

Author(s)

  • Jia Li
  • Yujuan Si
  • Liuqi Lang
  • Lixun Liu
  • Tao Xu

Related Research Unit(s)

Detail(s)

Original languageEnglish
Article number1590
Journal / PublicationApplied Sciences (Switzerland)
Volume8
Issue number9
Publication statusPublished - 8 Sep 2018

Link(s)

Abstract

An accurate electrocardiogram (ECG) beat classification can benefit the diagnosis of the cardiovascular disease. Deep convolutional neural networks (CNN) can automatically extract valid features from data, which is an effective way for the classification of the ECG beats. However, the fully-connected layer in CNNs requires a fixed input dimension, which limits the CNNs to receive fixed-scale inputs. Signals of different scales are generally processed into the same size by segmentation and downsampling. If information loss occurs during a uniformly-sized process, the classification accuracy will ultimately be affected. To solve this problem, this paper constructs a new CNN framework spatial pyramid pooling (SPP) method, which solves the deficiency caused by the size of input data. The Massachusetts Institute of Technology-Biotechnology (MIT-BIH) arrhythmia database is employed as the training and testing data for the classification of heartbeat signals into six categories. Compared with the traditional method, which may lose a large amount of important information and easy to be over-fitted, the robustness of the proposed method can be guaranteed by extracting data features from different sizes. Experimental results show that the proposed architecture network can extract more high-quality features and exhibits higher classification accuracy (94%) than the traditional deep CNNs (90.4%).

Research Area(s)

  • Classification, Convolutional neural networks, ECG beats, Feature extraction, Spatial pyramid pooling

Download Statistics

No data available