cs.LGFeb 19, 2026

MDP Planning as Policy Inference

Authors: David Tolpin

Organizations: Offtopia

Abstract

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.

Figures & tables

Explore similar work

CardsList
  1. Expected Free Energy-based Planning as Variational Inference

    Jun 9, 2026Wouter W. L. Nuijten, Thijs van de Laar, Bert de VriesFree Energy CalculationActive Inference

  2. Sophisticated Policies from Epistemic Priors

    Jul 21, 2026Wouter W. L. Nuijten, Bert de VriesActive InferenceFeedback Control

  3. VariBASed: Variational Bayes-Adaptive Sequential Monte-Carlo Planning for Deep Reinforcement Learning

    Feb 21, 2026Joery A. de Vries, Jinke He, Yaniv Oren +3Bayesian RLSample-Efficient RL