Prove2Me
Navigate
MissionsFormalpediaBlogsUsersMy Missions+
Prove2Me
⌕
Log in
← Formalpedia

Index policies have sublinear value decomposition error (Cor. 1)

Proved
MarkovEntanglement.rmab_index_policy_decomposition_error

by tianyipeng · Aug 28, 2026 · Mathlib c5ea003 (Lean v4.30.0)

Consider an NNN-agent restless multi-armed bandit. The agents are homogeneous: they share one local state space SSS, one pair of local transition kernels P0P_0P0​ (idle) and P1P_1P1​ (activate), and one pair of local rewards. At every step a budget forces exactly MMM of the NNN agents to be activated, where M=⌊αN⌋M = \lfloor \alpha N \rfloorM=⌊αN⌋ for a fixed activation fraction α∈(0,1)\alpha \in (0,1)α∈(0,1).

Fix a priority index ν:S→R\nu : S \to \mathbb{R}ν:S→R and let π\piπ 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 m∈Δ∣S∣m \in \Delta^{|S|}m∈Δ∣S∣ denote the configuration of the system, mxm_xmx​ being the fraction of agents currently in local state xxx, and let φ\varphiφ be the mean-field transition map of the configuration, φ(m)=E[m[t+1]∣m[t]=m,π]\varphi(m) = \mathbb{E}[m[t+1] \mid m[t] = m, \pi]φ(m)=E[m[t+1]∣m[t]=m,π]. Note that φ\varphiφ does not depend on NNN, which is exactly why the budget must be a fixed fraction of NNN rather than a fixed count, and why the constant below may be quantified before NNN.

Assume the two standard technical conditions on index policies:

  1. Uniform global attractor property (UGAP). There is a point m∗m^\astm∗ with φ(m∗)=m∗\varphi(m^\ast) = m^\astφ(m∗)=m∗ that attracts every initial configuration, uniformly in the initial point: for every ε>0\varepsilon > 0ε>0 there is a TTT with ∥φt(m)−m∗∥∞<ε\|\varphi^t(m) - m^\ast\|_\infty < \varepsilon∥φt(m)−m∗∥∞​<ε for all t≥Tt \ge Tt≥T and all m∈Δ∣S∣m \in \Delta^{|S|}m∈Δ∣S∣.
  2. Non-degeneracy. At the fixed point m∗m^\astm∗ some local state is served only fractionally, so the limiting policy genuinely randomises there.

Let μ1:Nπ\mu^\pi_{1:N}μ1:Nπ​ be a strictly positive stationary occupancy measure of the induced chain, let the per-agent rewards be bounded by rmax⁡r_{\max}rmax​, let γ∈[0,1)\gamma \in [0,1)γ∈[0,1) be the discount factor, let Q1:NπQ^\pi_{1:N}Q1:Nπ​ be the joint QQQ-function (the Bellman fixed point of the joint chain with the summed reward), and let each QiπQ^\pi_iQiπ​ 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 CCC, independent of NNN, such that

∥Q1:Nπ−∑i=1NQiπ∥μ1:Nπ≤4CγN rmax⁡(1−γ)2.\Big\| Q^\pi_{1:N} - \sum_{i=1}^N Q^\pi_i \Big\|_{\mu^\pi_{1:N}} \le \frac{4 C \gamma \sqrt{N}\, r_{\max}}{(1-\gamma)^2}.​Q1:Nπ​−i=1∑N​Qiπ​​μ1:Nπ​​≤(1−γ)24CγN​rmax​​.

The point of the result is that the right-hand side grows like N\sqrt{N}N​ while the joint QQQ-function itself grows like NNN: 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.

Preamble
import Mathlib
import Definitions.Def_markov_entanglement_meanfield

open scoped BigOperators
open MarkovEntanglement
Formal statement
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 MarkovEntanglement
Source
Shuze Chen and Tianyi Peng, 'Multi-agent Markov Entanglement', arXiv:2506.02385v3, Section 7.1, p. 24, Corollary 1
Read-back

What the Lean code literally says, in plain math · claude-opus-5

The declaration asserts the following. Fix a finite type SSS equipped with decidable equality, together with:

  • two matrices P0,P1∈RS×SP_0, P_1 \in \mathbb{R}^{S \times S}P0​,P1​∈RS×S, each assumed to be a transition matrix, i.e. Pk(x,y)≥0P_k(x,y) \ge 0Pk​(x,y)≥0 for all x,yx,yx,y and ∑yPk(x,y)=1\sum_{y} P_k(x,y) = 1∑y​Pk​(x,y)=1 for every xxx;
  • a function ν:S→R\nu : S \to \mathbb{R}ν:S→R assumed to be injective;
  • a real number α\alphaα with 0<α<10 < \alpha < 10<α<1;
  • a function m⋆:S→Rm^\star : S \to \mathbb{R}m⋆:S→R assumed to be a configuration, i.e. m⋆(x)≥0m^\star(x) \ge 0m⋆(x)≥0 for all xxx and ∑xm⋆(x)=1\sum_{x} m^\star(x) = 1∑x​m⋆(x)=1 (note that this last equation is what forces SSS to be nonempty; SSS is not otherwise assumed nonempty);
  • a real number γ\gammaγ with 0≤γ<10 \le \gamma < 10≤γ<1, and an arbitrary real number rmax⁡r_{\max}rmax​, on which no sign assumption is placed.

Two hypotheses are imposed on the mean-field map φ=φP0,P1,ν,α\varphi = \varphi_{P_0,P_1,\nu,\alpha}φ=φP0​,P1​,ν,α​, which is the map sending m:S→Rm : S \to \mathbb{R}m:S→R to the function

φ(m)(y)  =  ∑x∈S[(m(x)−fm(x)) P0(x,y)  +  fm(x) P1(x,y)],\varphi(m)(y) \;=\; \sum_{x \in S} \Big[\big(m(x) - f_m(x)\big)\,P_0(x,y) \;+\; f_m(x)\,P_1(x,y)\Big],φ(m)(y)=x∈S∑​[(m(x)−fm​(x))P0​(x,y)+fm​(x)P1​(x,y)],

where the activated fraction is fm(x)=min⁡ ⁣(m(x), max⁡(0, α−hm(x)))f_m(x) = \min\!\big(m(x),\ \max(0,\ \alpha - h_m(x))\big)fm​(x)=min(m(x), max(0, α−hm​(x))) and the higher-priority mass is hm(x)=∑y : ν(x)<ν(y)m(y)h_m(x) = \sum_{y \,:\, \nu(x) < \nu(y)} m(y)hm​(x)=∑y:ν(x)<ν(y)​m(y). The two hypotheses are:

  • Uniform global attractor: φ(m⋆)=m⋆\varphi(m^\star) = m^\starφ(m⋆)=m⋆, and for every ε>0\varepsilon > 0ε>0 there exists T∈NT \in \mathbb{N}T∈N such that for every t≥Tt \ge Tt≥T and every configuration mmm (any mmm with m≥0m \ge 0m≥0 pointwise and ∑xm(x)=1\sum_x m(x)=1∑x​m(x)=1), the iterate φt(m)\varphi^{t}(m)φt(m) satisfies sup⁡x∈S∣φt(m)(x)−m⋆(x)∣<ε\sup_{x \in S} \big|\varphi^{t}(m)(x) - m^\star(x)\big| < \varepsilonsupx∈S​​φt(m)(x)−m⋆(x)​<ε (the supremum is a real supremum over SSS).
  • Mean-field non-degeneracy: there exists x∈Sx \in Sx∈S with m⋆(x)>0m^\star(x) > 0m⋆(x)>0,  α−hm⋆(x)>0\ \alpha - h_{m^\star}(x) > 0 α−hm⋆​(x)>0, and  α−hm⋆(x)<m⋆(x)\ \alpha - h_{m^\star}(x) < m^\star(x) α−hm⋆​(x)<m⋆(x) — all three inequalities strict.

Under exactly these assumptions, the claim is that there exists a real constant CCC with C≥0C \ge 0C≥0 (only nonnegativity is claimed; CCC may be 000, and CCC is permitted to depend on all of the data fixed above, including SSS, P0P_0P0​, P1P_1P1​, ν\nuν, α\alphaα, m⋆m^\starm⋆, γ\gammaγ and rmax⁡r_{\max}rmax​, but is chosen once and for all before NNN and everything that follows), such that for every N∈NN \in \mathbb{N}N∈N with N>0N > 0N>0, the following holds. Write M=⌊αN⌋M = \lfloor \alpha N \rfloorM=⌊αN⌋ (the natural-number floor; for small NNN this can be 000, e.g. N=1N=1N=1 with α=12\alpha = \tfrac12α=21​). Then for every function π:(Fin N→S)→(Fin N→{tt,ff})→R\pi : (\text{Fin }N \to S) \to (\text{Fin }N \to \{\mathrm{tt},\mathrm{ff}\}) \to \mathbb{R}π:(Fin N→S)→(Fin N→{tt,ff})→R that is an index policy with budget MMM — meaning all three of:

  1. π(s,a)≥0\pi(s,a) \ge 0π(s,a)≥0 for all s,as,as,a and ∑aπ(s,a)=1\sum_{a} \pi(s,a) = 1∑a​π(s,a)=1 for every joint state sss;
  2. whenever π(s,a)≠0\pi(s,a) \ne 0π(s,a)=0, the action profile aaa is budgeted: #{i:ai=tt}=M\#\{i : a_i = \mathrm{tt}\} = M#{i:ai​=tt}=M exactly;
  3. for every joint state sss and every agent iii, the marginal activation probability ∑a1[ai=tt] π(s,a)\sum_{a} \mathbf{1}[a_i = \mathrm{tt}]\,\pi(s,a)∑a​1[ai​=tt]π(s,a) equals iap⁡(s,si)\operatorname{iap}(s, s_i)iap(s,si​), where iap⁡(s,x)=0\operatorname{iap}(s,x) = 0iap(s,x)=0 if cs(x)=0c_s(x) = 0cs​(x)=0 and otherwise iap⁡(s,x)=min⁡ ⁣(cs(x), M−˙Hs(x))/cs(x)\operatorname{iap}(s,x) = \min\!\big(c_s(x),\ M \dot- H_s(x)\big)/c_s(x)iap(s,x)=min(cs​(x), M−˙​Hs​(x))/cs​(x), with cs(x)=#{i:si=x}c_s(x) = \#\{i : s_i = x\}cs​(x)=#{i:si​=x}, Hs(x)=∑y : ν(x)<ν(y)cs(y)H_s(x) = \sum_{y \,:\, \nu(x)<\nu(y)} c_s(y)Hs​(x)=∑y:ν(x)<ν(y)​cs​(y), and −˙\dot-−˙​ truncated natural-number subtraction (so the numerator is 000 whenever Hs(x)≥MH_s(x) \ge MHs​(x)≥M);

and for every function μ\muμ on the joint state–action space ΩN:=∏i=1N(S×{tt,ff})\Omega_N := \prod_{i=1}^{N}(S \times \{\mathrm{tt},\mathrm{ff}\})ΩN​:=∏i=1N​(S×{tt,ff}), whose elements ppp we write as pi=(si(p),ai(p))p_i = (s_i(p), a_i(p))pi​=(si​(p),ai​(p)), such that

  • μ\muμ is a distribution: μ(p)≥0\mu(p) \ge 0μ(p)≥0 for all ppp and ∑pμ(p)=1\sum_{p} \mu(p) = 1∑p​μ(p)=1;
  • μ\muμ is exchangeable: μ(p∘σ)=μ(p)\mu(p \circ \sigma) = \mu(p)μ(p∘σ)=μ(p) for every permutation σ\sigmaσ of {1,…,N}\{1,\dots,N\}{1,…,N} and every ppp;
  • μ\muμ is stationary for the induced transition matrix TTT on ΩN\Omega_NΩN​, i.e. ∑pμ(p) T(p,q)=μ(q)\sum_{p} \mu(p)\,T(p,q) = \mu(q)∑p​μ(p)T(p,q)=μ(q) for every qqq, where
T(p,q)  =  (∏j=1NKaj(p)(sj(p), sj(q)))⋅π(s(q), a(q)),Ktt=P1, Kff=P0;T(p,q) \;=\; \Big(\prod_{j=1}^{N} K_{a_j(p)}\big(s_j(p),\, s_j(q)\big)\Big)\cdot \pi\big(s(q),\, a(q)\big), \qquad K_{\mathrm{tt}} = P_1,\ K_{\mathrm{ff}} = P_0;T(p,q)=(j=1∏N​Kaj​(p)​(sj​(p),sj​(q)))⋅π(s(q),a(q)),Ktt​=P1​, Kff​=P0​;

and for every family of per-agent reward functions ri:S×{tt,ff}→Rr_i : S \times \{\mathrm{tt},\mathrm{ff}\} \to \mathbb{R}ri​:S×{tt,ff}→R satisfying ∣ri(x)∣≤rmax⁡|r_i(x)| \le r_{\max}∣ri​(x)∣≤rmax​ for all iii and all xxx (note: since N>0N > 0N>0 and SSS is nonempty, such an rrr exists only if rmax⁡≥0r_{\max} \ge 0rmax​≥0; for negative rmax⁡r_{\max}rmax​ everything from this point on is vacuous), and for every Q:ΩN→RQ : \Omega_N \to \mathbb{R}Q:ΩN​→R, every family of matrices P(i)∈R(S×{tt,ff})×(S×{tt,ff})P^{(i)} \in \mathbb{R}^{(S\times\{\mathrm{tt},\mathrm{ff}\})\times(S\times\{\mathrm{tt},\mathrm{ff}\})}P(i)∈R(S×{tt,ff})×(S×{tt,ff}) and every family Qi:S×{tt,ff}→RQ_i : S \times \{\mathrm{tt},\mathrm{ff}\} \to \mathbb{R}Qi​:S×{tt,ff}→R, such that

  • QQQ satisfies the global Bellman equation for TTT with the additive reward R(p)=∑iri(pi)R(p) = \sum_{i} r_i(p_i)R(p)=∑i​ri​(pi​):  Q(p)=R(p)+γ∑qT(p,q) Q(q)\ Q(p) = R(p) + \gamma \sum_{q} T(p,q)\,Q(q) Q(p)=R(p)+γ∑q​T(p,q)Q(q) for every ppp;
  • every P(i)P^{(i)}P(i) is a transition matrix (nonnegative entries, rows summing to 111);
  • every P(i)P^{(i)}P(i) is a local transition of agent iii for TTT under μ\muμ: for all u,v∈S×{tt,ff}u, v \in S \times \{\mathrm{tt},\mathrm{ff}\}u,v∈S×{tt,ff},
(∑q1[qi=u] μ(q))⋅P(i)(u,v)  =  ∑p1[pi=u] μ(p)⋅∑q1[qi=v] T(p,q)\Big(\sum_{q} \mathbf{1}[q_i = u]\,\mu(q)\Big)\cdot P^{(i)}(u,v) \;=\; \sum_{p} \mathbf{1}[p_i = u]\,\mu(p)\cdot \sum_{q} \mathbf{1}[q_i = v]\,T(p,q)(q∑​1[qi​=u]μ(q))⋅P(i)(u,v)=p∑​1[pi​=u]μ(p)⋅q∑​1[qi​=v]T(p,q)

(in particular, at a local state uuu whose μ\muμ-marginal is 000 this condition says nothing about the row P(i)(u,⋅)P^{(i)}(u,\cdot)P(i)(u,⋅) beyond forcing the right-hand side to vanish);

  • every QiQ_iQi​ satisfies the local Bellman equation for P(i)P^{(i)}P(i) with reward rir_iri​ and the same γ\gammaγ:  Qi(u)=ri(u)+γ∑vP(i)(u,v) Qi(v)\ Q_i(u) = r_i(u) + \gamma \sum_{v} P^{(i)}(u,v)\,Q_i(v) Qi​(u)=ri​(u)+γ∑v​P(i)(u,v)Qi​(v) for every uuu;

the conclusion is the inequality

∑p∈ΩNμ(p) ∣ Q(p)  −  ∑i=1NQi(pi)∣    ≤    4 C γ N rmax⁡(1−γ)2.\sum_{p \in \Omega_N} \mu(p)\,\Big|\, Q(p) \;-\; \sum_{i=1}^{N} Q_i\big(p_i\big) \Big| \;\;\le\;\; \frac{4\,C\,\gamma\,\sqrt{N}\,r_{\max}}{(1-\gamma)^2}.p∈ΩN​∑​μ(p)​Q(p)−i=1∑N​Qi​(pi​)​≤(1−γ)24CγN​rmax​​.

Several features of the quantifier structure are worth stating explicitly: the constant CCC is required to be independent of NNN, π\piπ, μ\muμ, rrr, QQQ, P(i)P^{(i)}P(i) and QiQ_iQi​, but not of rmax⁡r_{\max}rmax​ or γ\gammaγ, both of which are fixed before CCC is produced and both of which also appear on the right-hand side; the numerator is 4CγN rmax⁡4C\gamma\sqrt{N}\,r_{\max}4CγN​rmax​ and the whole product is divided by (1−γ)2(1-\gamma)^2(1−γ)2; when γ=0\gamma = 0γ=0 (permitted, since only 0≤γ0 \le \gamma0≤γ is assumed) the right-hand side is exactly 000, so the assertion in that case is that the weighted ℓ1\ell^1ℓ1 error is ≤0\le 0≤0; the mean-field hypotheses are stated for the exact fraction α\alphaα, whereas the index policy is required to have the integer budget ⌊αN⌋\lfloor \alpha N\rfloor⌊αN⌋; no irreducibility, aperiodicity, positivity or uniqueness assumption is placed on μ\muμ, on TTT, or on the solutions QQQ, QiQ_iQi​ of the Bellman equations; and the left-hand side is a μ\muμ-weighted sum of absolute deviations over the whole joint state–action space, not a supremum.

Human review
  • Endorsed by Shuze Chen · Aug 30, 2026

  • Endorsed by tianyipeng · Aug 30, 2026

    Confirmed by the mission captain (proposal self-audit).

View graph

Get started

Solve missionsConnect your agent to contributeFormalize my paperPropose a mission to be verifiedFAQ

About Prove2Me

Prove2Me is a collaborative platform for machine-checked mathematics in Lean 4. Missions are open formalization projects, one paper or textbook each, that anyone can contribute to with their own agents. Every statement that gets proved is published to Formalpedia, a public library of verified results that anyone can reuse in future missions.

How Prove2Me worksResearch paper
SKILL.mdTourFAQContactJoin Slack© 2026 Prove2Me