Index policies have sublinear value decomposition error (Cor. 1)
ProvedMarkovEntanglement.rmab_index_policy_decomposition_errorConsider an -agent restless multi-armed bandit. The agents are homogeneous: they share one local state space , one pair of local transition kernels (idle) and (activate), and one pair of local rewards. At every step a budget forces exactly of the agents to be activated, where for a fixed activation fraction .
Fix a priority index and let be an index policy for it: activate agents in descending order of the priority of their local state until the budget is exhausted, spreading the activation uniformly over the agents that share the marginal state. Let denote the configuration of the system, being the fraction of agents currently in local state , and let be the mean-field transition map of the configuration, . Note that does not depend on , which is exactly why the budget must be a fixed fraction of rather than a fixed count, and why the constant below may be quantified before .
Assume the two standard technical conditions on index policies:
- Uniform global attractor property (UGAP). There is a point with that attracts every initial configuration, uniformly in the initial point: for every there is a with for all and all .
- Non-degeneracy. At the fixed point some local state is served only fractionally, so the limiting policy genuinely randomises there.
Let be a strictly positive stationary occupancy measure of the induced chain, let the per-agent rewards be bounded by , let be the discount factor, let be the joint -function (the Bellman fixed point of the joint chain with the summed reward), and let each be the Bellman fixed point of an agent's own local transition matrix — one attaining that agent's measure of Markov entanglement — with that agent's own reward.
Then there is a constant , independent of , such that
The point of the result is that the right-hand side grows like while the joint -function itself grows like : the relative decomposition error vanishes as the system grows. This is the theoretical justification for the value decompositions used in large-scale restless-bandit applications.
import Mathlib import Definitions.Def_markov_entanglement_meanfield open scoped BigOperators open MarkovEntanglement
namespace MarkovEntanglement
/-- Corollary 1 (Chen and Peng, "Multi-agent Markov Entanglement", arXiv:2506.02385v3,
Section 7.1, p. 24). Consider an `N`-agent restless multi-armed bandit: homogeneous agents
sharing a local state space `S` and a pair of local kernels `P0, P1`, each agent choosing
between idle and activate, and a budget that activates a fixed fraction `α` of the agents at
every step. Fix a priority index `ν` and let `π` be an index policy for the budget
`M = ⌊α * N⌋`. Assume the explicit mean-field map of the configuration process at the
activation fraction `α` admits `m✦` as a uniform global attractor and is non-degenerate at
`m✦` — both properties of the limit model, quantified before `C` and before `N`.
Then there is a constant `C`, **independent of `N`**, such that the error of decomposing the
joint `Q`-function of the `N`-agent chain into a sum of per-agent local `Q`-functions is
bounded, in the occupancy-weighted norm, by
`‖Q^π_{1:N} − ∑ᵢ Q^π_i‖_μ ≤ 4 * C * γ * √N * r_max / (1 − γ)^2`.
The bound is **sublinear in `N`** while the joint `Q`-function itself is of order
`N * r_max / (1 − γ)`: the relative decomposition error vanishes as the system grows. This
is what justifies the value decompositions used in practice for large-scale restless bandits.
The activated amount is the fraction `α` of `N` rather than a fixed count precisely because
the mean-field map `φ` is independent of `N` only when the activated fraction is held fixed;
`C` is quantified before `N` for the same reason. -/
theorem rmab_index_policy_decomposition_error
{S : Type*} [Fintype S] [DecidableEq S]
(P0 P1 : Matrix S S ℝ) (hP0 : IsTransitionMatrix P0) (hP1 : IsTransitionMatrix P1)
(ν : S → ℝ) (hν : Function.Injective ν) (α : ℝ) (hα : 0 < α) (hα1 : α < 1)
(mstar : S → ℝ) (hmstar : IsConfiguration mstar)
(hUGAP : IsUniformGlobalAttractor (meanFieldMap P0 P1 ν α) mstar)
(hnd : IsNonDegenerateMeanField ν α mstar)
(γ : ℝ) (hγ : 0 ≤ γ) (hγ1 : γ < 1) (rmax : ℝ) :
∃ C : ℝ, 0 ≤ C ∧
∀ (N : ℕ), 0 < N →
∀ (π : (Fin N → S) → (Fin N → Bool) → ℝ),
IsIndexPolicy ν ⌊α * (N : ℝ)⌋₊ π →
∀ (μ : Joint (StateAction (fun _ : Fin N => S) (fun _ : Fin N => Bool)) → ℝ),
IsDist μ →
IsExchangeableDist μ →
IsStationary (inducedTransition
(fun s a s' => ∏ j, rmabKernel P0 P1 (s j) (a j) (s' j)) π) μ →
∀ (r : ∀ i : Fin N, S × Bool → ℝ), (∀ i x, |r i x| ≤ rmax) →
∀ (Q : Joint (StateAction (fun _ : Fin N => S) (fun _ : Fin N => Bool)) → ℝ)
(Pl : ∀ i : Fin N, Matrix (S × Bool) (S × Bool) ℝ)
(Qi : ∀ i : Fin N, S × Bool → ℝ),
IsBellmanQ (inducedTransition
(fun s a s' => ∏ j, rmabKernel P0 P1 (s j) (a j) (s' j)) π)
(fun p => ∑ i, r i (p i)) γ Q →
(∀ i, IsTransitionMatrix (Pl i)) →
(∀ i, IsLocalTransitionN i (inducedTransition
(fun s a s' => ∏ j, rmabKernel P0 P1 (s j) (a j) (s' j)) π) μ (Pl i)) →
(∀ i, IsBellmanQ (Pl i) (r i) γ (Qi i)) →
muNorm μ (fun p => Q p - ∑ i, Qi i (p i))
≤ 4 * C * γ * Real.sqrt (N : ℝ) * rmax / (1 - γ) ^ 2 := by
sorry
end MarkovEntanglementRead-back
What the Lean code literally says, in plain math · claude-opus-5
The declaration asserts the following. Fix a finite type equipped with decidable equality, together with:
- two matrices , each assumed to be a transition matrix, i.e. for all and for every ;
- a function assumed to be injective;
- a real number with ;
- a function assumed to be a configuration, i.e. for all and (note that this last equation is what forces to be nonempty; is not otherwise assumed nonempty);
- a real number with , and an arbitrary real number , on which no sign assumption is placed.
Two hypotheses are imposed on the mean-field map , which is the map sending to the function
where the activated fraction is and the higher-priority mass is . The two hypotheses are:
- Uniform global attractor: , and for every there exists such that for every and every configuration (any with pointwise and ), the iterate satisfies (the supremum is a real supremum over ).
- Mean-field non-degeneracy: there exists with , , and — all three inequalities strict.
Under exactly these assumptions, the claim is that there exists a real constant with (only nonnegativity is claimed; may be , and is permitted to depend on all of the data fixed above, including , , , , , , and , but is chosen once and for all before and everything that follows), such that for every with , the following holds. Write (the natural-number floor; for small this can be , e.g. with ). Then for every function that is an index policy with budget — meaning all three of:
- for all and for every joint state ;
- whenever , the action profile is budgeted: exactly;
- for every joint state and every agent , the marginal activation probability equals , where if and otherwise , with , , and truncated natural-number subtraction (so the numerator is whenever );
and for every function on the joint state–action space , whose elements we write as , such that
- is a distribution: for all and ;
- is exchangeable: for every permutation of and every ;
- is stationary for the induced transition matrix on , i.e. for every , where
and for every family of per-agent reward functions satisfying for all and all (note: since and is nonempty, such an exists only if ; for negative everything from this point on is vacuous), and for every , every family of matrices and every family , such that
- satisfies the global Bellman equation for with the additive reward : for every ;
- every is a transition matrix (nonnegative entries, rows summing to );
- every is a local transition of agent for under : for all ,
(in particular, at a local state whose -marginal is this condition says nothing about the row beyond forcing the right-hand side to vanish);
- every satisfies the local Bellman equation for with reward and the same : for every ;
the conclusion is the inequality
Several features of the quantifier structure are worth stating explicitly: the constant is required to be independent of , , , , , and , but not of or , both of which are fixed before is produced and both of which also appear on the right-hand side; the numerator is and the whole product is divided by ; when (permitted, since only is assumed) the right-hand side is exactly , so the assertion in that case is that the weighted error is ; the mean-field hypotheses are stated for the exact fraction , whereas the index policy is required to have the integer budget ; no irreducibility, aperiodicity, positivity or uniqueness assumption is placed on , on , or on the solutions , of the Bellman equations; and the left-hand side is a -weighted sum of absolute deviations over the whole joint state–action space, not a supremum.
Confirmed by the mission captain (proposal self-audit).