Abstract:k nearest neighbor (kNN) method is a popular classification method in data mining and statistics because of its simple implementation and significant classification performance. However, it is impractical for traditional kNN methods to assign a fixed k value (even though set by experts) to all test samples. Previous solutions assign different k values to different test samples by the cross validation method but are usually time-consuming. This paper proposes a kTree method to learn different optimal k values for different test/new samples, by involving a training stage in the kNN classification. Specifically, in the training stage, kTree method first learns optimal k values for all training samples by a new sparse reconstruction model, and then constructs a decision tree (namely, kTree) using training samples and the learned optimal k values. In the test stage, the kTree fast outputs the optimal k value for each test sample, and then, the kNN classification can be conducted using the learned optimal k value and all training samples. As a result, the proposed kTree method has a similar running cost but higher classification accuracy, compared with traditional kNN methods, which assign a fixed k value to all test samples. Moreover, the proposed kTree method needs less running cost but achieves similar classification accuracy, compared with the newly kNN methods, which assign different k values to different test samples. This paper further proposes an improvement version of kTree method (namely, k*Tree method) to speed its test stage by extra storing the information of the training samples in the leaf nodes of kTree, such as the training samples located in the leaf nodes, their kNNs, and the nearest neighbor of these kNNs. We call the resulting decision tree as k*Tree, which enables to conduct kNN classification using a subset of the training samples in the leaf nodes rather than all training samples used in the newly kNN methods. This actually reduces running cost of test stage. Finally, the experimental results on 20 real data sets showed that our proposed methods (i.e., kTree and k*Tree) are much more efficient than the compared methods in terms of classification tasks.

Improving the Predictive Performances of $k$ Nearest Neighbors Learning by Efficient Variable Selection

Efficient Estimation of k for the Nearest Neighbors Class of Methods

Improved K nearest neighbor nonparametric regression and its application in short-term traffic flow forecasting

Introduction to Machine Learning: K-Nearest Neighbors

A New Hashing based Nearest Neighbors Selection Technique for Big Datasets

An Efficient Variable Selection Method for Predictive Discriminant Analysis

Neighborhood Ranking-Based Feature Selection

Twin Neural Network Improved k-Nearest Neighbor Regression

Ensemble Learning with Extremely Randomized k-Nearest Neighbors for Accurate and Efficient Classification

Improved neighborhood-based algorithms for large-scale collaborative-filtering

Bayes-Decisive Linear KNN with Adaptive Nearest Neighbors

Selecting Optimal Decisions Via Distributionally Robust Nearest-Neighbor Regression.

Enhancing K-nearest neighbor algorithm: a comprehensive review and performance analysis of modifications

Efficient Knn Classification with Different Numbers of Nearest Neighbors.

Minimum discrepancy principle strategy for choosing $k$ in $k$-NN regression

Back-propagating Errors Through Artificial Neural Networks for Variable Selection

A New k-Nearest Neighbors Algorithm for Learning from Multiple Experts' Uncertain Decisions

A Theory-Based Evaluation of Nearest Neighbor Models Put Into Practice

A new two-layer nearest neighbor selection method for kNN classifier

Surprisal Driven $k$-NN for Robust and Interpretable Nonparametric Learning

A Sparse Reconstructive Evidential -Nearest Neighbor Classifier for High-dimensional Data