Improved Algorithms for Linear Stochastic Bandits
Yasin Abbasi-YadkoriDávid PálCsaba Szepesvari
Develops a novel self-normalized tail inequality for vector-valued martingales to construct tighter confidence sets for linear stochastic bandits, yielding improved regret bounds and substantial empirical performance gains over existing optimism-based algorithms.
In sequential decision-making systems—such as online recommendation engines, dynamic web advertising, and clinical trials—algorithms must balance exploring unknown options against exploiting the best currently known option. The linear stochastic bandit framework models this dilemma by assuming expected rewards depend linearly on chosen actions. Prior algorithms following the optimism-in-the-face-of-uncertainty principle suffered from overly conservative confidence estimates and high computational overhead, largely due to statistical dependencies created across sequential decisions.
The article establishes tighter theoretical confidence bounds and develops practical, high-performance algorithms for both multi-armed and linear stochastic bandits. It evaluates these methods mathematically and validates their performance through simulated empirical experiments.
The authors developed a novel self-normalized tail inequality for vector-valued martingales, which constructs valid confidence sets that hold uniformly across all time steps without requiring conservative union bounds over time. Using this theoretical foundation, they introduced the OFUL (Optimism in the Face of Uncertainty Linear bandit) algorithm along with a computationally efficient variant called Rarely Switching OFUL, which updates its optimization model only when empirical data volume changes by a set factor. The authors evaluated these algorithms across both problem-dependent and worst-case scenarios and benchmarked them against prior state-of-the-art methods in simulation trials.
The key findings are as follows:
- The new martingale tail inequality eliminates the need for standard union bounds across time horizons, significantly tightening the confidence sets surrounding unknown system parameters.
- In empirical linear bandit simulations, the OFUL algorithm achieved an order-of-magnitude reduction in cumulative regret compared to previous baseline algorithms.
- The Rarely Switching OFUL algorithm reduced required computations by approximately 350 times in test experiments while maintaining identical regret performance to the standard OFUL method.
- In standard multi-armed bandit settings, modifying the Upper Confidence Bound algorithm achieved constant regret with high probability, removing the explicit dependence on the total time horizon.
These findings indicate that linear bandit algorithms can be deployed effectively in production environments without the severe computational costs or conservative exploration penalties that hindered earlier approaches. By basing confidence sets on empirical rather than worst-case theoretical quantities, systems can converge significantly faster toward optimal decision strategies, cutting exploration costs and increasing cumulative payoffs.
For practical implementation, engineering and data science teams should adopt Rarely Switching OFUL for high-frequency or resource-constrained environments, tuning the switching threshold to balance runtime overhead against decision accuracy. Further real-world validation in live customer-facing systems is recommended before replacing legacy production models.
The conclusions are subject to standard mathematical assumptions, primarily that noise is sub-Gaussian and true expected rewards are linear. While analytical proofs provide high confidence under these modeled conditions, decision-makers should remain cautious in production contexts characterized by non-linear payoffs or severe non-stationary drift.
- Paper: Finite-time Analysis of the Multiarmed Bandit Problem, Peter Auer et al. (2002). This foundational work establishes finite-time regret bounds for UCB algorithms, which the source directly modifies to achieve constant regret in multi-armed bandits.
- Paper: Using Confidence Bounds for Exploitation-Exploration Trade-offs, P. Auer (2003). This paper introduces upper confidence bound methods for linear associative reinforcement learning and linear bandits, providing the baseline algorithm and framework that the source directly refines and improves.
- Paper: A contextual-bandit approach to personalized news article recommendation, Lihong Li et al. (2010). This work introduces the LinUCB algorithm for linear contextual bandits, serving as a primary contextual bandit method whose theoretical confidence bounds the source explicitly improves upon.
- Paper: An Empirical Evaluation of Thompson Sampling, Olivier Chapelle et al. (2011). This paper evaluates Thompson sampling empirically against linear UCB methods like those refined in the source, offering a key complementary perspective on exploration in contextual bandit tasks.
- Paper: Time-uniform, nonparametric, nonasymptotic confidence sequences, Steven R. Howard et al. (2021). This work develops general time-uniform, nonasymptotic confidence sequences using martingale techniques that substantially generalize and extend the vector-valued martingale tail inequalities introduced in the source.
- Paper: Counterfactual Risk Minimization: Learning from Logged Bandit Feedback, Adith Swaminathan et al. (2015). This work builds on sequential bandit principles to address offline policy optimization from logged bandit feedback using counterfactual risk minimization.
