K nearest neighbours with mutual information for simultaneous classification and missing data imputation

  • Authors:
  • Pedro J. García-Laencina;José-Luis Sancho-Gómez;Aníbal R. Figueiras-Vidal;Michel Verleysen

  • Affiliations:
  • Department of Information and Communications Technologies, Universidad Politécnica de Cartagena, Plaza del Hospital 1, 30202 Cartagena, Murcia, Spain;Department of Information and Communications Technologies, Universidad Politécnica de Cartagena, Plaza del Hospital 1, 30202 Cartagena, Murcia, Spain;Department of Signal Theory and Communications, Universidad Carlos III de Madrid, Avda. de la Universidad 30, 28911 Leganés, Madrid, Spain;Université catholique de Louvain, Machine Learning Group, DICE. 3 place du Levant, B-1348 Louvain-la-Neuve, Belgium

  • Venue:
  • Neurocomputing
  • Year:
  • 2009

Quantified Score

Hi-index 0.01

Visualization

Abstract

Missing data is a common drawback in many real-life pattern classification scenarios. One of the most popular solutions is missing data imputation by the K nearest neighbours (KNN) algorithm. In this article, we propose a novel KNN imputation procedure using a feature-weighted distance metric based on mutual information (MI). This method provides a missing data estimation aimed at solving the classification task, i.e., it provides an imputed dataset which is directed toward improving the classification performance. The MI-based distance metric is also used to implement an effective KNN classifier. Experimental results on both artificial and real classification datasets are provided to illustrate the efficiency and the robustness of the proposed algorithm.