arXivDaily arXiv每日学术速递 周一至周五更新
今晚才能更新了~~大家休息下再来吧
arXiv 2008.08718stat.MLcs.LGmath.STstat.TH
收藏

最小差异原则策略用于选择 $k$-NN 回归中的 $k$

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

Inria(法国国家信息与自动化研究所);Paris 1 - Panthéon Sorbonne University(巴黎第一大学先贤祠-索邦大学)

Yaroslav Averyanov, Alain Celisse

AI总结:

本文提出一种基于早停和最小差异原则的无留出数据策略来选择 $k$-NN 回归中的 $k$,该方法在保持最小最大最优性的同时降低计算时间,并在人工和真实数据上优于其他模型选择策略。

摘要:

我们提出了一种新颖的数据驱动策略,在不使用任何留出数据的情况下,为 $k$-NN 回归估计器选择超参数 $k$。我们将超参数选择问题视为一个(关于 $k$ 的)迭代过程,并提出使用一种基于早停和最小差异原则思想的易于在实践中实施的策略。该模型选择策略被证明在某些光滑函数类(例如,有界域上的 Lipschitz 函数类)上是最小最大最优的。与留出法、5折交叉验证和 AIC 准则等其他模型选择策略相比,这种新方法在人工和真实数据集上通常能提高统计性能。该策略的新颖之处在于,在保持所得估计器的统计(最小最大)最优性的同时,减少了模型选择过程的计算时间。更准确地说,给定一个大小为 $n$ 的样本,如果应从 $\{1, \ldots, n\}$ 中选择 $k$,且 $\{f^1, \ldots, f^n\}$ 是回归函数的估计器,则最小差异原则需要计算估计器的一小部分,而广义交叉验证、Akaike 的 AIC 准则或 Lepskii 原则则不是这种情况。

英文摘要:

We present a novel data-driven strategy to choose the hyperparameter $k$ in the $k$-NN regression estimator without using any hold-out data. We treat the problem of choosing the hyperparameter as an iterative procedure (over $k$) and propose using an easily implemented in practice strategy based on the idea of early stopping and the minimum discrepancy principle. This model selection strategy is proven to be minimax-optimal over some smoothness function classes, for instance, the Lipschitz functions class on a bounded domain. The novel method often improves statistical performance on artificial and real-world data sets in comparison to other model selection strategies, such as the Hold-out method, 5-fold cross-validation, and AIC criterion. The novelty of the strategy comes from reducing the computational time of the model selection procedure while preserving the statistical (minimax) optimality of the resulting estimator. More precisely, given a sample of size $n$, if one should choose $k$ among $\left\{ 1, \ldots, n \right\}$, and $\left\{ f^1, \ldots, f^n \right\}$ are the estimators of the regression function, the minimum discrepancy principle requires the calculation of a fraction of the estimators, while this is not the case for the generalized cross-validation, Akaike's AIC criteria, or Lepskii principle.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑