Nearest Neighbor Sampling for Covariate Shift Adaptation - Archive ouverte HAL Access content directly
Preprints, Working Papers, ... Year : 2024

Nearest Neighbor Sampling for Covariate Shift Adaptation

Abstract

Many existing covariate shift adaptation methods estimate sample weights given to loss values to mitigate the gap between the source and the target distribution. However, estimating the optimal weights typically involves computationally expensive matrix inversion and hyper-parameter tuning. In this paper, we propose a new covariate shift adaptation method which avoids estimating the weights. The basic idea is to directly work on unlabeled target data, labeled according to the k-nearest neighbors in the source dataset. Our analysis reveals that setting $k=1$ is an optimal choice. This property removes the necessity of tuning the only hyper-parameter $k$ and leads to a running time quasi-linear in the sample size. Our results include sharp rates of convergence for our estimator, with a tight control of the mean square error and explicit constants. In particular, the variance of our estimators has the same rate of convergence as for standard parametric estimation despite their non-parametric nature. The proposed estimator shares similarities with some matching-based treatment effect estimators used, e.g., in biostatistics, econometrics, and epidemiology. Our experiments show that it achieves drastic reduction in the running time with remarkable accuracy.
Fichier principal
Vignette du fichier
2312.09969v2.pdf (1.31 Mo) Télécharger le fichier
Origin Files produced by the author(s)

Dates and versions

hal-04645530 , version 1 (11-07-2024)

Licence

Identifiers

  • HAL Id : hal-04645530 , version 1

Cite

François Portier, Lionel Truquet, Ikko Yamane. Nearest Neighbor Sampling for Covariate Shift Adaptation. 2024. ⟨hal-04645530⟩

Collections

GENES CREST ENSAI
39 View
30 Download

Share

Gmail Mastodon Facebook X LinkedIn More