Minimax-Optimality of Posterior Sampling for Reinforcement Learning
Organizations: KAIST
Abstract
Posterior sampling for reinforcement learning (PSRL) is one of the simplest and most effective exploration methods, but a basic question has remained open: does unmodified PSRL achieve minimax regret without structural assumptions on the prior? We answer yes. Exact vanilla PSRL is minimax optimal in leading-order Bayesian regret under arbitrary correlated priors. The difficulty is that a posterior-sampled transition model is coupled with its own continuation value. We overcome this with a common empirical transition reference that isolates the resulting value mismatch and a Bellman-based variance argument that controls it without an extra leading-order state-space factor. For finite-horizon, time-inhomogeneous tabular MDPs with unknown stochastic rewards, this yields the minimax regret rate under arbitrary joint priors over rewards and transitions. The same proof principle gives the minimax rate for linear-mixture MDPs under arbitrary joint parameter priors.
Figures & tables
| Work | Time | Prior | Rewards | Bayesian regret | Comparison |
| Tabular MDPs | |||||
| Osband et al. [20] | Homog. | Arbitrary | Unknown stochastic | gap | |
| Osband and Van Roy [22] | Homog. | Independent Dirichlet rows | Independent reward prior; sub-Gaussian noise | gap | |
| Moradipari et al. [19] | Inhomog. | Independent stages/components; consistency and occupancy assumptions | Unknown; independent of transitions | worst-case gap prior-dependent | |
| This work | Inhomog. | Arbitrary joint | Unknown stochastic | minimax | |
| Linear-mixture MDPs | |||||
Appendix figures & tables1 asset
Supplementary material from the paper’s appendix.