Deep Learning Framework with ECG Feature-Based Kernels for Heart Disease Classification

Abstract

Heart disease classification with high accuracy can support the physician’s correct decision on patients. This paper proposes a kernel size calculation based on P, Q, R, and S waves of one heartbeat to enhance classification accuracy in a deep learning framework. In addition, Electrocardiogram (ECG) signals were filtered using wavelet transform with dmey wavelet, in which the shape of the dmey is closed to that of one heartbeat. With this selected dmey, each heartbeat was standardized with 300 samples for calculation of kernel sizes so that it contains most features in each heartbeat. Therefore, in this research, with 103,459 heart rhythms from the MIT-BIH Arrhythmia Database, the proposed approach for calculation of kernel sizes is effective with seven convolutional layers and other fully connected layers in a Deep Neural Network (DNN). In particular, with five types of heart disease, the result of the high classification accuracy is about 99.4 %. It means that the proposed kernel size calculation in the convolutional layers can achieve good classification performance and it may be developed for classifying different types of disease.