JAO Section 6, equations (34)-(37) and Lemma 13, run directly on a two-class MDP with plantable actions: regret from a given initial state
ProvedBanditAlgorithm.jao_planted_two_class_mdp_regret_core_per_initial_stateThe two-class planted gadget. The MDP is described by its defining equations rather than through an auxiliary definition. The data are a class map , an escape map , a return map and a navigation map , and the MDP satisfies
- the reward is the class, for every action ;
- from a class- state every action goes to the class- state with probability and to the class- state otherwise;
- from a class- state every action goes to the class- state with probability and stays at otherwise.
Each row is supported on two states and the two masses sum to , so these equations pin the transition function down completely. For this is the reference MDP ; for the planting at it is . When and this is the gadget of JAO Figure 3 with ; for it is the composite MDP of Figure 4, whose class- states are the and whose navigation map moves between the copies.
Statement. There is a universal with the following property. Let , and , and set . Suppose given a two-class gadget shape on states and actions together with an injection whose image consists of class- pairs, each of which is a fixed point of the navigation map and whose state satisfies . Let be the reference MDP and the MDP planted at . Then for every learning algorithm and every initial state there is a planting with
The planting is chosen after the initial state, which is what makes the statement true; see the discussion of the quantifier on the parent. The initial state is otherwise unrestricted -- it may have either class.
Why not reduce to the two-state gadget. JAO argue the composite by reducing it to the collapsed MDP in which all are identified (p. 1583, "Note that learning this MDP is easier"). That is an intuition they never make precise, and the natural precise version -- an inequality between the two regrets -- is not available: the simulating policy for the collapsed MDP would have to be produced from the composite's policy before the planting is chosen, and it observes neither which copy the composite is in nor which of the actions was played, both of which the composite's policy uses. This statement keeps the argument on the composite itself, where every step of Section 6 goes through unchanged.
Proof. Write for the probability that the state of round has class . Conditioning on one step gives , where is the probability that round plays the planted pair, and the reward is . Comparing with the reference recursion started at the same , the difference obeys and , which telescopes on summation to ; this is equation (34), and unlike JAO's version it needs no stochastic-domination coupling. The reference recursion is solved explicitly, giving from either class, which is equation (35). Since is injective and lands in class- pairs, the counts are disjoint and bounded by the time spent in class , so their reference expectations sum to at most . The planted and reference MDPs differ in exactly one row, and both rows are two-point with masses and , so the divergence decomposition with the Bernoulli relative entropy and Pinsker's inequality give Lemma 13. The optimal gain is at least because the planted pair is a navigation fixed point, so always playing it from the planted state reproduces the two-state gadget. Averaging over the plantings and choosing leaves a positive multiple of .
import Mathlib.Data.Real.Sqrt import Mathlib.Analysis.SpecialFunctions.Log.Basic import Definitions.Def_FiniteMDPLearning open MeasureTheory ProbabilityTheory
theorem BanditAlgorithm.jao_planted_two_class_mdp_regret_core_per_initial_state :
∃ c : ℝ, 0 < c ∧
∀ (S A m : ℕ), 20 ≤ m → ∀ δ : ℝ, 0 < δ → δ ≤ 1 / 3 →
∀ T : ℕ, (16 : ℝ) * m ≤ δ * T →
∀ ε : ℝ, ε = 1 / 5 * Real.sqrt (δ * m / T) →
∀ (ρ : Fin S → ℝ) (up down : Fin S → Fin S) (nav : Fin S → Fin A → Fin S)
(arm : Fin m → Fin S × Fin A)
(M : Fin m → FiniteMDP S A) (M₀ : FiniteMDP S A),
Function.Injective arm →
(∀ i, ρ (arm i).1 = 0) →
(∀ i, nav (arm i).1 (arm i).2 = (arm i).1) →
(∀ i, down (up (arm i).1) = (arm i).1) →
(∀ s, ρ s = 0 ∨ ρ s = 1) →
(∀ s b, M₀.r s b = ρ s) →
(∀ s b, ρ s = 0 →
ρ (up s) = 1 ∧ ρ (nav s b) = 0 ∧ up s ≠ nav s b ∧
(M₀.P s b (up s) : ℝ) = δ ∧
(M₀.P s b (nav s b) : ℝ) = 1 - δ) →
(∀ s b, ρ s = 1 →
ρ (down s) = 0 ∧ down s ≠ s ∧
(M₀.P s b (down s) : ℝ) = δ ∧
(M₀.P s b s : ℝ) = 1 - δ) →
(∀ i, ∀ s b, (M i).r s b = ρ s) →
(∀ i, ∀ s b, ρ s = 0 →
((M i).P s b (up s) : ℝ)
= δ + (if (s, b) = arm i then ε else 0) ∧
((M i).P s b (nav s b) : ℝ)
= 1 - δ - (if (s, b) = arm i then ε else 0)) →
(∀ i, ∀ s b, ρ s = 1 →
((M i).P s b (down s) : ℝ) = δ ∧
((M i).P s b s : ℝ) = 1 - δ) →
∀ π : MDPPolicy S A, ∀ s₀ : Fin S,
∃ i : Fin m,
c * Real.sqrt ((T : ℝ) * m / δ) ≤
∫ h, mdpRegret (M i) T h
∂(mdpMeasure (M i) (mdpStateDirac s₀) π T) := by
sorry