MDP Planning as Policy Inference
Not on the map. Why?
Abstract (arXiv)
We formulate episodic Markov decision process (MDP) planning as Bayesian inference over policies. The primary contribution is conceptual: the policy itself is treated as the latent variable, and expected return defines an unnormalized posterior density over policies. This preserves the standard expected-return objective, in contrast to trajectory-centric planning-as-inference formulations that introduce auxiliary optimality variables and to entropy-regularized policy optimization methods that solve a different objective. In the exact formulation, the posterior over deterministic policies induces what we define here as an optimal stochastic policy under preference uncertainty, namely the stochastic policy induced by that posterior. For discrete MDPs with stochastic transitions, we study variational sequential Monte Carlo (VSMC) as one approximate inference method for this posterior, introducing policy consistency under state revisitation and coupled transition randomness across particles. Experiments on grid worlds, Blackjack, Triangle Tireworld, and Academic Advising examine the consequences of inference over policies and compare its induced behavior with entropy-regularized policy optimization. The results support the view that MDP planning can be naturally cast as Bayesian inference over policies.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 17 of its 35 references that are in the map. First posted on arXiv 19 Feb 2026.
Builds on
17 of 35 references.
References · 17
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,635
- 2016Variational Inference: A Review for Statisticianscited by 1,466
- 2017Reinforcement Learning with Deep Energy-Based Policiescited by 659
- 2018Reinforcement Learning and Control as Probabilistic Inference: Tutorial and Reviewcited by 466
- 2015Importance Weighted Autoencoderscited by 454
- 2019Monte Carlo Gradient Estimation in Machine Learningcited by 281
- 2017Equivalence Between Policy Gradients and Soft Q-Learningcited by 172
- 2019Soft Actor-Critic for Discrete Action Settingscited by 161
- 2015Gradient Estimation Using Stochastic Computation Graphscited by 120
- 2017Variational Sequential Monte Carlocited by 93
- 2017Filtering Variational Objectivescited by 91
- 2018Tighter Variational Bounds are Not Necessarily Bettercited by 76
5 more
- 2019Stochastic gradient Markov chain Monte Carlocited by 56
- 2018Yes, but Did It Work?: Evaluating Variational Inferencecited by 52
- 2017Reinterpreting Importance-Weighted Autoencoderscited by 36
- 2023Soft Robust MDPs and Risk-Sensitive MDPs: Equivalence, Policy Gradient, and Sample Complexitycited by 16
- 2015Black-Box Policy Search with Probabilistic Programscited by 4
Built on by
Not cited yet.