Enhanced tolerance-based intuitionistic fuzzy rough set theory feature selection and ResNet-18 feature extraction model for arrhythmia classification

Abstract

Arrhythmia classification on Electrocardiogram (ECG) signals is an important process for the diagnosis of cardiac disease and arrhythmia disease. The existing researches in arrhythmia classification have limitations of imbalance data problem and overfitting in classification. This research applies Fuzzy C-Means (FCM) – Enhanced Tolerance-based Intuitionistic Fuzzy Rough Set Theory (ETIFRST) for feature selection in arrhythmia classification. The selected features from FCM-ETIFRST were applied to the Multi-class Support Vector Machine (MSVM) for arrhythmia classification. The ResNet18 – Convolution Neural Network (CNN) was applied for feature extraction in input signal to overcome imbalance data problem. Conventional feature extraction along with CNN features are applied for FCM-ETIFRST feature selection process. The FCM-ETIFRST method in arrhythmia classification is evaluated on MIT-BIH and CPCS 2018 dataset. The FCM-ETIFRST has 98.95% accuracy and Focal loss-CNN has 98.66% accuracy on MIT-BIH dataset. The FCM-ETIFRST method has 98.45% accuracy and Explainable Deep learning Model (XDM) method have 93.6% accuracy on CPCS 2018 dataset.