Dimension Reduction in Contextual Online Learning Via Nonparametric Variable Selection

Wenhao Li,Ningyuan Chen,L. Jeff Hong
2023-01-01
Journal of Machine Learning Research
Abstract:We consider a contextual online learning (multi-armed bandit) problem with high-dimensional covariate x and decision y. The reward function to learn, f (x, y), does not have a particular O similar to(T(dx+dy+1)/(dx+dy+2)), parametric form. The literature has shown that the optimal regret is where dx and dy are the dimensions of x and y, and thus it suffers from the curse of dimensionality. In many applications, only a small subset of variables in the covariate affect the value of f, which is referred to as sparsity in statistics. To take advantage of the sparsity structure of the covariate, we propose a variable selection algorithm called BV-LASSO, which incorporates novel ideas such as binning and voting to apply LASSO to nonparametric settings. Using it as a subroutine, we can achieve the regret where d*x is the effective covariate dimension. The regret matches the optimal regret when the covariate is d*x-dimensional and thus cannot be improved. Our algorithm may serve as a general recipe to achieve dimension reduction via variable selection in nonparametric settings. O similar to(T(d*x+dy+1)/(d*x+dy+2)),
What problem does this paper attempt to address?