Basics

Notation and conventions

For a set , is the set of probability distributions on , and is the set of finite sequences over . In a kernel written , the semicolon separates the point at which the output distribution is evaluated () from the inputs (), so .

Time runs as , so an agent acts on the history available so far and the observation is generated from the post-action state . There is no observation before the first action.

An uppercase symbol such as , , or denotes a random variable, and the matching bold lowercase symbol , , denotes a value it can take. Bold symbols correspond to vector-valued entities.

Negative subscripts, like , denote ”… of all agents other than ”.

Definition (PED-POMDP)

A policy-extended Dec-POMDP (PED-POMDP) is a 10-tuple

where every family written is indexed by , while are global, and

  • is the state space,
  • is the agent index space,
  • is the action space of agent ,
  • is the state transition kernel,
  • is the initial state distribution,
  • is the observation space of agent ,
  • is the observation kernel of agent ,
  • is the (stochastic) reward of agent ,
  • is the discount factor, shared by all agents, and
  • is the policy of agent .

Under a joint policy, agent seeks to maximize . Rewards are per-agent, so the model is general-sum, and the cooperative Dec-POMDP is the special case . The codomain is WLOG: any bounded real-valued reward affinely rescales into and is realized as a Bernoulli draw with that mean, preserving expected return up to affine equivalence and hence the optimal-policy set.

Definition (Rollout)

A -step rollout of a PED-POMDP is a tuple

where , , and : the state sequence, joint actions, and joint observations of a gameplay of exactly time steps. Under the time convention, is the post-action state of step and is emitted from , so the state sequence has entries and ends at . Write for agent ‘s history at step read off . Since the PED-POMDP fixes every policy, it induces a probability distribution over -step rollouts,

and carries all information needed to describe the gameplay: in particular every reward is determined, . In this sense a PED-POMDP is a closed stochastic system rather than a game to be solved, and the rollout distribution is its complete behavioral description.

Remark (state can encode metacognitive information)

Consider a PED-POMDP with . For each , instead of one , give a space of observability configurations, a parameterized pair , and a prior from which is drawn once at the start. This lets each agent hold uncertainty over how its peer (or itself) accesses the world, which makes explicit what we may call metacognition, cognition about one’s own or another’s cognitive process. An agent can wonder whether a peer really saw what it claims, whether a news item was fabricated, or whether its own reasoning slipped somewhere.

The base PED-POMDP already captures this implicitly. Replace by , fold into the initial distribution through , hold fixed under the transition through , take , and set . The result is equivalent to the configuration-augmented model with no explicit or . We take this route from here on, folding metacognitive information into , which also lets object-level and metacognitive information be handled the same way.

Remark (state can encode policy information)

The same maneuver lets us reintroduce parameterized policies should we need them. Consider an extension that replaces each parameter-free with a parameter space and a parameterized policy , with each drawn from a prior at the start, independently across . The base PED-POMDP already captures this. Replace by , fold the priors into the initial distribution through , and hold fixed under the transition through . The equivalence further requires a private revelation of to agent , since agent needs to compute its own action distribution: extend to , set , and deliver the component as a one-time private observation to agent alone, before its first action. The base PED-POMDP’s parameter-free policy then reads off its augmented history and reproduces of the parametric model. So parameterized policies and priors over them fold into alongside object-level and metacognitive information; we keep the base 10-tuple from here on.

Definition (Sibling)

Call two PED-POMDPs -siblings if they agree on every component except possibly . Varying agent ‘s behavior while holding the rest of the world fixed means moving within the -sibling family, and definitions that compare agent ‘s behavior against alternatives are posed across the family, so that each definition stays internal to a single PED-POMDP.

Coherence Optimization

Definition (Egocentric POMDP)

Fix a PED-POMDP and an agent . Write

and let a generic egocentric state be , where is the others’ joint history of actions and observations. The egocentric POMDP of agent is

Agent treats the world state and the others’ private histories as a single hidden environment. It observes neither nor the others’ observations or actions; the channel exposes only . The others’ actions belong in the hidden state for the same reason their observations do, namely that each depends on ‘s own past actions, so a belief about ‘s next action requires a belief about ‘s past actions. Type uncertainty, when present, is folded into by the previous remark and so already enters through .

Write , with the -th coordinate of . For a source state with , own action , and a target state of the form ,

All target states not of this form ( not extended by exactly one pair) receive value . The product runs over , since agent ‘s own observation is produced by rather than by the latent transition. The remaining components are

with all other initial-state values . The notation records that agent ‘s observation depends on the world component alone and is constant in . The reward keeps the expectation over , because the others’ time- action is realized during the transition and is not yet in the source state. The policy of is itself a policy of , since both act on agent ‘s own action-observation history.

No component of involves , so all -siblings share one egocentric POMDP, and with it everything built from below. Every definition below relativizes verbatim to a subjective prior substituted for in , should a misspecified initial belief ever be needed; we do not use this generality.

Definition (Egocentric Belief-State MDP)

We further define the egocentric belief-state MDP

For a belief and action , let

be the one-step predictive distribution after . For an observation with

let the posterior be

Then

and . The belief-state reward overloads the state-level by its argument type.

Definition (Belief-State Operator)

We then define the belief-state operator

a partial map defined at every history that occurs, meaning it has positive probability under the rollout distribution of . For an occurring , write , , and for the final state, agent ‘s history , and the others’ joint history of a random -step rollout , and set

the conditional distribution of the hidden environment given agent ‘s own history. Every factor in depends on only through , which the conditioning fixes, so these factors cancel between numerator and denominator; the conditional is therefore computed by the -free recursion and, for each , the one-step update

with . Two consequences. First, while the domain of references through the rollout distribution, its values do not: all -siblings of assign the same belief to any history occurring in both, so beliefs are comparable across the family. Second, every is a reachable belief of , since the recursion conditions only on observations of positive probability, matching the branching of . The belief is a sufficient statistic of the history for control in , which is what makes an equivalent fully observed problem.

Definition (Policy Translation)

A stationary policy of the belief-state MDP is a map , since is fully observed and the belief is its state. Its translation is the agent- policy that filters the history so far into its belief and then applies ,

Substituting for yields an -sibling of , and the defining equation is read within . This is well-founded because the recursion computing is shared by all -siblings: construct forward in time, acting on at the empty history, updating the belief on each realized pair, and acting on the updated belief. This defines at every history occurring in , which is all that the rollout distribution of ever consults; elsewhere it may be set arbitrarily. A translated policy depends on the history only through its belief: two occurring histories with must receive the same action distribution. A general policy may act differently at and , so it need not be the translation of any . This is the precise sense in which belief-state policies are a special kind of history policy but not conversely.

Definition (Belief-State Individual Optimality)

A policy of is belief-state individually optimal (b.i.o.) if it satisfies the Bellman optimality equation of under reward and discount . Writing for the optimal value,

and is b.i.o. when at every reachable . Belief-state optimality is a property of alone; it involves neither nor the rollout distribution, so the b.i.o. set is shared by all -siblings.

Definition (Individual Optimality)

Agent is individually optimal (i.o.) in if some b.i.o. policy satisfies at every occurring history , that is, agrees with the translation wherever agent ‘s gameplay actually reaches. Restricting to occurring histories loses nothing: agreement propagates forward from the empty history, so an i.o. induces the same rollout distribution as the sibling and in particular attains the optimal value. Since the b.i.o. set and the belief values are shared across -siblings, asking in which members of the family agent is i.o. is well-posed; this is how alternative behaviors of agent are compared.

Definition (Belief-Groundedness)

For an occurring history of agent , agent ‘s belief-groundedness at in is (we assume below that the JS divergence uses base-2 logarithm)

the negated smallest JS gap between agent ‘s action distribution at and an optimal action distribution at the induced belief. Like individual optimality, it compares across -siblings: each sibling is scored at its own occurring histories against the shared benchmark.

If agent is i.o. in , then a single b.i.o. witness matches at every occurring history, so for all occurring . The converse needs an extra step. From at every occurring one obtains, for each such , some optimal action distribution equal to , and these can come from different optimal policies. To recover i.o. one also needs to depend on the history only through the belief and to select a single optimal action distribution at each belief, so that one works everywhere. Without that, does not give i.o.

Definition (Inferred Coherence)

The -step rollout distributions are projectively consistent — marginalizing the -step distribution over recovers the -step one — so they extend uniquely to a distribution over infinite rollouts ; the expectation below is under this distribution. For agents and an occurring history of agent , the -inferred coherence of agent is

the expected -discounted temporal sum of agent ‘s belief-groundedness at its actualized history, conditioned on agent ‘s history being , with the prefactor normalizing the range to . Every realization of compatible with the conditioning occurs, so is defined almost surely, and the sum converges since . For the conditional expectation infers what agent ‘s gameplay has already been from agent ‘s vantage point; for it forecasts. The case reads as self-assessed coherence.

Remark (empirical methods for optimizing inferred coherence)

arXiv:2601.13566 gives a candidate empirical method for optimizing for inferred coherence. That paper focuses on the case where is a language model, playing the role of its prior policy, and is another language model, playing the role of its deterministic policy; we hope that similar methods can be extended to as a more general learning agent and as a human. However, the method relies on a pretrained prior that specifies marginal coherence, which may significantly limit its usefulness; looking for better methods is a priority.