Shared rewards add a reward-entanglement term to the error
OpenMarkovEntanglement.cooperative_decomposition_error_with_reward_entanglementStatement
Proposition. For a fully cooperative system whose global reward need not decompose, let be the measure of reward entanglement, the -norm distance from to the nearest sum of local rewards. Then
Notes
The main bound assumes the reward already splits across agents. In genuinely cooperative problems it usually does not — there is one shared reward — and this proposition covers that case by adding a second source of error.
The two terms separate cleanly and are worth reading side by side. Reward entanglement enters with a single factor , because a reward misfit is a zeroth-order error that is merely summed along the trajectory. Transition entanglement enters with , because a one-step transition error compounds through the value recursion. So an imperfectly decomposable reward is the milder of the two defects.
Search terms: shared reward multi-agent, reward decomposition error, cooperative MARL value decomposition, credit assignment error bound.
import Mathlib import Definitions.Def_markov_entanglement_multi open scoped BigOperators open MarkovEntanglement
namespace MarkovEntanglement
theorem cooperative_decomposition_error_with_reward_entanglement
{N : ℕ} {S : Fin N → Type*} [∀ i, Fintype (S i)] [∀ i, DecidableEq (S i)]
(P : Matrix (Joint S) (Joint S) ℝ) (μ : Joint S → ℝ) (γ : ℝ) (rmax : Fin N → ℝ)
(r : Joint S → ℝ) (rl : ∀ i, S i → ℝ) (Q : Joint S → ℝ)
(Pl : ∀ i, Matrix (S i) (S i) ℝ) (Qi : ∀ i, S i → ℝ)
(hγ : 0 ≤ γ) (hγ1 : γ < 1) (hP : IsTransitionMatrix P)
(hμ : IsPositiveDist μ) (hstat : IsStationary P μ)
(hr : ∀ i s, |rl i s| ≤ rmax i)
-- `rl` must attain the measure of reward entanglement, exactly as `Pl` attains
-- the measure of Markov entanglement; without this the local rewards are free
-- and the bound is false.
(hrl : muNorm μ (fun p => r p - ∑ i, rl i (p i)) = rewardEntanglement μ r)
(hQ : IsBellmanQ P r γ Q)
(hPl : ∀ i, IsTransitionMatrix (Pl i))
(hopt : ∀ i, muAgentTVDistN i μ P (Pl i) = entanglementN i μ P)
(hQi : ∀ i, IsBellmanQ (Pl i) (rl i) γ (Qi i)) :
muNorm μ (fun p => Q p - ∑ i, Qi i (p i))
≤ rewardEntanglement μ r / (1 - γ)
+ 4 * γ * (∑ i, entanglementN i μ P * rmax i) / (1 - γ) ^ 2 := by
sorry
end MarkovEntanglementRead-back
What the Lean code literally says, in plain math · claude-opus-5
Read-back: cooperative_decomposition_error_with_reward_entanglement
The setting and all the data
Fix a natural number (implicit) and a family of types (implicit), each of which is finite and has decidable equality. Write
for the joint space: an element is a tuple assigning to each agent a point . is finite. The following data are given, all explicitly quantified over (the theorem asserts its conclusion for every choice of them satisfying the hypotheses below):
- , a matrix on the joint space;
- , a weight function on the joint space;
- , a scalar;
- , one real number per agent;
- , a function on the joint space;
- for each agent (a family of local functions);
- ;
- for each agent ;
- for each agent .
The non-standard notions, expanded
All of the following are definitions from this bundle; they are unfolded here so the statement can be read without them.
- Row-stochastic ("
IsTransitionMatrix") for a square matrix indexed by a finite set: for all , and for every . Nothing is required of columns. - Strictly positive distribution ("
IsPositiveDist") for : for every , and . - Stationarity ("
IsStationary") of for : for every , . - -norm ("
muNorm") of a function on a finite index set: . (The weights enter without an absolute value of their own; here they are positive by hypothesis.) - Agent- marginal of the joint matrix ("
marginalN"): for and ,
i.e. the total mass that row of puts on joint points whose -th coordinate is . Note this depends on the whole tuple , not only on .
- -weighted agent-wise total variation distance ("
muAgentTVDistN") between the joint matrix and a candidate local matrix :
It is a -weighted average over joint points, not a maximum.
- Agent-wise measure of Markov entanglement ("
entanglementN") for agent :
the greatest lower bound of the above distance over all row-stochastic local candidates.
- Measure of reward entanglement ("
rewardEntanglement"):
the greatest lower bound, over all families of local functions, of the -weighted average absolute deviation of from a sum of local functions.
- Bellman fixed point ("
IsBellmanQ") for a square matrix on a finite set , a function , a scalar , and a function : for every .
The bundle also defines separability of a joint matrix, tensor products of local matrices, product-form transitions, local-transition compatibility, decomposable rewards, and value decomposition. None of those notions occurs in this statement; the only entanglement quantities used are and as expanded above.
The hypotheses
- .
- (strict).
- is row-stochastic on the joint space.
- is a strictly positive probability distribution on the joint space.
- is stationary for .
- For every agent and every : . (This bounds the local functions ; it says nothing about .)
- — i.e. the given family exactly attains the reward-entanglement infimum.
- is the Bellman fixed point for , , : for every , .
- Each is row-stochastic.
- For every agent : — i.e. each exactly attains the agent-wise entanglement infimum.
- For every agent , is the Bellman fixed point for , , : for every .
The conclusion
The grouping is as displayed: the whole sum over agents is multiplied by and then divided by ; the first term is divided by (first power). It is a non-strict inequality ().
Every variable of the conclusion: constrained or free
- and the family — free, apart from finiteness and decidable equality. In particular is permitted. Each is forced to be nonempty, but only indirectly: hypothesis 4 requires , which is impossible if is empty, and nonempty forces every nonempty. (This matters for ; see below.)
- — constrained: strictly positive, sums to , stationary for (hyps. 4, 5).
- — constrained: row-stochastic with stationary (hyps. 3, 5); it also feeds hypotheses 8 and 10. Otherwise arbitrary; nothing forces it to be separable, product-form, or close to any local dynamics.
- — constrained to (hyps. 1, 2). Both denominators and are therefore strictly positive, so no division-by-zero junk value can arise.
- — entirely free: no bound, no sign condition, no decomposability condition is imposed on . It is not itself controlled by . It appears on both sides of the conclusion: it determines through hypothesis 8, and it determines on the right.
- — constrained by hypothesis 7 to be an exact minimizer of the reward-entanglement objective (and by hypothesis 6 to be bounded by ). Minimizers need not be unique (for instance, a constant may be shifted between two agents' local functions without changing the objective); the conclusion is asserted for every minimizer, together with the it induces.
- — constrained only from below: hypothesis 6 forces (using that is nonempty, as forced by hypothesis 4). Any larger value is legal, so is free upward. Because and , enlarging can only enlarge the right-hand side; it appears nowhere on the left. Had some been empty, hypothesis 6 would have been vacuous for that and could have been taken arbitrarily negative, which — given a strictly positive — would drive the right-hand side to while the left-hand side stayed fixed; hypothesis 4 is what blocks this.
- — constrained and in fact uniquely determined by hypothesis 8: the equation has a unique solution because is row-stochastic and .
- — constrained: row-stochastic and an exact minimizer of (hyps. 9, 10). Such a minimizer always exists (the row-stochastic matrices on a nonempty finite form a nonempty compact set and is continuous in its matrix argument), but it need not be unique; the conclusion is asserted for every choice.
- — constrained and uniquely determined by hypothesis 11, given , and , for the same invertibility reason.
- and are not free variables: they are determined by .
On falsification by a free variable. The two data with genuinely unbounded freedom are and . occurs only on the right, with the nonnegative coefficient , so increasing it only weakens the claim, and hypothesis 6 together with the nonemptiness of prevents it from being negative. occurs on both sides. The residual freedom — the choice of , , , and of which minimizers , are taken — is not resolved by inspection of the statement: no variable appears solely on the left-hand side in a way that a sign or scaling argument alone would exploit. I therefore report no explicit falsifying instantiation.
Joint satisfiability, and degenerate cases
- The hypotheses are jointly satisfiable, including non-degenerately. Take any and any nonempty finite ; let be the matrix all of whose entries equal and the uniform distribution on . Then hypotheses 3, 4, 5 hold. Take anywhere in and arbitrary; a minimizing family exists (the objective is the -norm distance from to a fixed finite-dimensional linear subspace of functions on , so the infimum is attained), giving hypothesis 7; set for hypothesis 6; take uniform on , which makes , giving hypotheses 9, 10; and let , be the (unique) Bellman solutions. So the statement is not vacuous.
- . Permitted. Then is a one-point set (the empty tuple), every sum over agents is , of that point is , is the matrix , , hypothesis 7 holds automatically, and . Both sides of the conclusion equal , so it holds with equality.
- . Permitted. Then and , the second term on the right vanishes, and both sides equal by hypothesis 7 — equality again.
- Empty / empty joint space. Ruled out by hypothesis 4, as noted above.
- The infima are genuine. Both defining sets are nonempty (for : take all ; for : take any row-stochastic , which exists since is nonempty and finite) and are bounded below by , since and the summands are absolute values. Hence and , and neither infimum falls back on the convention that assigns the value to the infimum of an empty or unbounded-below set of reals.
- Both attainment hypotheses (7 and 10) are equalities, not inequalities: the given and must sit exactly at the respective infima, not merely near them.
Confirmed by the mission captain (proposal self-audit).