Experimental Design for Regret Minimization in Linear Bandits
Andrew Wagenmaker, Julian Katz-Samuels, Kevin Jamieson
INTRODUCTION
Existing regret minimization algorithms for linear bandits suffer from several important shortcomings. First, they typically rely on naive union bounds, which yield regret guarantees scaling as either or . Such union bounds ignore the geometry present in the problem and, as such, can be very wasteful. As the union bound often appears in the confidence interval within the algorithm, this is not simply an analysis issue—it can also affect real performance. Second, in the moderate, non-asymptotic time regime, existing algorithms tend to rely on the principle of optimism—pulling only the arms they believe may be optimal. Algorithms relying on this principle are very myopic, foregoing initial exploration which could lead to better long-term reward and instead focusing on obtaining short-term reward, leading to suboptimal long-term performance. This is a well-known effect in the bandit setting but, as we show, is also present in the semi-bandit setting.
In this paper, we develop an algorithm overcoming both of these shortcomings. Rather than employing a naive union bound, we appeal to tools from empirical process theory for controlling the suprema of a Gaussian process, allowing us to obtain confidence bounds that are geometry-dependent and potentially much tighter. In addition, our algorithm relies on careful planning to balance the exploration-exploitation tradeoff, taking into account both the potential information gain as well as the reward obtained when pulling an arm. This planning allows us to collect sufficient information for good long-term performance without incurring too much initial regret and, to the best of our knowledge, is the first planning-based algorithm in the linear bandit setting that provides finite-time guarantees.
We emphasize that we are interested in the non-asymptotic regime and aim to optimize the whole regret bound, including lower-order terms. While several recent works achieve instance-optimal regret, they suffer from loose lower-order terms which dominate the regret for small to moderate . Our results aim to minimize such terms through employing tighter union bounds. We summarize our contributions:
We develop a single, general algorithm that achieves a state-of-the-art finite-time regret bound in stochastic linear bandits, in combinatorial bandits with bandit feedback, and in combinatorial bandits with semi-bandit feedback. In addition, our framework is general enough to extend to settings as diverse as partial monitoring and graph bandits.
We show that in the combinatorial semi-bandit regime, our algorithm is computationally efficient, relying only on calls to a linear maximization oracle, and state-of-the-art, yielding a significant improvement on existing works in the non-asymptotic time horizon regime.
We give the first example for combinatorial bandits with semi-bandit feedback that shows that optimistic strategies such as UCB and Thompson Sampling can do arbitrarily worse than the asymptotic lower bound, and show that our algorithm improves on optimism in this setting by an arbitrarily large factor.
As a corollary, we obtain the first computationally efficient algorithm for pure exploration in combinatorial bandits with semi-bandit feedback, and achieve a state-of-the-art sample complexity.
This work can be seen as obtaining problem-dependent minimax bounds—minimax bounds that depend on the arm set but hold for all values of the reward vector—and are similar in spirit to the bounds on regret minimization in MDPs given by Zanette and Brunskill (2019). For some favorable arm sets , our bounds are tighter than prior -independent minimax bounds by large dimension factors. To the best of our knowledge, we are the first to obtain such geometry-dependent minimax bounds for linear bandits.
PRELIMINARIES
Throughout, we assume that . We consider two observation models: semi-bandit feedback and bandit feedback. In the bandit feedback setting, at every timestep we observe:
where . In the semi-bandit feedback setting, we assume that our bandit instance is combinatorial, , and at every timestep we observe:
where . Note that, while we assume Gaussian noise for simplicity, all our results will hold with sub-Gaussian noise (Katz-Samuels et al., 2020).
In the bandit setting, after observations, our estimate of will be the standard least squares estimate:
In the semi-bandit setting, we will estimate coordinate-wise, forming the estimate:
where is the number of times . We denote:
In the combinatorial setting, can often be exponentially large in the dimension, making computational efficiency non-trivial since cannot be efficiently enumerated. As such, much of the literature on combinatorial bandits has focused on obtaining algorithms that rely only on an argmax oracle:
Efficient argmax oracles are available in many settings, for instance finding the minimum weighted matching in a bipartite graph and finding the shortest path in a directed acyclic graph.
MOTIVATING EXAMPLES
Before presenting our algorithm and main results, we present several examples that motivate the necessity of planning and the wastefulness of naive union bounds, and illustrate how our algorithm is able to make improvements in both these aspects.
and Algorithm 1 has expected regret bounded as, for any :
Thus, treating as a constant, the asymptotic regret of the generic optimistic algorithm is loose by a square root dimension factor, and Algorithm 1 in the current paper improves over optimism by an arbitrarily large factor. As it also relies on the principle of optimism, albeit in a randomized fashion, Thompson Sampling will be suboptimal by this same factor on this instance. A similar instance can also be found in the bandit feedback setting. The improvement in Algorithm 1 is due to its ability to pull informative but suboptimal arms if the information gain outweighs the regret incurred, reducing the cumulative regret. Optimistic algorithms, in contrast, will only pull arms they believe may be optimal, and so do not effectively take into account the information gain which, in some cases, causes them to be very suboptimal.
To illustrate the improvement we gain by applying a less naive union bound, we will consider the following combinatorial class:
while algorithms employing naive union bounds will achieve regret bounds scaling at best as:
In the appendix we discuss in more detail how the regret scales for specific algorithms in this setting. The regret bound we present for our algorithm in Proposition 2 is in fact state-of-the-art—all other existing algorithms will incur the larger dimension dependence.
EXPERIMENTAL DESIGN FOR REGRET MINIMIZATION
Intuitively, is the largest Gaussian width of any subset of formed by taking all with gap bounded by . The following results are helpful in giving some sense of the scaling of .
Note that these upper bounds are often loose. The following results shows that, in some cases, we pay a instead of .
The Gaussian width is critical in avoiding wasteful union bounds, allowing instead for geometry-dependent confidence intervals. The following confidence interval will form a key piece in our analysis.
2 Algorithm Overview
We next present our algorithm, RegretMED, in Algorithm 1. Inspired by several recent algorithms achieving asymptotically optimal regret (Lattimore and Szepesvari, 2017), at every epoch our algorithm finds a new allocation by solving an experimental design problem (2). This minimizes an upper bound on the regret incurred in the epoch while ensuring the allocation produced will explore enough to improve the estimates of the gaps for each arm, thereby balancing exploration and exploitation and allowing us to obtain a tight bound on finite-time regret. We apply the TIS inequality to bound the estimation error of our gaps, which motivates the constraint in (2). Critically, this yields a regret bound scaling with the Gaussian width of the action set.
Key Theoretical Tools: We briefly describe the key theoretical tools employed by RegretMED. First, we note that an experimental design based algorithm is novel in the setting of regret minimization. As we have shown, this approach allows us to perform properly on challenging instances by explicitly balancing the information gain and reward, while also yielding a computationally feasible solution in the semi-bandit regime. Our second innovation is the use of the TIS inequality to obtain tight concentration bounds. While we are not the first to utilize this in the linear bandit setting (Katz-Samuels et al., 2020), it previously was only utilized in the best arm identification setting, and our work therefore shows how it can be applied in the regret minimization setting as well. The use of the TIS inequality yields two important improvements over more naive union bounds. First, it provides tighter confidence intervals in the non-asymptotic time regime and therefore yields improved regret bounds. Second, as we will see, it allows us to write the constraint for our experiment design problem (2) in a form that is linear in the the decision variable. This allows us to reduce solving the optimization to calls of a linear maximization oracle, and is a key piece in showing our algorithm is computationally efficient.
3 Main Regret Bound
We now state our main regret bound. Define
for absolute constants and .
4 Computationally Efficient Algorithm
While Algorithm 1 can be run in settings where is enumerable, it becomes computationally infeasible for very large , as (2) cannot be solved via a linear maximization oracle. In place of (2), consider instead solving:
As we show in Theorem 4, we can solve this problem with a computationally feasible algorithm in the semi-bandit feedback regime. Running this modified version of Algorithm 1, we obtain the following regret bound.
and the minimax regret will be bounded as:
In the semi-bandit setting, we can apply Theorem 4 to compute an approximate solution to (3) in polynomial time, as described below. See Section 6 and Table 1 for an in-depth discussion of how our result compares to existing works.
5 Pure Exploration with Semi-Bandit Feedback
Although our algorithm is designed to minimize regret, a slight modification gives a computationally efficient algorithm for best arm identification in the semi-bandit feedback setting. In particular, instead of (2) consider solving:
Let . Run Algorithm 1 but replace (2) with (4) and omit the break on line 7. Invoke Theorem 4 to efficiently find an approximate solution to (4). Then, with probability , the algorithm will terminate after collecting at most:
samples and we will have .
We state and prove a lower bound for this problem in the appendix, Theorem 6, which shows that this sample complexity is near-optimal. To the best of our knowledge, this is the first general, computationally efficient, and near optimal algorithm for pure exploration with semi-bandit feedback.
6 Optimization
The following result shows that there exists a polynomial-time algorithm that finds an approximately optimal solution, i.e., it is within a constant approximation factor of the optimal solution.
Let opt be the optimal value of (5). There exists an Algorithm that returns such that , , and, with probability at least :
Furthermore, the number linear maximization oracle calls is polynomial in .
which can be solved using only linear maximization oracle calls via the binary search procedure from Katz-Samuels et al. (2020). The proof of this result and full algorithm is given in Section D.
We prove this result and state how this rounded distribution can be computed in Appendix E.
EXPERIMENTAL RESULTS
We next present experimental results for RegretMED in both the semi-bandit and bandit feedback settings. Every point in each plot is the average of 50 trials. The error bars indicate one standard error.
We illustrate the result in Figures 3 and 3 for different values of . In both cases, RegretMED yields a significant improvement over CTS-Gaussian and CombUCB1. Note that is growing exponentially in and for we have . In all experiments we set .
As Figure 3 illustrates, the performance of RegretMED is almost unaffected by the choice of , while the performance of both TS and LinUCB degrades significantly. Optimistic algorithms are suboptimal on this instance as they do not pull the suboptimal but informative arm, . Our results indicate that RegretMED is able to overcome this difficulty by continuing to pull even when it has been determined suboptimal, recognizing the information gain outweighs the regret incurred.
DISCUSSION AND PRIOR ART
While prior algorithms have tended to be based on the principle of optimism (Kveton et al., 2015; Combes et al., 2015; Degenne and Perchet, 2016; Wang and Chen, 2018; Perrault et al., 2020a), we have shown that optimistic strategies are asymptotically suboptimal (see Proposition 1), motivating our planning-based algorithm. Additional work includes (Chen et al., 2016; Talebi and Proutiere, 2016; Perrault et al., 2020b). We summarize our results in Table 1.
Asymptotically Optimal Regret in Linear Bandits: Another related line of work focuses on asymptotic performance (Lattimore and Szepesvari, 2017; Combes et al., 2017; Hao et al., 2020; Degenne et al., 2020; Cuvelier et al., 2020). In the bandit setting asymptotic lower bounds have been shown to scale as:
While we do not claim RegretMED is asymptotically optimal, we note that the optimization we are solving (2) closely resembles the above optimization. Indeed, at the final epoch of RegretMED, our estimates of the gaps will be sufficiently accurate so as to ensure we are playing approximately the asymptotically optimal distribution. Furthermore, as Proposition 1 and Figure 3 show, RegretMED appears to be playing the asymptotically optimal strategy in situations where optimism fails. We leave a rigorous proof of the asymptotic qualities of RegretMED to future work.
Concurrent to this work, several works appeared which simultaneously achieve asymptotically optimal and sub- regret (Tirinzoni et al., 2020; Kirschner et al., 2020b). In particular, Tirinzoni et al. (2020) achieves instance-optimal regret in finite time. We remark that their regret bound contains large additive terms which will dominate the leading term for moderate time horizons. Our primary concern is in this non-asymptotic regime, where the union bound applied is still significant, and we therefore see our work as complementary, addressing issues they do not address.
Asymptotically optimal regret has been relatively unexplored in the semi-bandit setting. Following the acceptance of this work, a very recent work (Cuvelier et al., 2021) proposed a computationally efficient asymptotically optimal algorithm in the semi-bandit setting, which was the first of its kind. As with the bandit setting, our concern is with the non-asymptotic time regime, so this result is complementary to ours.
Stochastic Multi-Armed Bandits with Side Observations: In the stochastic multi-armed bandits with side observations problem, the agent is given a graph of nodes where each node is associated with an independent distribution. When the agent pulls a node , she observes and suffers its stochastic reward and she also observes the stochastic reward of any node with an edge connected to node . Caron et al. (2012) proposed a UCB-like algorithm and Buccapatnam et al. (2014) used a linear programming solution to show that the regret scales with the minimum dominating set.
Pure Exploration in Multi-Armed Bandits: There has not been a significant amount of previous work on pure exploration combinatorial bandits with semi-bandit feedback. Chen et al. (2020) provide a general framework that subsumes combinatorial bandits with semi-bandit feedback but their algorithm is non-adaptive and suboptimal. Several special cases of pure exploration combinatorial bandits with semi-bandit feedback have been studied. Best arm identification (where ) has received much attention (Even-Dar et al., 2006; Jamieson et al., 2014; Karnin et al., 2013; Kaufmann et al., 2016; Chen and Li, 2015). The setting in Jun et al. (2016) subsumes the top-K problem, but their approach does not generalize to other combinatorial problem instances. Concurrent to this work, Jourdan et al. (2021) derived an asymptotically optimal best arm identification algorithm for the semi-bandit setting. We note that our result focuses on optimality in the finite-time regime, so our results our complementary.
Our algorithmic technique bridging empirical process theory and experimental design is inspired by the work on pure exploration combinatorial bandits in Katz-Samuels et al. (2020). The semi-bandit feedback setting in the present paper poses a new and non-trivial computational challenge since, unlike in Katz-Samuels et al. (2020), the number of variables in the optimization is potentially exponential in the dimension.
Acknowledgements
AW is supported by an NSF GFRP Fellowship DGE-1762114. JKS is supported by an Amazon Research Award. The work of KJ is supported in part by grants NSF RI 1907907 and NSF CCF 2007036.
References
Appendix A Action Elimination with Gaussian Width
We first state an algorithm inspired by Lattimore and Szepesvári and prove a regret bound. This algorithm, while naive, incorporates the TIS inequality to obtain regret scaling with the Gaussian width. Furthermore, the analysis is simple and helps aid in the intuition of the proof of our main theorems.
and , so long as . From Katz-Samuels et al. and Allen-Zhu et al. , we know such a rounding procedure exists and can be computed efficiently, and that it suffices to choose .
where .
with probability at least and minimax regret as:
with probability at least . Here are absolute constants.
We can now follow the same argument as Lemma 12 of Katz-Samuels et al. . Take for some and let be the distribution that minimizes:
and the distribution that minimizes:
Let . Then we will have that:
where the last inequality holds by Kiefer-Wolfowitz and Proposition 9. Also:
Optimizing this over gives the final regret of:
and choosing gives the absolute regret bound. ∎
Appendix B Regret Bound Proofs
Let and define the events:
Proposition 6 gives that with probability at least :
Estimation error: Henceforth we assume holds. We proceed by induction to show that the gaps are always well-estimated. First we prove the base case. Let and consider any . Then:
where follows by Proposition 7.5.2 of Vershynin and follows since is a feasible solution to (3).
For the inductive step, assume that, for all :
Consider round and take . There then exists some such that . Then:
where follows by Proposition 7.5.2 of Vershynin , follows since by virtue of the fact that , so , follows since and for any , we will have theta , so , holds by the inductive hypothesis and Lemma 1 of Katz-Samuels et al. and taking to be the estimate of at round , and holds since is a feasible solution to (3). We can perform a similar calculation to get the same thing for , allowing us to conclude that, for all :
Bounding the Round Regret: From the previous section, we know that on the good event all our gaps will be well-estimated. From (6), it follows that the constraint in (3) is tighter than the following constraint:
so any satisfying this inequality is also a feasible solution to (3).
where follows since we will always have:
where uses the fact that for all , , and the last inequality follows as above. We therefore have that:
The first term can be bounded by the regret bounded given in Lemma 2:
is the only non-negative solution. It follows then that:
Finally, by Theorem 4 we can choose , , and we will be able to compute the solution efficiently. ∎
The proof of this result is very similar to the proof of Theorem 2 but we include the points where it differs for the sake of completeness. Unless otherwise noted, all notation is defined as in the proof of Theorem 2.
Estimation error: Henceforth we assume holds. We proceed by induction to show that the gaps are always well-estimated. First we prove the base case. Let and consider any . Then:
where follows by Proposition 7.5.2 of Vershynin and follows since is a feasible solution to (2). For the inductive step, assume that, for all :
Consider round and take . There then exists some such that . Then:
where follows by Proposition 7.5.2 of Vershynin , follows since by virtue of the fact that , so , follows since and for any , we will have theta , so , holds by the inductive hypothesis and Lemma 1 of Katz-Samuels et al. and taking to be the estimate of at round , and holds since is a feasible solution to (3). We can perform a similar calculation to get the same thing for , allowing us to conclude that, for all :
From here the remaining calculations on the gap estimates performed in the proof of Theorem 2 hold almost identically.
Bounding the Round Regret: From (6), it follows that the constraint in (2) is tighter than the following constraint:
so any satisfying this inequality is also a feasible solution to (2).
From here we follow the same pattern as in the proof of Theorem 2. We handle each term in the constraint separately. For the second term, note that we can upper bound:
For the second term, by the Kiefer-Wolfowitz Theorem in the bandit case, and Proposition 9 in the semi-bandit case, we’ll have:
Following the same argument as in Theorem 2, it follows that:
From here the argument follows identically to the proof of Theorem 4, so we omit the remainder of the proof. ∎
We can think of this procedure as a deterministic variant of action elimination. We can bound the regret incurred as:
The results on the rounding procedure follow from Katz-Samuels et al. , Allen-Zhu et al. . ∎
Given a , Algorithm 1 will run for at most:
rounds. Furthermore, regardless of , Algorithm 1 will run for at most:
where follows by Proposition 7.5.2 of Vershynin , follows since for any :
Appendix C Pure Exploration Proofs
For the sake of clarity, we rewrite the pure exploration algorithm (see Algorithm 3).
Theorem (2) shows that we can solve (12) in polynomial-time, but note that it is easier to solve (12) approximately by calling stochastic Frank-Wolfe to solve
and the convergence rate shown in Lemma 5 applies.
The MINGAP subroutine (Algorithm 4), originally provided in Chen et al. , is a computationally scalable method to compute the empirical gap between the empirically best arm and the empirically second best arm. It uses at most calls to the linear maximization oracle.
We note that the correctness and sample complexity proofs are quite similar to the proof of Theorem in Katz-Samuels et al. , but we include it for the sake of completeness. The main contribution of our paper for the pure exploration problem is a computational method to solve (12) even when the number of variables is exponential in the dimension.
Step 1: A good event and well-estimated gaps Using the identical argument to the first two steps of the proof of Theorem 2, we have that with probability at least at every round , for all :
For the remainder of the proof we suppose that this good event holds.
Step 2: Correctness. It is enough to show at round , if , then the Unique returns false. Inspecting Unique, a sufficient condition is to show that . By (13) and (14), we have that
Thus, the sample complexity is upper bounded by
where we used the fact that the rounding procedure can use points in the semi-bandit case. Thus, it suffices to upper bound the second term in the above expression. Fix . Then,
Fix . The first term is bounded as follows.
where we obtained line (16) using exercise 7.6.9 in Vershynin .
where line (18) follows since (13), (14), and Lemma 1 in Katz-Samuels et al. imply that .
In this section, we prove a lower bound for the combinatorial bandit setting with semi-bandit feedback. Fix a model and let denote the distribution of the observations when arm is pulled. In this setting, at each round , is drawn and
We say that an Algorithm is -PAC if for any instance , it returns with the largest mean with probability at least .
Fix an instance such that and is unique. Let be a -PAC algorithm and let be its total number of pulls on . Then,
The proof is quite similar to the proof of Theorem 1 in Fiez et al. .
For simplicity, label and . Define the set of alternative instances . Let denote the random number of times that is pulled during the game. Then, noting that the standard transportation Lemma from Kaufmann et al. easily generalizes to semi-bandit feedback, we have that for any ,
By a standard argument (see for example Theorem 1 Fiez et al. ), this implies that
Let . For each , define
showing that . Note that using the identity for the KL-divergence for a multivariate Gaussian, we have that
Since was arbitrary, we may let , obtaining the result. ∎
Next, we state and prove a lower bound for the non-interactive MLE: it chooses an allocation prior to the game, then observes where , and forms the MLE and outputs . Since the non-interactive MLE may use knowledge of in choosing its allocation and the estimator and recommendation rules are very natural, we view the sample complexity of the non-interactive MLE as a good benchmark to measure the sample complexity of algorithms against. The following lower bound for the non-interactive MLE resembles Theorem 3 in Katz-Samuels et al. .
The proof is quite similar to the proof of Theorem 3 in Katz-Samuels et al. , so we merely sketch it here.
Theorem 3 in Katz-Samuels et al. shows that there exists a universal constant such that if or , the with probability at least , the oracle MLE makes a mistake.
Now, rearranging the above inequality,we have that
Note that the allocation for the semi-bandit problem specifies an allocation for the combinatorial bandit problem and the stochastic process (and non-interactive MLE algorithm) is the same on both problems. Thus, can be interpreted as in the combinatorial bandit protocol for some allocation , and we may apply the proof of Theorem 3 to obtain that with probability at least , the oracle MLE makes a mistake.
Appendix D Computational Complexity Results
Algorithm 5 is the main algorithm (see Theorem 8 for its guarantee); it essentially does a grid search over the time horizon variable, . Note that for a fixed , we have that for all
and thus we can ignore the term . Thus, Algorithm 5 calls Algorithm 6 to solve for a fixed the following optimization problem.
and perform binary search over . To solve each of these convex feasibility programs, we employ the Plotkin-Shmoys-Tardos reduction to online learning and apply Algorithm 7, a multiplicative weights update style algorithm. Lemmas 6 and 7 provide the guarantees for the multiplicative weights update algorithm and for the binary search procedure, respectively.
The Plotkin-Shmoys-Tardos reduction requires a method for solving for arbitrary :
See Lemma 5 for our convergence result on stochastic Frank-Wolfe.
Finally, we note that each of our algorithms uses a global variable tol, which for the theory we set to . We note that scales as and thus a polynomial dependence on results in a polynomial dependence on .
Algorithm 9, originally provided in Katz-Samuels et al. , uses binary search and calls to the linear maximization oracle to compute
D.2 Main Optimization Proofs
Note these are polynomial in . Our algorithms share a global parameter tol; it suffices to set . Define
The following Lemma provides the convergence guarantee for stochastic Frank-Wolfe in the semi-bandit setting (see Algorithm 8).
Furthermore, with probability at least , the number of oracle calls is bounded by
For simplicity, we focus on the case where (the other cases are similar). We write and as abbreviations for and .
Smoothness: for an appropriate choice of
Small deviation with high probability: is chosen sufficiently large to ensure that with probability at least
For the sake of abbreviation, define . By Lemma 13, we have that is twice differentiable and that
where we used Jensen’s inequality and is a universal constant.
Now, by the mean value theorem, there exists such that
where the second inequality follows by Holder’s Inequality. Thus,
For the sake of brevity, we write for the remainder of the proof.
Step 1.2: Small deviation with high probability. Now, we show that is chosen sufficiently large to ensure that with probability at least
we then have that by Lemma 2.6.8 in Vershynin ,
Therefore, since and since for an appropriately chosen universal constant, by a standard sub-Gaussian tail bound (21) follows.
The following Lemma shows that the Multiplicative Weight Update algorithm (Algorithm 7) either finds an approximately feasible solution or if there is no approximately feasible solution, determines infeasibility.
Fix and let . Define
With probability at least , if MW() does not declare infeasibility, then MW() returns and if MW() declares infeasibility, then is infeasible. Furthermore, on the same event, MW() uses at most linear maximization oracle calls.
The algorithm uses the Plotkin-Shmoys-Tardos reduction to online learning and essentially runs the multiplicative weights update algorithm (see Arora et al. ) where there is an expert for each constraint. Define
At each round , the algorithm chooses a distribution, and , over the constraints and the adversary uses the stochastic Frank-Wolfe algorithm to find such that
The reward for expert/constraint 1 is and the reward for expert/constraint 2 is .
Let denote the event that satisfies
uses at most linear maximization oracle calls. Define Further, define the following events
By Lemmas 5 and 8 applied with and the law of total probability, we have that
Now, for the remainder of the proof we assume that occurs.
Suppose that at some round Algorithm 8 returns such that . Then, since implies that
we have that on
Thus, the algorithm correctly declares infeasibility of the convex feasibility program.
where is defined in Algorithm 7. We have that
since , , and we assume that . Furthermore,
for a suitably chosen constant where we used that fact that .
Thus, we have shown (22) and therefore may apply Theorem 9, which implies on that
Now, finally, applying Lemma 7, we have that
This shows that approximately satisfies one of the constraints; showing approximate satisfaction of the other constraint follows by a similar argument. Thus, we conclude that . ∎
The following Lemma shows that Algorithm 6 approximately solves the optimization problem (20).
Fix and let . Let be the value of
then with probability at least Algorithm 6 declares the program infeasible. If
Furthermore, Algorithm 6 uses a number of oracle calls that is upper bounded by .
Algorithm 6 applies Algorithm 7 at most times on a using a predetermined set of values for , which we denote . Define the event
where is defined in Lemma 6. Then, by the union bound, we have that . Suppose occurs for the remainder of the proof.
Then, on the event , we have that the Algorithm 6 declares infeasibility of the program.
Note that for any , we have that
and thus the objective does not depend on and can be dropped from the objective. Using the event , if
is empty, then Algorithm 7 declares the program infeasible; otherwise, Algorithm 7 finds . Then, by a standard binary search argument, the result follows. ∎
The following Theorem establishes that Algorithm 5 approximately solves the main optimization problem (5). It directly implies Theorem 4.
Let . Suppose , . Let opt be the value of
With probability at least , Algorithm 5 returns such that , , and
Furthermore, Algorithm 5 uses a number of oracle calls that is polynomial in
Step 0. Let be the value of
and let be the value of
then declares the program infeasible and if
then returns that satisfies
Further, define . By Lemma 7 and a union bound, we have that . We suppose holds for the rest of the proof.
Step 1. First, we show that Algorithm 5 returns such that
By assumption the optimization problem in (5) is feasible and, hence, and thus by the event , the algorithm finds at least one nearly feasible solution, i.e., is not False for all . Let attain the optimal value in the optimization problem (5). Let such that . By event finds such that
Algorithm 5 outputs , which satisfies by Lemma 7 and by construction,
where in the last line we used , which bounds the objective value of .
Next, we show feasiblity of . Observe that
where we used the fact that and .
Step 2: Relate to . Next, we show that
Recall that we let attain the optimal value in the optimization problem (5). Let such that . Note that
Step 3: Relate to opt. Next, we show that
where in the last line we used .
Step 4: Putting it together. Putting together (26), (25), (27), and (28), we have that Algorithm 5 returns such that , , and
D.3 Miscellaneous Optimization Lemmas
and the number of linear maximization oracle calls is bounded above by
The estimation results by applying a standard subGaussian tail bound. The bound on the number of oracle calls follows since Algorithm 9 is applied times and by Lemma 9 and a union bound. ∎
The following Lemma shows that the binary search procedure in Algorithm 9 is efficient with very high probability and it follows immediately from the proof of Lemma 2 of Katz-Samuels et al. .
Draw and consider the optimization problem
Next, we describe a result on the multiplicative weights update algorithm that follows immediately from Corollary 4 in Arora et al. . Consider the experts problem. The set of events is denoted by . Suppose there are experts. At each round , the agent picks an expert and the adversary picks an outcome and the agent obtains reward . The multiplicative weights update algorithm mains a distribution over the experts and chooses an expert randomly from (see Arora et al. for details on how this distribution is chosen). The adversary may have knowledge of the when choosing . The following provides a lower bound on the expected reward obtained by the multiplicative weights update algorithm.
Let denote an error parameter. Suppose there are experts and . If the multiplicative weights algorithm sets the learning rate as , after , then the multiplicative weights algorithm achieves the following bound on its average expected reward: for any expert ,
D.4 Convergence Lemmas
The objective in semi-feedback is convex (by a similar argument to the proof in Katz-Samuels et al. ).
Fix and . By matrix convexity,
Furthermore, since the above matrices are diagonal,
Then, by Sudakov-Fernique inequality (Theorem 7.2.11 in Vershynin ),
Next, we turn to analyzing stochastic Frank-Wolfe. Although a convergence result for stochastic frank wolfe is provided in Hazan and Luo , our setup is slightly different, so we include a convergence analysis for our setting for the sake of completeness. The proof is quite similar to the proof in Hazan and Luo .
Suppose that . Suppose that is convex, , and in Algorithm 11 is chosen such that with probability at least
where . Then, with probability at least ,
The proof follows closely the analysis of SFW in Hazan and Luo but uses smoothness wrt . We have that
where line (29) uses smoothness (Lemma 11), line (30) uses the optimality of , and line (31) uses the definition of the dual norm. Now, define the event
By hypothesis, is chosen such that with probability at least , . Therefore, we have that
The proof is concluded by simple induction. ∎
This follows by a straightforward case by case analysis. ∎
The following is standard smoothness Lemma from convex optimization.
D.5 Differentiability Lemmas
In this section, we show that is twice-differentiable wrt . We set for simplicity and write instead of for the sake of brevity. The following Lemma shows that is differentiable wrt .
The calculation of follows by the chain rule.
Step 1: First, we show that is Lipschitz with an absolutely integrable Lipschitz constant. Define
Since and the maximum of -Lipschitz functions is -Lipschitz, we have that
Step 2: Now, we show that the partial derivatives exist. Define the event
is distinct, if , then with probability holds and is differentiable at .
so the maximizer will be unique. As this is true for all , it follows that . An identical argument implies , so . Then, by the dominated convergence theorem,
The following Lemma shows that is twice-differentiable wrt .
Thus, we may apply the dominating convergence theorem to obtain
for some constant . Therefore, by the bounded convergence theorem for limits, we have that
However, , so the above implies:
By construction, we have , which proves the result.
and . Thus, by the dominating convergence theorem,
Step 4. Putting together (34), (35), and (36), we have shown that
Thus, we have that that the second order partial derivatives exist and derived an expression for them. Showing that the second order partial derivatives are continuous proceeds as in the proof of Lemma 12 (apply the dominating convergence theorem).
Appendix E Rounding
This is a standard result in convex geometry, see for instance Eggleston . ∎
Given any , in the bandit setting, there exists a distribution that is -sparse and:
In the semi-bandit setting, when , there exists a distribution that is -sparse and:
Given some allocation , let the corresponding distribution, and (so ).
Since we only care about the sparsity of , consider fixed. Then, given a solution to (2) or (3), the value of the constraint and objective the solution achieves achieves are fully specified by and . To see the latter, note that . Lemma 14 then implies that there exists a distribution that is -sparse in the bandit case and -sparse in the semi-bandit case that achieves the same value of the constraint and objective of (2) or (3).
Appendix F Gaussian Width Results
This proof closely mirrors the proof of Theorem 21.1 of Lattimore and Szepesvári .
Note also that, by the identity above, for any :
Choosing to be the distribution putting all its mass on , we have:
To see the equality, note that the above implies:
Let for some fixed . Therefore, . Define
We begin by bounding the first term. Notice that since for some fixed and thus if , then . Furthermore, the span of the vectors in has dimension at most since for any , for all , we have that
Thus, by the Kiefer-Wolfowitz Theorem Lattimore and Szepesvári :
To lower bound , note that:
For the regret bound of competing algorithms, LinUCB will scale as . Given the above lower bound on , the regret of action elimination will scale as . In the semi-bandit setting, Kveton et al. obtain a regret bound of and, ignoring logarithmic terms, Degenne and Perchet obtain the same bound. Other existing works [Combes et al., 2015, Perrault et al., 2020a] do not state minimax bounds but, using the standard analysis to obtain a minimax bound from a gap-dependent bound, their regret will also scale as . Note that in this comparison we have ignored terms and have taken the dominate term to be the term with leading dependence that hits the . ∎
Taking the infimum over , in the bandit feedback case Kiefer-Wolfowitz gives , and in the semi-bandit case, Proposition 9 gives the same result. Since was chosen arbitrarily, it follows that .
For the second bound, Exercise 7.5.10 of Vershynin gives that:
from which the result follows immediately. ∎
If and , then at most contains all subsets of size and less so:
where the last inequality follows since the gaussian complexity is within a constant of the Gaussian width when the set contains 0, by Exercise 7.6.9 of Vershynin . The result then follows by choosing . ∎
The proof in the bandit setting is identical to the proof given in Katz-Samuels et al. and we therefore omit it.
Appendix G Lower Bound for Semi-Bandit Feedback and Optimistic Strategies
A policy is consistent if for all and , . Let denote the number of times that is pulled and the number of times that is pulled.
We use a similar argument to the proof of Theorem 1 in Lattimore and Szepesvari . We construct an alternative instance to obtain an asymptotic lower bound. Let denote the probability measure of the associated instance (which we will specify shortly). We note that the Divergence Lemma (Lemma 15.1 Lattimore and Szepesvári ) is easily adapted to the semi-bandit feedback setting. Thus, by a standard argument that applies the Divergence Lemma and the Bretagnolle–Huber inequality (Theorem 14.2 in Lattimore and Szepesvári ), we have that
Let denote the regret of on the alternative instance . Choose . We have that
Then, inequalities (38) and (39) imply that
Dividing both sides by , we have that
Consistency of the policy implies that
This establishes the first claim in the lower bound. The second claim follows by a similar argument to the argument in Corollary 2 of Lattimore and Szepesvari .
Proof of lower bound for optimism: Define the following problem instance
with . Let for and . Note that if and . Then, the optimization problem in Theorem 12 becomes
Consider the solution is and otherwise. This attains a value of
Now, consider the performance of the generic optimistic algorithm. Let denote the number of times that arm is chosen. Define the event
Suppose holds. Now, suppose that . Then,
for all , which together with (40) implies that
is sufficient. Since this is a feasible solution, we’ll then have that:
where the last inequality holds since . Ignoring factors that do not involve , and noting that there are at most rounds, the total regret is bounded as:
Choosing completes the proof.
Failure of Thompson Sampling for semi-bandit feedback: We now provide a sketch as to why Thompson sampling fails on the instance in Proposition 1. Intuitively, Thompson Sampling is optimistic in a randomized fashion, so we would expect it to fail in the same way as optimistic algorithms. Slightly more formally, consider a typical version of Thompson sampling where at each round , where is the arm chosen at time and . Note that with high probability, we will have that:
so we will essentially only pull an arm when. In the case of , we will have:
where are the total pulls of . Since , the above inequality reduces to:
so arm will only be pulled a logarithmic number of times in , which, as with optimism, is not sufficient to achieve optimal regret.
Appendix H Additional Experimental Results
We remark that, when running RegretMED, we do not use the exact constants specified in the algorithm. These constants are likely somewhat loose due to looseness in our analysis. In addition, we do not run the computationally efficient procedure derived formally but instead found that a much simpler heuristic—running stochastic Frank-Wolfe on the Lagrangian relaxation—works well in practice. We also do not use the precise value of , and instead use an upper bound that can be computed using only knowledge of the arms.
The algorithms we compare against do not contain significant hyperparameters, and we choose reasonable values for the parameters they do require. In particular, for LinUCB, we use the regularization .