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

Journal: BioMed research international
Published Date:

Abstract

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.

Authors

  • Fei Yang
    Hunan Province Key Laboratory of Typical Environmental Pollution and Health Hazards, School of Public Health, University of South China, Hengyang 421001, China.
  • Jiazhi Du
    School of Computer Science and Technology, Harbin Institute of Technology, Harbin, China.
  • Jiying Lang
    School of Mechanical, Electrical and Information Engineering, Shandong University, Weihai, China.
  • Weigang Lu
    Department of Educational Technology, Ocean University of China, Qingdao, China.
  • Lei Liu
    Department of Science and Technology, Beijing Shijitan Hospital, Capital Medical University, Beijing, China.
  • Changlong Jin
    School of Mechanical, Electrical and Information Engineering, Shandong University, Weihai, China.
  • Qinma Kang
    School of Mechanical, Electrical and Information Engineering, Shandong University, Weihai, China.