Original title: The Dually Flat Geometry of Planning as Inference
Nikola Milosevic, Asaki Kataoka, Nicolas Hinrichs, Kenji Doya +1 · September 2026
In reinforcement learning, computers learn good decisions by trial and error, but tracking how often an agent visits each situation is hard. Researchers tackled this by building a new mathematical setup called a resetting planning process. It embeds the decision-making goal directly into the agent's movement dynamics. Its steady pattern, which they call the visitation measure, creates a smooth geometric landscape called a dually flat statistical manifoldA geometric structure representing the space of possible plans with two linked ways to measure distances.. Here, visitation probabilities and decision strategies connect neatly through conditional entropy. This geometric perspective expands planning-as-inference so it handles complex, nonlinear goals instead of just simple, linear rewards. Each step requires only a single natural-gradient update. It also reveals that the common temporal-difference errorA signal used in reinforcement learning to update estimates of future rewards based on current experiences. acts as a marginal-utility estimate, aiding reinforcement learning and theoretical neuroscience.
You're organizing a community picnic where the fun depends on how many people show up, but the turnout itself depends on what games you plan. If you plan games assuming a fixed number of guests, you might end up with too few or too many, ruining the day. In this work, the picnic planning is like AI decision-making, where rewards are nonlinearly tied to choices. The geometry helps adjust plans dynamically, so each game choice considers its effect on attendance.
This work affects AI researchers and neuroscientists by allowing planning systems to handle more realistic, nonlinear goals. It provides a unified geometric framework that could improve AI algorithms and insights into brain function.
The abstract does not specify whether this framework has been experimentally validated or implemented in practical systems. It focuses on theoretical development, so computational efficiency and real-world applicability are not addressed.
Ask anything about this paper. Answers come from its abstract, and say so when the abstract doesn't cover your question.
The Dually Flat Geometry of Planning as Inference — Nikola Milosevic, Asaki Kataoka, Nicolas Hinrichs et al.
Abstract licence: CC0
Explanation generated by Synthesis from the abstract — it is not the original paper. This is a preprint, so it hasn't necessarily been peer reviewed.