Table of Contents
Fetching ...

Design Experiments to Compare Multi-armed Bandit Algorithms

Huiling Meng, Ningyuan Chen, Xuefeng Gao

TL;DR

A new analytical framework is developed and three key properties of the resulting estimator are proved: it is unbiased; it requires only $T + o(T)$ user interactions instead of $2T$ for a run of the treatment and control policies, nearly halving the experimental cost when both policies have sub-linear regret; and its variance grows sub-linearly in $T$, whereas the estimator from a na\"ive design has a linearly-growing variance.

Abstract

Online platforms routinely compare multi-armed bandit algorithms, such as UCB and Thompson Sampling, to select the best-performing policy. Unlike standard A/B tests for static treatments, each run of a bandit algorithm over $T$ users produces only one dependent trajectory, because the algorithm's decisions depend on all past interactions. Reliable inference therefore demands many independent restarts of the algorithm, making experimentation costly and delaying deployment decisions. We propose Artificial Replay (AR) as a new experimental design for this problem. AR first runs one policy and records its trajectory. When the second policy is executed, it reuses a recorded reward whenever it selects an action the first policy already took, and queries the real environment only otherwise. We develop a new analytical framework for this design and prove three key properties of the resulting estimator: it is unbiased; it requires only $T + o(T)$ user interactions instead of $2T$ for a run of the treatment and control policies, nearly halving the experimental cost when both policies have sub-linear regret; and its variance grows sub-linearly in $T$, whereas the estimator from a naïve design has a linearly-growing variance. Numerical experiments with UCB, Thompson Sampling, and $ε$-greedy policies confirm these theoretical gains.

Design Experiments to Compare Multi-armed Bandit Algorithms

TL;DR

A new analytical framework is developed and three key properties of the resulting estimator are proved: it is unbiased; it requires only user interactions instead of for a run of the treatment and control policies, nearly halving the experimental cost when both policies have sub-linear regret; and its variance grows sub-linearly in , whereas the estimator from a na\"ive design has a linearly-growing variance.

Abstract

Online platforms routinely compare multi-armed bandit algorithms, such as UCB and Thompson Sampling, to select the best-performing policy. Unlike standard A/B tests for static treatments, each run of a bandit algorithm over users produces only one dependent trajectory, because the algorithm's decisions depend on all past interactions. Reliable inference therefore demands many independent restarts of the algorithm, making experimentation costly and delaying deployment decisions. We propose Artificial Replay (AR) as a new experimental design for this problem. AR first runs one policy and records its trajectory. When the second policy is executed, it reuses a recorded reward whenever it selects an action the first policy already took, and queries the real environment only otherwise. We develop a new analytical framework for this design and prove three key properties of the resulting estimator: it is unbiased; it requires only user interactions instead of for a run of the treatment and control policies, nearly halving the experimental cost when both policies have sub-linear regret; and its variance grows sub-linearly in , whereas the estimator from a naïve design has a linearly-growing variance. Numerical experiments with UCB, Thompson Sampling, and -greedy policies confirm these theoretical gains.
Paper Structure (31 sections, 10 theorems, 78 equations, 12 figures, 2 tables, 2 algorithms)

This paper contains 31 sections, 10 theorems, 78 equations, 12 figures, 2 tables, 2 algorithms.

Key Result

Lemma 1

Fix a horizon $T \geq 1$. For any $i \in \{0, 1\}$, let $\{(A_t^{\pi_i}, R_t^{\pi_i})\}_{t=1}^T$ be the action-reward trajectory defined in eq:A-R-single-conditional-dist under the canonical model of the naı̈ve design, and let $\{(A_t^{\pi_i\text{-s}}, R_t^{\pi_i\text{-s}})\}_{t=1}^T$ be the action-

Figures (12)

  • Figure 1: Walmart search results for "socks". The left panel displays the default "All results" interface, while the right panel displays the filtered view triggered by the "New Arrivals" button. The red boxes highlight the new arrival item appearing in the regular search result.
  • Figure 2: Naı̈ve design for comparing two multi-armed bandit policies.
  • Figure 3: Artificial Replay: run one policy, then replay compatible rewards for the other policy.
  • Figure 4: An illustration of artificial replay with $K=3$ and $T=10$. The top row of each policy is the index of the pulled arm and the bottom row shows the realized reward.
  • Figure 5: An illustration of the sample path of $\pi_0$ in Figure \ref{['fig:AR-arrow-diagram']} using the reward-stack model. The integer at the top left corner indicates the period the reward is used.
  • ...and 7 more figures

Theorems & Definitions (14)

  • Lemma 1
  • Theorem 1
  • Corollary 1
  • Proposition 1
  • Theorem 2: Symmetry
  • Theorem 3
  • Remark 1
  • Theorem 4: Unbiasedness
  • Theorem 5: Asymptotic Variance Reduction
  • Remark 2
  • ...and 4 more