← all papers · overview

Diffusion Models Meet Contextual Bandits

Abstract

Efficient online decision-making in contextual bandits is challenging, as methods without informative priors often suffer from computational or statistical inefficiencies. In this work, we leverage pre-trained diffusion models as expressive priors to capture complex action dependencies and develop a practical algorithm that efficiently approximates posteriors under such priors, enabling both fast updates and sampling. Empirical results demonstrate the effectiveness and versatility of our approach across diverse contextual bandit settings.

Related papers

Ranked by semantic similarity — how closely each paper's abstract matches this one (100% = near-identical topic).