Online Decision Making with High-Dimensional Covariates: Regret Bound of the LASSO BanditResearch Paper
Motivation
Many sequential decisions are personalised: a physician chooses a drug dose for each arriving patient, a platform chooses which offer to show each arriving user. Each decision is made after observing a vector of covariates describing the individual, and its outcome is observed only for the option chosen. This is the contextual (covariate) bandit problem, studied in operations research and machine learning since Auer (JMLR 2002) and Goldenshluger and Zeevi (Stochastic Systems 2013).
In medical and e-commerce applications the covariate vector is often high-dimensional: the number of covariates is comparable to or larger than the number of decisions that will ever be made, while the outcome of each option depends on a few of them. Low-dimensional bandit algorithms then incur regret that grows polynomially with . Bastani and Bayati (Operations Research 2020) proposed the LASSO Bandit, which estimates each option's reward model with the LASSO, and proved a regret bound that grows only logarithmically in . The paper evaluates the method on warfarin dosing data.
Timeline:
- 2002–2003: Auer introduces linear-reward contextual bandits with confidence bounds.
- 2013: Goldenshluger and Zeevi give a forced-sampling algorithm for two arms in low dimension with regret under a margin condition and an arm-optimality condition, and an information-theoretic lower bound of the same order.
- 2020: Bastani and Bayati extend the forced-sampling scheme to arms and high-dimensional sparse parameters, with regret .
Setting
There are arms with unknown parameters . At each time a covariate vector arrives; the are i.i.d. with law and take values in a fixed set . If arm is pulled, the reward is , where the noises are independent, -subgaussian ( for all ), and independent of the covariates. A policy chooses the arm from and the past covariates, arms and observed rewards. Its cumulative expected regret is
The sparsity is the smallest integer with for all .
The four assumptions are: (1) on and ; (2) a margin condition ; (3) arm optimality: every arm is either suboptimal by a margin at every covariate, or optimal by margin on a region of probability at least ; (4) a compatibility condition: the conditional second-moment matrix of each optimal arm lies in the set of matrices with whenever .
The LASSO estimator on samples is any minimizer of . The LASSO Bandit forces arm at the prescribed times . At every other time it keeps the arms whose forced-sample estimate is within of the best. Among them it plays the arm with the largest all-sample estimate , trained on every past pull of the arm, with .
Formalization targets
Goal: Theorem 1 (regret of the LASSO Bandit)
For , , , , and ,
with the explicit constants , of the paper (p. 285).
Milestones
- Proposition 1: a LASSO tail inequality for adaptively collected rows with conditionally subgaussian noise.
- Lemma 1: a LASSO tail inequality when a constant fraction of the rows is i.i.d. with a compatible second-moment matrix.
- Proposition 2: the forced-sample estimator of an optimal arm is within of except with probability .
- Proposition 3: the all-sample estimator of an optimal arm is within of except with probability .
Significance
The theorem shows that exploiting sparsity makes the regret depend on the ambient dimension only through , while its dependence on the horizon is within one factor of the lower bound known in low dimension. Proposition 1 is a LASSO oracle inequality for adapted designs, where each row may depend on earlier observations. It applies whenever a LASSO is fitted to data gathered by a feedback policy: adaptive experiments, dynamic pricing, sequential treatment assignment.
The results are proved in the paper and its online appendix; none of them has a machine-checked proof. This mission produces a formal model of the covariate bandit with a non-anticipating algorithm, a formal LASSO for adapted designs, and, when complete, a verified regret bound with every constant explicit. Proposition 1 and Lemma 1 are reusable beyond bandits.
Difficulty
The all-sample estimator is trained on the times at which the algorithm chose an arm, and those choices depend on earlier estimates. Its design rows are therefore neither independent nor identically distributed, and the standard LASSO analysis, which starts from i.i.d. rows and a restricted-eigenvalue bound on their population covariance, does not apply. The forced samples are i.i.d. but only in number, too few for the rate the regret bound needs. Controlling the compatibility constant of the adaptively selected sample covariance, and the martingale noise term, is where the naive argument breaks.
Formalization scope
Arms are Fin K (paper arm is i.val + 1), coordinates Fin d, times are natural numbers from . The model is a structure IsCovariateNoiseModel on a probability space: i.i.d. measurable covariates in a measurable set , independent subgaussian noises (Mathlib's HasSubgaussianMGF with parameter ), noise independent of covariates. Assumptions 1–4 are separate predicates. is Mathlib's sup norm, logarithms are natural, and is the uncentred conditional second moment.
The LASSO minimizer and the arg max need not be unique, so the algorithm takes a selection rule and a tie-breaking rule as parameters, and the theorems hold for all of them. Each round reads only the current covariate, the past covariates, the past arms and their observed rewards. The regret theorem and Proposition 3, whose data set is chosen by the algorithm, require both rules to be measurable. Otherwise the trajectory would not be a random variable, and the expectations in could be integrals of non-measurable functions, which Lean evaluates to and which would make the goal trivially true. For the same reason every assumption constant is required to be positive, and is imposed on the horizon. Only the explicit inequality of Theorem 1 is stated, not the trailing or . Proposition 2 is stated for optimal arms (see its note).
A complete development needs matrix concentration for bounded i.i.d. rows, the Azuma–Hoeffding inequality, and the deterministic LASSO basic inequality under a compatibility condition. Contributions of any of these as standalone lemmas are welcome.
Selected references
- H. Bastani and M. Bayati, Online Decision Making with High-Dimensional Covariates, Operations Research 68(1):276–294, 2020. https://doi.org/10.1287/opre.2019.1902
- A. Goldenshluger and A. Zeevi, A Linear Response Bandit Problem, Stochastic Systems 3(1):230–261, 2013. https://doi.org/10.1287/11-SSY032
- P. Auer, Using Confidence Bounds for Exploitation-Exploration Trade-offs, Journal of Machine Learning Research 3:397–422, 2002. https://www.jmlr.org/papers/v3/auer02a.html
- P. Bühlmann and S. van de Geer, Statistics for High-Dimensional Data, Springer, 2011. https://doi.org/10.1007/978-3-642-20192-9