ECG feature extraction based on the bandwidth properties of variational mode decomposition

Abstract

It is a difficult process to detect abnormal heart beats, known as arrhythmia, in long-term ECG recording. Thus, computer-aided diagnosis systems have become a supportive tool for helping physicians improve the diagnostic accuracy of heartbeat detection. This paper explores the bandwidth properties of the modes obtained using variational mode decomposition (VMD) to classify arrhythmia electrocardiogram (ECG) beats. VMD is an enhanced version of the empirical mode decomposition (EMD) algorithm for analyzing non-linear and non-stationary signals. It decomposes the signal into a set of band-limited oscillations called modes. ECG signals from the MIT-BIH arrhythmia database are decomposed using VMD, and the amplitude modulation bandwidth BAM, the frequency modulation bandwidth BFM and the total bandwidth B of the modes are used as feature vectors to detect heartbeats such as normal (N), premature ventricular contraction (V), left bundle branch block (L), right bundle branch block (R), paced beat (P) and atrial premature beat (A). Bandwidth estimations based on the instantaneous frequency (IF) and amplitude (IA) spectra of the modes indicate that the proposed VMD-based features have sufficient class discrimination capability regarding ECG beats. Moreover, the extracted features using the bandwidths (BAM, BFM and B) of four modes are used to evaluate the diagnostic accuracy rates of several classifiers such as the k-nearest neighbor classifier (k-NN), the decision tree (DT), the artificial neural network (ANN), the bagged decision tree (BDT), the AdaBoost decision tree (ABDT) and random sub-spaced k-NN (RSNN) for N, R, L, V, P, and A beats. The performance of the proposed VMD-based feature extraction with a BDT classifier has accuracy rates of 99.06%, 99.00%, 99.40%, 99.51%, 98.72%, 98.71%, and 99.02% for overall, N-, R-, L-, V-, P-, and A-type ECG beats, respectively.