What type of inference is planning?
About
Multiple types of inference are available for probabilistic graphical models, e.g., marginal, maximum-a-posteriori, and even marginal maximum-a-posteriori. Which one do researchers mean when they talk about "planning as inference"? There is no consistency in the literature, different types are used, and their ability to do planning is further entangled with specific approximations or additional constraints. In this work we use the variational framework to show that, just like all commonly used types of inference correspond to different weightings of the entropy terms in the variational problem, planning corresponds exactly to a different set of weights. This means that all the tricks of variational inference are readily applicable to planning. We develop an analogue of loopy belief propagation that allows us to perform approximate planning in factored-state Markov decisions processes without incurring intractability due to the exponentially large state space. The variational perspective shows that the previous types of inference for planning are only adequate in environments with low stochasticity, and allows us to characterize each type by its own merits, disentangling the type of inference from the additional approximations that its practical use requires. We validate these results empirically on synthetic MDPs and tasks posed in the International Planning Competition.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Active Inference Planning | Frozen Lake | Success Rate54.5 | 7 | |
| Active Inference Planning | Wumpus World | Success Rate35.2 | 5 | |
| Active Inference Planning | RockSample | Average Reward1.98 | 5 | |
| Active Inference Planning | T-maze 200 episodes | Success Rate52 | 5 | |
| Reactivity Maze | Reactivity Maze (200 episodes) | Mean Reward0.18 | 5 | |
| Sequential Decision Making | MiniGrid DoorKey-8x8 (100 episodes) | Success Rate89 | 3 |