English

Optimal choice of $k$ for $k$-nearest neighbor regression

Statistics Theory 2020-02-18 v4 Statistics Theory

Abstract

The kk-nearest neighbor algorithm (kk-NN) is a widely used non-parametric method for classification and regression. We study the mean squared error of the kk-NN estimator when kk is chosen by leave-one-out cross-validation (LOOCV). Although it was known that this choice of kk is asymptotically consistent, it was not known previously that it is an optimal kk. We show, with high probability, the mean squared error of this estimator is close to the minimum mean squared error using the kk-NN estimate, where the minimum is over all choices of kk.

Keywords

Cite

@article{arxiv.1909.05495,
  title  = {Optimal choice of $k$ for $k$-nearest neighbor regression},
  author = {Mona Azadkia},
  journal= {arXiv preprint arXiv:1909.05495},
  year   = {2020}
}