A novel kNN algorithm with data-driven k parameter computation

被引:136
作者
Zhang, Shichao [1 ]
Cheng, Debo [1 ,2 ]
Deng, Zhenyun [1 ]
Zong, Ming [3 ]
Deng, Xuelian [4 ]
机构
[1] Guangxi Normal Univ, Guangxi Key Lab Multisource Informat Min & Secur, Guilin, Guangxi, Peoples R China
[2] Univ South Australia, Informat Technol & Math Sci, Adelaide, SA, Australia
[3] Massey Univ, Inst Nat & Math Sci, Auckland, New Zealand
[4] Guangxi Univ Chinese Med, Coll Publ Hlth & Management, Nanning, Guangxi, Peoples R China
关键词
kNN method; kNN prediction; Parameter computation; REGRESSION; IMPUTATION; CLASSIFICATION; OPTIMIZATION; SELECTION;
D O I
10.1016/j.patrec.2017.09.036
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
This paper studies an example-driven k-parameter computation that identifies different k values for different test samples in kNN prediction applications, such as classification, regression and missing data imputation. This is carried out with reconstructing a sparse coefficient matrix between test samples and training data. In the reconstruction process, an l(1)-norm regularization is employed to generate an element-wise sparsity coefficient matrix, and an LPP (Locality Preserving Projection) regularization is adopted to keep the local structures of data for achieving the efficiency. Further, with the learnt k value, k NN approach is applied to classification, regression and missing data imputation. We experimentally evaluate the proposed approach with 20 real datasets, and show that our algorithm is much better than previous k NN algorithms in terms of data mining tasks, such as classification, regression and missing value imputation. (C) 2017 Elsevier B.V. All rights reserved.
引用
收藏
页码:44 / 54
页数:11
相关论文
共 50 条