Stochastic bandits robust to adversarial corruptions
Thodoris Lykouris, Vahab Mirrokni, Renato Paes Leme
Introduction
In online learning with bandit feedback, a learner needs to decide at each time between alternative actions or arms of unknown quality, facing a trade-off between exploiting profitable past actions or exploring new actions about which she has little information. Bandit problems are typically classified according to how the rewards are generated. In stochastic bandits, rewards are drawn from fixed but unknown distributions, which models settings where the alternatives follow particular patterns and do not react to the learner. The other extreme is adversarial bandits, which are robust to rewards that are specifically designed to trick the learner, as in game-theoretic settings.
In this paper, we focus on settings where the overall behavior is essentially stochastic but a small fraction of the rewards can be adversarially changed. Classic stochastic bandit algorithms, like Upper Confidence Bound (UCB) [ACBF02] or Active Arm Elimination (AAE) [EMM06], base most of their decisions on a few observations made in an initial phase of the algorithm and therefore can be easily tricked into incurring linear regret if very few arms are corrupted. Adversarial bandit algorithms like EXP3 are not fooled by such tricks, but cannot exploit the fact that the input is mostly stochastic.
Our goal is to robustify the stochastic setting by designing algorithms that can tolerate corruptions and still be able to exploit the stochastic nature of the input. The algorithms we design are agnostic to the corruption, i.e. they can tolerate any level of corruption, and the guarantee degrades gracefully as more corruption is added. Moreover, we prove lower bounds showing that our results are tight up to a logarithmic factor. Before we explain our technical contribution in detail, we describe examples of settings we have in mind.
In pay-per-click online advertising, the platform selects for each pageview an ad to display and obtains a certain reward if the user clicks on the ad. The click probabilities are unknown. The tension between repeatedly displaying a particular profitable ad that provides reliable revenue and exploring other potentially more rewarding options is a major application of stochastic bandits in the ads industry.
If it weren’t for a phenomenon known as click fraud, this would be a textbook example of stochastic bandits. In click fraud, botnets maliciously simulate users clicking on an ad to trick learning algorithms. One example is a bot consistently making searches to trigger some ad and not clicking on it to make it seem like a certain ad has very low click-through-rate in order to boost its competitor.
Recommendation systems:
A platform recommending activities or services to a user faces the same trade-off. Suggesting new restaurants leads to faster learning of the best spots but may result to dissatisfaction of the customers who are led to disappointing experiences. While most inputs follow a stochastic pattern, some inputs are typically corrupted: either maliciously, e.g. fake reviews by competitors, or non-maliciously, e.g. construction next-door makes the restaurant less desirable in certain interval. This corruption may again exhibit arbitrary patterns and is not identically distributed over time, yet it is dwarfed by the fact that most of the input is stochastic.
There are several other such examples: emails mostly follow a stochastic pattern except a fraction of them which are spam and are designed to trick algorithms. Internet searches follow a predictable pattern except certain spikes caused by unpredictable events. Data collection used in the econometric process often suffers from errors that affect a small part of the input. In all those cases, the vast majority of the input follows a predictable pattern, but a fraction of the samples are corrupted.
1 Our contribution
Our model. In this paper, we introduce a new model of stochastic bandits with adversarial corruptions. The goal of this model is to encourage the design of bandit algorithms that (i) work well in mixed adversarial and stochastic models, and (ii) whose performance deteriorates gracefully as we move from fully stochastic to fully adversarial models.
In this model there are arms, each associated with a fixed reward distribution . At each round , a random reward is drawn and an adversary can change the reward to , possibly using information about the realizations of from both the current and previous rounds as well as the probability that the learner puts on each arm. The learner then draws an arm and obtains both as reward and feedback. We say that the adversary is -corrupted if in every sample path we have .
Our results. The main result (Theorem 3 in Section 3) is a learning algorithm we term Multi-layer Active Arm Elimination Race that with probability has regret
where is the gap of arm , i.e. the difference in stochastic means of arm and the optimal arm . For arms with very small gap, i.e. when , the inverse dependence on the gap can be replaced by . It is possible to improve the bound by a log factor for pseudo-regret, i.e. maximum expected regret against any fixed arm, obtaining: . Two important features of the algorithm are that the guarantee is:
Agnostic: The algorithm does not need to know the corruption level . The guarantee is provided with respect to how much corruption was added in retrospect. If the corruption level is known, we can remove the dependence on as shown in Theorem 1.
High Probability: Our bounds hold with high probability which is important for practical applications as the ones described above. In contrast, the weaker definition of pseudo-regret often hides events with large regret that are offset by events with large negative regret.
The stochastic case corresponds to in which case we recover a bound that is slightly worse than the guarantee provided by UCB. Our algorithm obtains with probability , while UCB obtains this bound without the term.
En route to the result, in Theorem 2 we show an algorithm that, for any fixed known , provides regret for stochastic input and if it is -corrupted. In other words, if we only need to tolerate either a known level or zero corruptions, we save a logarithmic factor from the bound, and match the bound provided by UCB in the stochastic case.
Another question is whether the linear dependence on the corruption level is tight. In Section 4, we show that it cannot be improved upon without decay in the stochastic guarantee (i.e. while still guaranteeing logarithmic regret when the input is stochastic). The lower bound is an adaptation from the adversarial to the corrupted setting of a result from Auer and Chiang [AC16]. This holds even for the case where the corruptions are either or a known level (where our algorithm provides a matching upper bound). We prove in Theorem 4 that an algorithm with pseudo-regret in the stochastic setting () then for every constant , there is a -corrupted instance where the algorithm incurs regret with constant probability.
Our algorithm can also be viewed through the lens of the best of both worlds literature [BS12, SS14, AC16, SL17], where the goal is to design algorithms that simultaneously provide logarithmic regret guarantees in the stochastic regime and square-root guarantees in the adversarial. In Section 5, we sketch how our algorithm can be appropriately modified to obtain, for any constant , pseudo-regret for and pseudo-regret otherwise. We observe that the results in the best of both worlds literature correspond to the case where . We note that such bounds are obtained for pseudo-regret and not regret with high-probability.
Our techniques. The starting point of our design are classical stochastic bandit learning algorithms like UCB and Active Arm Elimination. Such algorithms are very susceptible to corruptions since they base most of their decisions on a small initial exploration phase. Therefore, with a small number of corruptions it is possible to completely trick the algorithm into eliminating the optimal arm.
We address this issue by robustifying them using a multi-layer approach. The learning algorithm consists of multiple layers running in parallel. The layers have decreasing speed and increasing tolerance to corruption. The first layer finishes very fast selecting an arm as optimal, but provides no tolerance to corruption. Subsequent layers are more robust but also slower.
The resulting algorithm is a race between different layers for picking the optimal arm. Once the fastest layer finishes, it provides a first crude estimate of the optimal arm. Once slower layers finish, we obtain finer and finer estimates of the optimal arm.
Our second main idea is that we can obtain more robust algorithms by subsampling. If a layer is only selected with probability , it only receives in expectation a -fraction of the corruption injected by the adversary. If is low enough, the layer behaves almost as if it was stochastic.
Finally, we couple the different layers together by a process of global eliminations. This process enables slower layers to eliminate arms in faster layers. Such a process is necessary for preventing inaccurate layers from pulling suboptimal arms too often.
2 Related work
Online learning with stochastic rewards goes back to the seminal work of Lai and Robbins [LR85]. The case of adversarial rewards was introduced by Auer et al. [ACBFS03]. The reader is referred to the books of Cesa-Bianchi and Lugosi [CBL06], Bubeck and Cesa-Bianchi [BCB12], and Slivkins [Sli17] for an elaborate overview of the area. These two extremes suffer from orthogonal problems; the one is overoptimistic expecting that all rewards come from the same distribution while the other one is too pessimistic in order to be protected against malicious adversaries. Our work addresses the middle ground: rewards come from distributions but are often adversarially corrupted. This is motivated by the non-robustness of stochastic learning algorithms to even small corruption levels.
Closely related to our work lie the works on best of both worlds guarantees [BS12, SS14, AC16, SL17]. These works achieve (up to logarithmic factors) the optimal pseudo-regret guarantee for stochastic rewards and the optimal pseudo-regret or actual regret guarantee for adversarial rewards. Bubeck and Slivkins [BS12] and Auer and Chiang [AC16] begin from a stochastic algorithm and test whether they encounter non-stochastic behavior in which case they switch to adversarial algorithm. In contrast, Seldin et al. [SS14, SL17] begin from an adversarial algorithm with very optimistic learning rate and adapt it if they encounter such behavior. Recently and independently to this work, Wei and Luo [WL18] provide a best of both worlds result with a small-loss pseudo-regret guarantee on the adversarial setting, via a novel analysis of the log-barrier OMD algorithm of Foster et al. [FLL+16]. Although the aforementioned algorithms are very elegant, their analysis is not robust to inputs that are slightly away from stochastic. Our work bridges this gap by designing algorithms with a more smooth behavior for close-to-stochastic instances.
There have been other works that attempt to provide improved guarantees than the adversarial setting when instances are well behaved. Hazan and Kale [HK09] offer regret guarantees that scale with the variance of the losses instead of the time horizon. This guarantee is meaningful in settings that have a very predictable nature and have usually the same performance such as routing. However they do not address most applications of stochastic bandits. In Click Fraud, for example the rewards come from Bernoulli distributions and the variance of such a distribution is high even if the input is totally stochastic. Another approach is the work of Shamir and Szlak [SS17], who consider an input that is adversarial but random local permutations are applied to obtain a more benign instance. This approach is very relevant in settings like buffering, but is again not applicable to our settings.
On the opposite side, attempting to provide improved guarantees for the stochastic setting or enhancing their range is a very active area of research. For instance, the MOSS algorithm [AB09] of Audibert and Bubeck provides the optimal non-distribution-based upper bound for stochastic bandits while retaining the optimal distribution-based stochastic guarantee. The KL-UCB algorithm of Garivier and Cappé [GC11] provides improved constants in the upper bound of the stochastic guarantee matching the lower bound of Lai and Robbins [LR85] for Bernoulli rewards. The Robust UCB algorithm [BCBL13] extends the results to non-bounded rewards replacing with the weaker assumption of bounded variance. However, all the above results are not robust to corruptions from an adaptive adversary due to their deterministic nature. Since the adversary knows the arm the learner will select, they can always corrupt the optimal arm whenever it is about to be selected and therefore cause the learner to either play it multiple times even if it is suboptimal or decide against playing it even with a small amount of corruption (similarly as in our lower bound).
There is also prior work on incorporating corruptions in online decision making. In the online learning front, there are two such attempts, to the best of our knowledge. In their best of both worlds result, Seldin and Slivkins [SS14] allow for some contamination in the data as long as they are obliviously selected and they do not decrease the gap by more than a factor of . The second work is a recent paper by Gajane et al. [GUK18] who suggest a model of corrupted feedback aiming for differential privacy. Unlike our model, their corruptions are neither adversarial nor adaptive. Both of these works make benign assumptions about the nature of corruption and do not address the main roadblock in the settings we consider: an adversarial saboteur will try to add faulty data in the beginning to change the order between the two arms and, with a minimal corruption, she will achieve this goal. Closer to our model are the works on robust allocation such as online matching with corrupted data [MGZ12, EKM15]; unlike online matching though, in online learning we cannot evaluate the optimum at every round since the algorithm’s decisions affect the information it observes.
Last, learning in the presence of corruptions has recently received great attention in the batch learning setting. For instance, recent works study inference under the presence of adversarially corrupted data [MRT15], designing estimators that are robust to corrupted data [DKK+16], learning in auctions with some faulty data due to econometrics errors [CD17]. Our work suggests a similar framework for the study of online learning that is robust to adversarial corruptions in the more challenging problem of sequential decision making where decisions also affect the information observed.
Model
Corrupted stochastic bandits. We study an online bandit learning setting with arms. Each arm is associated with a distribution with mean . The distributions are assumed to have positive measure only on rewards in $a^{\star}=\arg\max_{a}\mu(a)\Delta(a)=\mu\left(a^{\star}\right)-\mu(a)a^{\star}a^{\star}a\neq a^{\star}\Delta(a)=0$.
We consider an adversary who can corrupt some of the stochastic rewards. The adversary is adaptive, in the sense that the corrupted rewards can be a function of the realization of the stochastic rewards up to that point and of the learner’s choices in previous rounds. More formally, the protocol between learner and adversary, at each round , is as follows:
The learner picks a distribution over the arms.
Stochastic rewards are drawn for each arm: .
The adversary observes the realizations of as well as rewards and choices of the learner in previous steps and returns a corrupted reward .
The learner draws arm and observes .
We refer to as the amount of corruption injected in round . The instance is -corrupted if the total injected corruption is at most for all realizations of the random variables:
Note that the adversary is assumed to be adaptive, in the sense that she has access to all the realizations of random variables for all rounds and the realization of rewards at round but only knows the player’s distribution at round and not the arm . Our guarantees gracefully degrades with the total corruption injected by the adversary.
Regret notions. Regret corresponds to the difference between the reward obtained by the algorithm and the reward of the best arm in hindsight:
The regret is a random variable that depends on the random rewards, the randomness used by the learner, and the randomness of the adversary. We say that a regret bound holds with probability if
where the probability is taken over all the three sources of randomness described.
Finally pseudo-regret is a weaker notion that compares the expected performance of the learner with the arm with the highest expected performance. In other words:
The upper bound: Multi-layer Active Arm Elimination Race
Active arm elimination. The starting point of our design is the Active Arm Elimination algorithm for stochastic bandits [EMM06], which can be viewed as an alternative presentation of the more famous UCB algorithm [ACBF02]. It is based on the following idea: in an initial exploration phase, we pull arms in a round-robin fashion and compute an estimate as the average empirical reward of arm . After pulls of arm , usual concentration arguments establish that with probability at least , the difference of the empirical and actual means is at most . We say that is the confidence interval of arm .
This means in particular that given two arms and , if the difference in empirical means becomes larger than the widths of the confidence intervals, i.e., , then with high probability arm is not optimal. Once this happens, the algorithm eliminates arm by removing it from the round-robin rotation. After both arms and the optimal arm are pulled times, the confidence intervals will be small enough that arm will be eliminated.
Eventually all arms but the optimal are eliminated and we enter what is called the exploitation phase. In this phase we only pull the arm with optimal mean. Before we enter exploitation we pulled each suboptimal arm at most times. Each of those suboptimal pulls incurs regret in expectation which leads to the pseudo-regret bound of . This bound can also be converted to a high probability bound if we replace by .
Arms with small . We note that, for the arms that have , the inverse dependence on the gap may initially seem vacuous; for instance, when there are two optimal arms with the same mean, the upper bound becomes infinite as . However, the inverse dependence on the gap can be replaced by in the case of pseudo-regret and in the case of actual regret (due to variance reasons). For simplicity of exposition, we omit this in the current section but we demonstrate how to perform this replacement in Section 5.
The active arm elimination algorithm is clearly not robust to corruption since by corrupting the first steps, the adversary can cause the algorithm to eliminate the optimal arm. As the algorithm never pulls the suboptimal arms after exploration, it is not able to ever recover. One initial idea to fix this problem is to enlarge the confidence intervals. We can decompose the rewards in two terms where the first term comes from the stochastic reward and the second is the corruption introduced by the adversary. If the total corruption introduced by the adversary is at most , then with width , a similar analysis to above gives us the following regret bound:
If is a valid upper bound for the total corruption then active arm elimination with has regret with probability .
The proof follows the standard analysis of active arm elimination. We first establish that, with high probability the optimal arm is never inactivated (Lemma 3.1) and then upper bound the number of times each suboptimal arm is played (Lemma 3.2). The pseudo-regret guarantee directly follows by multiplying the number of plays for each arm by its gap . For the high-probability guarantee, we need to also show that the regret incurred in the meantime is not much more than the above. We provide proof details about the theorem and lemmas in Appendix A. ∎
With probability at least , arm never becomes inactivated.
With probability at least , all arms become inactivated after plays.
2 Stochastic bandits robust to known corruption
The drawback of the active arm elimination algorithm with enlarged confidence intervals (Theorem 1) is that, even if there are no corruptions, it still incurs a regret proportional to . As a warm up to the main theorem, we provide an algorithm that achieves the usual bound of if the input is purely stochastic and, at the same time, achieves if the input is -corrupted for a known . In the next subsection, we modify the algorithm to make it agnostic to the corruption level .
Two instances of Active Arm Elimination. The first idea is to run two instances of active arm elimination: the first is supposed to select the correct arm if there is no corruption and the second is supposed to select the right arm if there is corruption. The first instance is very fast but it is not robust to corruptions. The second instance is slower but more precise, in the sense that it can tolerate corruptions. Since the second instance is more trustworthy, if the second instance decides to eliminate a certain arm , we eliminate the same arm in the faster instance.
Decrease corruption by sub-sampling. To keep the regret low if the input is stochastic, the second instance of active arm elimination cannot pull a suboptimal arm too many times. Therefore, the technique in Theorem 1 alone is not enough. The main idea of the algorithm is to make arm behave as if it was almost stochastic by running the second instance with low probability. If the learner selects to run the second instance with probability then, when the adversary adds a certain amount of corruption to a certain round, the second instance observes that corruption with probability . Therefore, the expected amount of corruption the learner observes in the second instance is constant. This makes the arms behave almost like stochastic arms in that instance.
Learning algorithm. We obtain our algorithm by combining those ideas. We have two instances of active arm elimination which we denote by (fast) and (slow). Each instance keeps an estimate of the mean and corresponding to the average empirical reward of that arm and also keeps track of how many times each arm was pulled in that instance and . This allows us to define a notion of confidence interval in each of the instances. We define as usual and for the slow instance we define slighly larger confidence intervals: (the reason will be clear in a moment). Also, each instance keeps a set of eliminated arms for that instance: and .
In each round, with probability we make a move in the fast instance: we choose the next active arm in the round robin order, i.e., arm which was played less often, pull this arm and increase and update accordingly. As usual, if there are two active arms and such that we eliminate by adding it to .
With the remaining probability we make a move in the slow instance by executing the exact same procedure as described for the other instance. There is only one difference (which causes the two instances to be coupled): when we inactivate an arm in we also eliminate it in . This leaves us with a potential problem: it is possible that all arms in the instance end up being eliminated. If we reach that point, we play an arbitrary active arm of the slow instance, i.e., any arm .
The resulting algorithm is formally provided in Algorithm 1.
Towards the performance guarantee, Lemma 3.3 bounds the amount of corruption that actually enters the slow active arm elimination algorithm, which enables the regret guarantee in Theorem 2.
In Algorithm 1, the slow active arm elimination algorithm observes, with probability at least , corruption of at most during its exploration phase (when picked with probability ).
If one cared just about the expected corruption that affects , this is at most a constant number since the total corruption is at most and it affects with probability . To prove a high-probability guarantee we require a concentration inequality on martingale differences (since the corruptions can be adaptively selected by the adversary). We provide the details in Appendix B. ∎
Algorithm 1 run with widths and has for the stochastic case and for the -corrupted case with probability at least .
The result for the stochastic case follows standard arguments for stochastic algorithms (since we obtain double the regret of this setting as we run two such algorithms with essentially the same confidence intervals). For the -corrupted case, we establish via Lemma 3.3 an upper bound on the corruption that will affect the slow active arm elimination algorithm . Thanks to the sub-sampling, this upper bound is close to a constant instead of depending on which allows to not incur dependence on in the stochastic case. Having this upper bound, we can apply it to the algorithm of the previous section to get an upper bound on the number of plays of suboptimal arms in . Since the algorithms are coupled, such a bound implies an upper bound on the regret that it can cause in as well. This is because in expectation the arm is played at most times more in as it may be selected every single time in prior to getting eliminated by and is selected times more often than . To obtain the above guarantee with high probability, we lose an extra logarithmic factor. The details of the proof are provided in Appendix B. ∎
3 Stochastic bandits robust to agnostic corruption
Multi-layer active arm elimination race. We now describe our main algorithm in the paper. We call it a race since we view it as multiple layers racing to pick the optimal arm. The less robust layers are faster so they arrive first and we keep choosing (mostly) according to them until more robust but slower layers finish and correct or confirm the current selection of the best arm.
Figure 1 provides an example of the state of the algorithm, which is formally defined in Algorithm 2.
We now provide the main result of the paper, a regret guarantee for Algorithm 2.
The lower bound
For the two arms case where the gap between the arms is , Theorem 2 presents an algorithm which achieves pseudo-regret if the input is stochastic and with probability if the input is at most -corrupted. We show below that this dependence is tight.
Consider a multi-armed bandits algorithm that has the property that for any stochastic input in the two arm setting, it has pseudo-regret bounded by , where . For any , there is a corruption level with and a -corrupted instance such that with constant probability the regret is .
Extensions
In this section, we discuss some extensions that our algorithm can accommodate.
Definition of corruption. We presented all results measuring the corruption as the sum over all rounds of the maximum across arms of the corruption injected by the adversary:
In fact all our results can be improved via using and replacing by for summand . More formally, our main theorem (Theorem 3) becomes:
The proof follows the same arguments since it only compares each arm with . This result is nice since the contribution of each arm to the regret is a function only of its own gap and the corruption injected to it and the one injected to arm . The latter dependence on the corruption on the optimal arm is essential since the main attack we presented to the classical arguments only corrupts arm – the lower bound of the previous section also only adds corruption to .
Dependence on the gap. In Section 3, all our guarantees have an inverse dependence on the gap of all arms . Note that such a guarantee is completely meaningless for arms with a very small gap; for instance, if there exist two optimal arms then there is an arm with which makes the presented bound infinite and therefore vacuous. As we hinted there though, this inverse dependence can be improved for arms with small . Our proofs generally relied on setting an upper bound on the number of times that a suboptimal arm is played and thereby providing an upper bound on the regret they cause.
For arms with , an alternative analysis is to say that, even if they are erroneously selected every single time, we can upper bound the loss in performance they cause. For pseudo-regret, the performance loss if they were selected every single time is . For actual regret, one needs to also take into consideration the variance but, even if they are selected every single time, a Hoeffding bound shows that their total reward is with high probability at most lower than its expectation. As a result, the inverse dependence on in our bound can be replaced by for pseudo-regret and for actual regret.
Moreover, the careful reader may have noticed that in Theorem 1, the dependence can be replaced by a sole dependence on without the gap. However, this does not extend to the subsequent theorems since the dependence on there does not come from the upper bound on the corruption experienced (this is at most due to subsampling). Instead, the dependence on comes from projecting the correct layer (smallest layer robust to corruption) to the previous layers via the number of times it will take to eliminate any suboptimal arm.
Uncorrupted objective. In applications such as spam, the corruptions should not be counted as part of the rewards. Our algorithm provides the same guarantee in the case of uncorrupted rewards (the difference between the performances in the two objectives is at most ). One can also observe that the linear dependence on is still necessary: consider arms with and an adversary that corrupts the first steps making them look identical. The learner has no better option than randomly selecting between the two which gives him a regret of under the uncorrupted objective. We note that, in this setting, the linear dependence is necessary unconditionally of the performance of the algorithm in the stochastic setting.
Towards best of all worlds. In the previous section, we showed that a logarithmic dependence in the stochastic setting comes at the expense of linear dependence on in the -corrupted setting if we focus on actual regret. A very interesting direction is to achieve such an improvement with either a higher power on the logarithm in the stochastic setting or aiming for pseudo-regret instead.
In fact, we can combine our algorithm with the SAPO algorithm of Auer and Chiang [AC16] and achieve a bicriteria guarantee for pseudo-regret. For an specified by the algorithm, we achieve our guarantee if the corruption is and at most otherwise; notice that the case corresponds to the best of both worlds. This is done via running the SAPO algorithm at the level with probability instead of having higher layers. The SAPO algorithm guarantees that the pseudo-regret caused by any particular arm is at most logarithmic if the instance is stochastic and at most if it is adversarial via a beautiful analysis that keeps negative regret of time intervals that have performed well to avoid testing eliminated arms too often. In our setting, if the corruption level is less than , the instance behaves as stochastic causing at most logarithmic regret. Else the instance is corrupted and we can extrapolate the regret in this layer to the whole algorithm as arms that are eliminated in this layer are also eliminated before via global eliminations. Since the regret there is at most and this is multiplied by , this implies a bound of on pseudo-regret.
Acknowledgements The authors would like to thank Sid Banerjee whose lecture notes on stochastic bandits proved very helpful, Andrés Munoz Medina, Karthik Sridharan, and Éva Tardos for useful discussions, Manish Raghavan for suggestions on the writeup, and the anonymous reviewers for the valuable feedback they provided that improved the presentation of the paper.
References
Appendix A Supplementary material on Section 3.1
In this section we provide the proof of Theorem 1. Note that in the lemma statements the width is defined as in the theorem: for any arm .
Lemma 3.1 (restated) With probability at least , arm never becomes eliminated.
The crux of the proof lies in establishing that, with high probability, the upper bound of the confidence interval of never becomes lower than the lower bound of the confidence interval of any other arm and therefore does not become eliminated.
More formally, let and be the empirical mean after samples of the stochastic part of the rewards and the empirical mean after samples of the corrupted rewards respectively. Recall that is the mean of arm . By Hoeffding inequality, for any arm , with probability at least :
We set to establish that this holds for all arms and all time steps (after arm has been played times). As a result, for any arm and any time: and .
Comparing now the actual (corrupted) empirical means, they can be altered by at most absolute corruption . Hence and .
Combining the above inequalities with the fact that the actual mean of is higher than the one of , i.e. , we establish that and therefore arm is not eliminated. Since this holds for all times and arms, the lemma follows. ∎
Lemma 3.2 (restated) With probability at least , all arms become eliminated after plays.
The proof stems from the following observations. By Lemma 3.1, arm is with high probability never eliminated. After rounds, with high probability, the lower confidence interval of arm is above the upper confidence interval of arm . This comes from the fact that, after plays of arm (and also of arm since it is not eliminated), the empirical stochastic mean of is, with high probability, at most below its actual mean and similarly the empirical stochastic mean of arm is at most above its actual mean. Since the corruptions are upper bounded by , they can only contribute to a decrease in the average empirical (corrupted) means by at most which is not enough to circumvent the gap .
More formally, let and denote the empirical means of the stochastic part of the rewards and the corrupted rewards respectively after plays of arm . By the same Hoeffding inequality as in the proof of the previous lemma, with probability at least , it holds that . Therefore, with the same probability, after plays for both arm and : and .
The absolute corruption is at most therefore and . By the choice of , we have . Combining with the above argument, this also implies that the widths are upper bounded by and .
Combining the above with the fact that the actual mean of is higher than the one of , i.e. , we establish
As a result arm becomes eliminated after plays if it is not already eliminated before. ∎
Theorem 1 (restated) If is a valid upper bound for the total corruption then arm elimination with has regret with probability .
The proof follows the classical stochastic bandit argument of measuring the regret caused by each arm as a function of its gap and the number of times it is played as established by Lemma 3.2.
For simplicity of presentation, we first provide the pseudo-regret guarantee. Pseudo-regret compares the expected performance of the algorithm to the expected performance one would have had, had they selected throughout the whole time horizon. The expected performance when one uses is . The loss compared to that every time is used instead is equal to its gap . As a result, the expected contribution to pseudo-regret from suboptimal arm is equal to . Lemma 3.2 establishes that with probability any suboptimal arm is played at most times. Each play of the suboptimal arm causes pseudo-regret of . Multiplying the times by the expected regret per time the guarantee (which equals to the gap) and setting the failure probability to be some inverse polynomial of the time horizon to ensure that the expected regret due to the bad event is at most a constant leads to the pseudo-regret guarantee.
To turn the above into a high-probability guarantee, we need to show that the regret incurred during the steps that we pull arm is not significantly higher than the expectation (therefore bounding the resulting variance). By the Hoeffding inequality of Lemma 3.1, the empirical cumulative reward of arm is, with high probability, at most less than its expectation. The same holds for arm for these steps (its realized performance is at most this much more than its expectation). The probability that these statements do not hold for some arm or some time is at most .
Regarding arms , the term can be upper bounded by by the definition of :
Regarding arm , let be the arm with the smallest gap. By Lemma 3.1, never gets eliminated but it is not necessarily the ex post optimal arm. In fact some other arm with may be the ex post optimal arm (arms with higher gap are with high probability not the ex post optimal arm by an analogous argument as in Lemma 3.2. However, by the same argument as above arm is with high probability at most below its expectation and the ex post optimal arm is at most this much above its expectation. This gives a bound of that is caused by the case where is not the ex post optimal arm.
Therefore the actual regret from times that arm is played is at most where the one term comes from the expectation and the other from the aforementioned bounds on the variance. The corruption can increase any cumulative reward by at most which is already existing in the regret bound. Replacing by Lemma 3.2, we obtain the high-probability guarantee. Note that the failure probabilities of the two lemmas are coupled as they correspond to the same bad events. ∎
Appendix B Supplementary material on Section 3.2
In this section, we provide the proof of Theorem 2. To handle the corruption, we bound with high probability the total corruption experienced by the slow active arm elimination instance (Lemma 3.3). To deal with an adaptive adversary, we need a martingale concentration inequality; specifically we apply a Bernstein-style inequality introduced in [BLL+11] (Lemma B.1).
Lemma 3.3 (restated) In Algorithm 1, the slow active arm elimination algorithm observes, with probability at least , corruption of at most during its exploration phase (when picked with probability ).
The first observation is that the expected corruption encountered by algorithm is at most a constant (total corruption of encountered with probability ). The rest of the proof focuses on bounding the variance of this random variable (actual corruption encountered by the layer). Crucially, since we want to allow the adversary to be adaptive, we should not assume independence across rounds but only conditional independence (conditioned on the history) and this is why some more involved concentration inequality is necessary. Therefore we create a martingale sequence (actual corruption minus expected corruption) and apply a Bernstein-style concentration inequality.
where corresponds to the history up to round . Note that
The last inequality holds as and by the definition of .
A trivial upper bound of is , since the rewards are in $1-\delta$:
Theorem 2 (restated) Algorithm 1 run with widths and has for the stochastic case and for the -corrupted case with probability at least .
The most interesting case is the -corrupted setting. Let be the failure probability in Lemma 3.3. By Lemma 3.3, with probability at least , the actual corruption experienced by the slow active arm elimination algorithm is at most which is less than for non-trivial values of and . Therefore we can apply the analysis of Theorem 1 with corruption level at least and get a handle on the actual regret coming from the slow active arm elimination algorithm.
What is left is to bound the regret coming from the fast active arm elimination algorithm. Towards this goal, we bound the number of times that a suboptimal arm is played in the fast active arm elimination by the expected time that it remains active at the slow active arm elimination. By Lemma 3.2, arm is played in the slow active arm elimination, with probability at least , at most
Having a bound on the number of plays of the arm in the slow active arm elimination instance, we use this to bound the number of plays in the fast active arm elimination instance. In expectation, this is at most times as every move in the slow active arm elimination occurs with probability and, at least of these moves are plays of while it is still active. Since every time arm is played it incurs pseudo-regret , this provides the pseudo-regret guarantee.
To obtain a high probability guarantee, let and observe that with probability at least , we make one move at the slow arm elimination algorithm every moves at the fast arm elimination algorithm. This can be seen by thinking the following process: One tosses coins with bias until she observes heads for the first time (heads is the -biased event). After tosses of the coins the probability that no heads have arrived is at most . To ensure that this is less than , we need to wait , which is achieved by .
By union bound on the failure probabilities for each of those draws, we get that with failure probability (since as it is at most the time horizon), arm gets inactivated in after
The last part is to prove that the regret experienced throughout those rounds is not too large. This follows by the two applications of Hoeffding inequality as before for arms and , analogously to Theorem 1. Combining the above arguments the theorem follows. The total failure probability of the guarantee is . ∎
Appendix C Supplementary material on Section 3.3
Last, we note that, since we used powers of to increase the corruption among layers, the fact that we did not apply the arguments of Theorem 2 with the exact but instead used a such that causes just an extra constant factor on the regret. ∎
Appendix D Supplementary material on Section 4
Theorem 4 (restated) Consider a multi-armed bandits algorithm that has the property that for any stochastic input in the two arm setting, it has pseudo-regret bounded by , where . For any , there is a corruption level with and a -corrupted instance such that with constant probability the regret is .
Step 3: create an adversary that forces a lot of regret in interval . The adversary is quite simple: for the first steps, the arms are Bernoulli with means and for the remaining timesteps, the arms are Bernoulli with means .
Now we establish some concentration on the regret that the learner achieves with respect to arm in the intervals , and . We note that if the learner pulls arm , she does not incur any regret. If she pulls arm , she incurs regret which can be positive or negative. To compute regret with respect to arm in each of those intervals, we sample every time that the arm is pulled.
We then use the Hoeffding bound in the last expression and get:
Step 5c: interval In this interval, pulling arm has again positive expected regret. We use the same technique used in 5a to argue that she cannot obtain large negative regret with high probability: Let be the number of times arm is pulled in that interval and again we abuse notation and let be the difference in rewards in the -th time the arm is pulled. Then:
For this probability is zero, since . Now, for larger , we can use the standard Chernoff bound:
We simply sum the regret of the learner in each of the intervals. For intervals and we can use the bounds computed in steps 5a andn 5c directly. For interval , we note that conditioned on , the learner probes arm a constant number of times, so his total regret differs from the regret by pulling arm in all iterations by at most a constant, therefore the total regret can be bounded by: