Missing Value Estimation Methods Research for Arrhythmia Classification Using the Modified Kernel Difference-Weighted KNN Algorithms

Abstract EN

Electrocardiogram (ECG) signal is critical to the classification of cardiac arrhythmia using some machine learning methods.

In practice, the ECG datasets are usually with multiple missing values due to faults or distortion.

Unfortunately, many established algorithms for classification require a fully complete matrix as input.

Thus it is necessary to impute the missing data to increase the effectiveness of classification for datasets with a few missing values.

In this paper, we compare the main methods for estimating the missing values in electrocardiogram data, e.g., the “Zero method”, “Mean method”, “PCA-based method”, and “RPCA-based method” and then propose a novel KNN-based classification algorithm, i.e., a modified kernel Difference-Weighted KNN classifier (MKDF-WKNN), which is fit for the classification of imbalance datasets.

The experimental results on the UCI database indicate that the “RPCA-based method” can successfully handle missing values in arrhythmia dataset no matter how many values in it are missing and our proposed classification algorithm, MKDF-WKNN, is superior to other state-of-the-art algorithms like KNN, DS-WKNN, DF-WKNN, and KDF-WKNN for uneven datasets which impacts the accuracy of classification.