Search and Belief-Space Planning › Planning Under Partial Observability
World Models HandbookAll chapters Search and Belief-Space Planning › Planning Under Partial ObservabilityOne section · Sections in this chapter4 / 13 ↓
Reading tools Definitions & inline links
Choose your expertise level to adjust how many terms are explained. Beginners see more tooltips, experts see fewer to maintain reading flow. Hover over underlined terms for instant definitions.
Article links
Make inline references clickable
Everything in the previous section assumed the planner knows the state . In a maze rendered as a grid of characters, that assumption is invisible because it is satisfied by construction. Many real systems violate it: a robot may have a drifting odometer, a dialogue agent may not know what the user knows, and a recommender may not know what the user has already seen. This section explains why that violation is not a technicality and what replaces the MDP.
The physical state does not become a distribution; it remains the latent state of the environment. What changes is the planner's information state. The planner replaces direct access to with a belief over possible states, evaluates a belief value , and combines prediction with Bayesian conditioning. Under the assumed POMDP model, the belief is a controlled Markov information state, so Bellman reasoning can be applied in belief space.
The POMDP and Why MDP Search Breaks
Recall from Markov and Partially Observable Decision Processes that a partially observable MDP adds an observation space and an observation model to the MDP tuple: . After taking action in state , the environment moves to and emits . Critically, the agent sees but not .
The immediate consequence is that the planner cannot index its decision by one known physical state. Suppose the agent assigns probability to both and , two cells with the same local wall signature. Moving right is on the goal route from but moves toward the dead-end branch from . Planning from an arbitrarily chosen candidate discards both the belief weights and the way future observations can separate the candidates. Its error rate is not generically one half; it depends on the belief, on the rewards, and on the candidate-state action values.
A second consequence is subtler and, for planning purposes, more important. Once actions can change what you know, an action may be valuable partly because of its observations. Turning around in a corridor may earn no immediate reward, and may even cost time, but it may tell you which way is north. A fully observed MDP planner has no information term with which to value that effect; it would choose the same physical motion only for its transition and reward consequences. In a POMDP, later decisions can make information valuable. This is the structural shift we flagged in the introduction: uncertainty can make policies that pay to observe outperform policies that only chase immediate task progress.
The Belief MDP
The standard resolution is to change what the planner treats as a state. Instead of planning over , plan over the space of probability distributions over . This space is called the belief space and is denoted ; an element assigns a probability to each state. You built the machinery for computing from observation histories in Bayesian Filtering and Belief States. Here we use it as the substrate for planning.
The mathematical shape of contrasts sharply with . When is finite with , is a set of discrete points, but is the -dimensional simplex, a continuous space whose dimension is one less than the number of states. The three-state case gives a triangle; the four-state case gives a tetrahedron; the twenty-two-state case, which is our maze, is a twenty-one-dimensional polytope that we can only visualize through projections. Even a small discrete state space therefore induces a continuous belief space, one major source of POMDP planning complexity alongside action-observation branching and horizon.
The key fact is that a correctly updated belief is sufficient for predicting future rewards and observations under future actions. The belief need not preserve the literal history; it preserves the history's decision-relevant probabilistic information under the assumed model. This means the POMDP induces an MDP over , with:
- Belief states: .
- Actions: the same .
- Belief reward: , the expected immediate reward.
- Belief transition: , defined below.
The belief transition is the composition of a prediction step and a Bayesian correction. The prediction step propagates belief through the dynamics:
Here is the current probability of state , is the transition probability after action , and is the predicted probability of successor before seeing the next observation.
The probability of observing under this predicted belief is
Here is the likelihood of observation after arriving in , and the sum marginalizes the unobserved successor state.
The corrected belief is then Bayes' rule applied cellwise:
The denominator must be positive; a zero-probability observation is outside the conditioning event. The result is the posterior probability of successor state . Before the observation is seen, the next belief is a random variable because different possible observations induce different posteriors. This remains true in our maze even though both the physical dynamics and sensor are deterministic: uncertainty about the current latent cell induces uncertainty about which deterministic successor and signature will occur. Once the action and observation are fixed, the Bayes update itself is deterministic.
Three properties of the belief MDP are worth keeping in mind.
First, an optimal policy can be expressed as a function of the current belief under the standard POMDP assumptions. For a finite horizon it may also depend on the remaining time; for an infinite-horizon discounted problem, a stationary belief policy can be optimal under the usual conditions. The belief is therefore a sufficient information state for control. We discussed sufficient state in Sufficient State and State Abstraction. A controller need not retain the full history when it can update the exact belief and any required time index.
Second, for a finite-state, finite-action, finite-observation POMDP with a finite horizon and expected-reward maximization, the optimal belief value function is piecewise linear and convex in . This means for a finite set of alpha vectors , where is the inner product of the -th vector with the belief. Each alpha vector represents a conditional plan, and the maximizing vector defines a region of belief space. This structural result (Sondik, 1971; Smallwood and Sondik, 1973) underlies exact finite-horizon POMDP methods. The convexity is what makes the structure exploitable: the value function is the upper envelope of a family of linear functions, and the choices of optimal action correspond to the faces of that envelope. A piecewise-linear convex value function is a much more constrained object than a general one, and in principle it can be computed exactly.
Third, the exact representation grows quickly in the worst case. Before pruning, the standard finite-horizon backup can generate a number of candidate alpha vectors that is doubly exponential in horizon when there is more than one observation; the number that survives pruning is problem-dependent. The belief space is also a simplex of dimension rather than a finite collection of states. Pruning helps, but exact methods become impractical as the state, action, observation, or horizon sizes increase. The two-state tiger problem is useful because its belief space is one-dimensional and can be visualized. State count alone does not determine the number of surviving alpha vectors; the structure of transitions, observations, rewards, and horizon also matters.
The Approximations
Large POMDPs usually require approximations, while exact belief-space methods remain useful for small finite problems. It helps to organize approximations by what they discard and to ask which discarded structure is least costly for the problem at hand.
Throw away uncertainty. The crudest family. QMDP computes the MDP Q-function and then evaluates actions by their expectation under the belief:
Here is the optimal fully observed MDP action value and weights each possible state. This is fast but structurally incapable of valuing future information, because it assumes that uncertainty disappears after the current action. For reward maximization, granting the controller that extra future information makes the QMDP value an optimistic upper bound on the corresponding POMDP value, not a lower bound. The resulting policy can favor actions that work only under the assumed future certainty. The observation model does not appear in the QMDP expression, so an otherwise reward-neutral sensing action receives no credit for improving later decisions. QMDP could still select an action called peek for its modeled reward or state transition, but not for the observation it supplies.
A close cousin is most-likely-state planning: take and plan as if were known. It is cheap and reuses an MDP planner, but it discards probability mass on every other state and cannot value observations. It can be adequate when the belief is tightly concentrated and nearby states have similar action values. A multimodal belief is a warning sign rather than a proof of failure: the approximation fails when plausible modes favor materially different actions.
Throw away most beliefs. Point-based methods apply backups at a finite set of representative beliefs rather than across the full simplex. PBVI (Pineau, Gordon, and Thrun, 2003) uses explorative stochastic trajectories to select belief points. SARSOP (Kurniawati, Hsu, and Lee, 2008) focuses sampling on an approximation to the beliefs reachable under optimal policies, not specifically on the boundary of the reachable set. HSVI (Smith and Simmons, 2004) maintains upper and lower bounds and directs trials using those bounds. These methods can produce alpha-vector policy representations. Their tractability depends on more than the raw dimension: reachable-set geometry, sparsity, action and observation counts, and the required error tolerance also matter. A finite sample of reachable beliefs can greatly reduce work, but it can miss decision-relevant regions and therefore does not preserve the instance's optimum automatically.
Throw away the global offline policy. Online methods plan from the current belief with a fixed compute budget, execute one action, and plan again. They may still use a fixed finite search depth or discounted infinite-horizon objective. If the planner only needs the next action, it can concentrate on beliefs reachable from the current one. The belief tree still branches over observations, so sampling and, in continuous spaces, observation progressive widening are used to keep the represented tree finite. POMCP is one online sampling approach.
Let us build the POMDP version of the maze. We keep the deterministic dynamics and replace the position sensor with a local wall signature: the agent perceives a four-bit pattern saying which of its four neighbors are blocked. Many cells share a signature, so the sensor is ambiguous rather than noisy. Ambiguous sensors are useful here because they separate two kinds of partial observability that learners often conflate: noise (the same state produces different observations on different visits) and aliasing (different states produce the same observation). Our maze has only aliasing, and every observation is produced deterministically by its state.
def belief_update(belief, action, obs):
"""Exact Bayes filter for this model."""
predicted = belief @ T[action]
posterior = predicted * O[obs]
total = posterior.sum()
if total <= 1e-12:
raise ValueError(
"observation has zero probability under predicted belief"
)
return posterior / total
def belief_grid(belief):
"""Reshape a belief vector back onto the 2D maze for plotting."""
canvas = np.full((H, W), np.nan)
for i, s in enumerate(FREE):
canvas[s] = belief[i]
return canvas
def entropy(belief):
positive = belief[belief > 0.0]
value = float(-np.sum(positive * np.log(positive)))
return max(0.0, value) # avoid displaying negative zero at a point massNow let us look at how much information the sensor carries. Grouping cells by signature shows how badly aliased the world is.
signature 3 [U=0 D=0 L=1 R=1] -> 6 cells: [(1, 0), (1, 5), (2, 5), (3, 3), (3, 5), (4, 5)] signature 5 [U=0 D=1 L=0 R=1] -> 1 cells: [(4, 3)] signature 6 [U=0 D=1 L=1 R=0] -> 1 cells: [(2, 0)] signature 7 [U=0 D=1 L=1 R=1] -> 2 cells: [(5, 0), (5, 5)] signature 9 [U=1 D=0 L=0 R=1] -> 2 cells: [(0, 5), (2, 3)] signature 10 [U=1 D=0 L=1 R=0] -> 2 cells: [(0, 0), (4, 0)] signature 12 [U=1 D=1 L=0 R=0] -> 8 cells: [(0, 1), (0, 2), (0, 3), (0, 4), (2, 1), (2, 2), (4, 1), (4, 2)]
There are seven distinct signatures for twenty-two free cells, so a signature does not identify the position. Signature 12 is shared by eight cells: four along the top corridor, two in the middle row, and two in the bottom-left arm. Under a uniform prior over free cells, conditioning on an initial observation of 12 gives a uniform belief over those eight cells with entropy nats. That is the largest initial alias class in this maze.
There is no sensor noise to average away; the ambiguity comes from perceptual aliasing. Repeating the same observation without changing viewpoint cannot separate the eight signature-12 cells, but filtering an action-observation sequence can. Noisy sensors and aliased sensors can both require history, and a most-likely-state controller is not guaranteed to become optimal in either case. The important distinction is that repeated independent measurements may reduce noise, whereas resolving aliasing here requires actions that lead candidate cells to different later signatures.
Let us watch a belief evolve under a fixed action sequence, to make the mechanics concrete before we search over them.
initial support: 8 cells, entropy 2.079 after another 'right' and observing 12: 5 cells, entropy 1.609 after another 'right' and observing 12: 2 cells, entropy 0.693 after another 'right' and observing 12: 1 cells, entropy 0.000
Starting from true cell , each right action produces the displayed signature and prunes the posterior support from eight cells to five, then two, then one. This is filtering under a fixed, physically consistent action-observation sequence. It shows how motion can disambiguate aliased positions; deciding whether that information is worth the motion is the planner's separate job.