Nearest Neighbour Based Estimates of Gradients: Sharp Nonasymptotic Bounds and Applications - Télécom Paris
Communication Dans Un Congrès Année : 2021

Nearest Neighbour Based Estimates of Gradients: Sharp Nonasymptotic Bounds and Applications

Résumé

Motivated by a wide variety of applications, ranging from stochastic optimization to dimension reduction through variable selection, the problem of estimating gradients accurately is of crucial importance in statistics and learning theory. We consider here the classical regression setup, where a real valued square integrable r.v. Y is to be predicted upon observing a (possibly high dimensional) random vector X by means of a predictive function f (X) as accurately as possible in the meansquared sense and study a nearest-neighbourbased pointwise estimate of the gradient of the optimal predictive function, the regression function m(x) = E[Y | X = x]. Under classical smoothness conditions combined with the assumption that the tails of Y − m(X) are sub-Gaussian, we prove nonasymptotic bounds improving upon those obtained for alternative estimation methods. Beyond the novel theoretical results established, several illustrative numerical experiments have been carried out. The latter provide strong empirical evidence that the estimation method proposed here performs very well for various statistical problems involving gradient estimation, namely dimensionality reduction, stochastic gradient descent optimization and disentanglement quantification.
Fichier principal
Vignette du fichier
ausset21a.pdf (493.07 Ko) Télécharger le fichier
Origine Fichiers éditeurs autorisés sur une archive ouverte

Dates et versions

hal-03559386 , version 1 (06-02-2022)

Identifiants

  • HAL Id : hal-03559386 , version 1

Citer

Stéphan Clémençon, Guillaume Ausset, François Portier. Nearest Neighbour Based Estimates of Gradients: Sharp Nonasymptotic Bounds and Applications. AISTATS 2021 : The 24th International Conference on Artificial Intelligence and Statistics, May 2021, Virtual conference, France. pp.1-12. ⟨hal-03559386⟩
49 Consultations
78 Téléchargements

Partager

More