English

Leveraging priors on distribution functions for multi-arm bandits

Machine Learning 2025-07-14 v2 Machine Learning

Abstract

We introduce Dirichlet Process Posterior Sampling (DPPS), a Bayesian non-parametric algorithm for multi-arm bandits based on Dirichlet Process (DP) priors. Like Thompson-sampling, DPPS is a probability-matching algorithm, i.e., it plays an arm based on its posterior-probability of being optimal. Instead of assuming a parametric class for the reward generating distribution of each arm, and then putting a prior on the parameters, in DPPS the reward generating distribution is directly modeled using DP priors. DPPS provides a principled approach to incorporate prior belief about the bandit environment, and in the noninformative limit of the DP posteriors (i.e. Bayesian Bootstrap), we recover Non Parametric Thompson Sampling (NPTS), a popular non-parametric bandit algorithm, as a special case of DPPS. We employ stick-breaking representation of the DP priors, and show excellent empirical performance of DPPS in challenging synthetic and real world bandit environments. Finally, using an information-theoretic analysis, we show non-asymptotic optimality of DPPS in the Bayesian regret setup.

Keywords

Cite

@article{arxiv.2503.04518,
  title  = {Leveraging priors on distribution functions for multi-arm bandits},
  author = {Sumit Vashishtha and Odalric-Ambrym Maillard},
  journal= {arXiv preprint arXiv:2503.04518},
  year   = {2025}
}

Comments

Camera ready version -- Reinforcement Learning Journal, 2025

R2 v1 2026-06-28T22:09:20.565Z