A Comparative Analysis of Deep Neural Network Models using Transfer Learning for Electrocardiogram Signal Classification

Nitin Rahuja,S. K. Valluru
DOI: https://doi.org/10.1109/RTEICT52294.2021.9573692
2021-08-27
Abstract:With the development of computational resources in the past decade, the adoption of deep learning methodologies has taken a steep rise. Particularly, the use of various Convolutional Neural Network (CNN) based architectures for various computer vision and image classification problems in healthcare research can be witnessed. In this scope, performance evaluation and comparison of three different CNN architectures for multi-class classification of Electrocardiogram (ECG) signals is presented in this paper. Three distinct classes of ECG signal i.e. Normal Sinus Rhythm (NSR), Arrhythmia (ARR), and Congestive Heart failure (CHF) are utilized in this work, which is representative of an individual's heart conditions. The classification methodology adopted makes use of Continuous Wavelet Transform (CWT) which provides a 2-dimensional time-frequency representation of available samples of ECG signals. ECG dataset was collected from different PhysioBank databases, which provides openly-accessible medical research data. Images with time-frequency representation of ECG signals were applied as input to three CNN architectures namely, AlexNet, GoogLeNet, and SqueezeNet. Using transfer learning approach and modification in certain output layers of three architectures, ECG classification was performed and the performance of three CNN architectures was studied. The results revealed classification accuracy of 97.80%, 97.78% and 97.22% by AlexNet, GoogLeNet and SqueezeNet respectively. AlexNet outperformed GoogLeNet and SqueezeNet models in terms of accuracy as well as training time.
Computer Science,Medicine
What problem does this paper attempt to address?