DE-PNN: Differential Evolution-Based Feature Optimization with Probabilistic Neural Network for Imbalanced Arrhythmia Classification

In this research, a heartbeat classification method is presented based on evolutionary feature optimization using differential evolution (DE) and classification using a probabilistic neural network (PNN) to discriminate between normal and arrhythmic heartbeats. The proposed method follows four steps: (1) preprocessing, (2) heartbeat segmentation, (3) DE feature optimization, and (4) PNN classification. In this method, we have employed direct signal amplitude points constituting the heartbeat acquired from the ECG holter device with no secondary feature extraction step usually used in case of hand-crafted, frequency transformation or other features. The heartbeat types include normal, left bundle branch block, right bundle branch block, premature ventricular contraction, atrial premature, ventricular escape, ventricular flutter and paced beat. Using ECG records from the MIT-BIH, heartbeats are identified to start at 250 ms before and end at 450 ms after the respective R-peak positions. In the next step, the DE method is applied to reduce and optimize the direct heartbeat features. Although complex and highly computational ECG heartbeat classification algorithms have been proposed in the literature, they failed to achieve high performance in detecting some minority heartbeat categories, especially for imbalanced datasets. To overcome this challenge, we propose an optimization step for the deep CNN model using a novel classification metric called the Matthews correlation coefficient (MCC). This function focuses on arrhythmia (minority) heartbeat classes by increasing their importance. Maximum MCC is used as a fitness function to identify the optimum combination of features for the uncorrelated and non-uniformly distributed eight beat class samples. The proposed DE-PNN scheme can provide better classification accuracy considering 8 classes with only 36 features optimized from a 253 element feature set implying an 85.77% reduction in direct amplitude features. Our proposed method achieved overall 99.33% accuracy, 94.56% F1, 93.84% sensitivity, and 99.21% specificity.