JAO Section 6: a two-class planted gadget has optimal average reward at least
ProvedBanditAlgorithm.jao_two_class_gadget_optimal_gain_geThe two-class planted gadget. The MDP is given 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 satisfies: the reward is the class, ; 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. Every row is supported on two distinct states whose masses sum to , so the transition function is pinned down completely. The reference MDP is the same shape with . For and this is JAO Figure 3 with ; for it is the composite of Figure 4, its class- states being the .
The whole of JAO Section 6 is run at this level of generality, on the composite MDP itself rather than on the collapsed two-state MDP they pass to on p. 1583. That reduction is not available as an inequality between regrets: the simulating policy would have to be produced before the planting is chosen, and it sees neither which copy the composite is in nor which of the actions was played. Working with the class in place of the state avoids it, and no step of the argument is lost.
Throughout, the initial state is arbitrary and of either class.
Statement. For , , and a planted pair at a class- state which is a fixed point of the navigation map, , and satisfies , the optimal gain satisfies .
This is the quantity JAO compute on p. 1586 ("Calculating the stationary distribution, we find that the optimal average reward for the MDP is "). Only the lower bound is needed: regret is , so a large is what makes the bound strong.
Proof. The optimal gain is a supremum over initial states and over all policies, so one policy from one state suffices. Take the memoryless deterministic policy playing everywhere, started at . The two hypotheses on the planted pair are exactly what confines the resulting chain to : the navigation fixed point keeps a failed escape at , and returns there. So the chain is the two-state gadget with escape probability and return probability , whose stationary mass on the class- state is . Writing for the probability that round is in class , one step gives with ; since the reward is the class indicator, the expected reward over rounds is the partial sum of , so it is minus a bounded geometric remainder, the Cesàro averages converge to , and the defining the gain equals it. Both suprema are then entered with le_ciSup, the boundedness coming from every reward lying in .
import Mathlib.Data.Real.Sqrt import Mathlib.Analysis.SpecialFunctions.Log.Basic import Definitions.Def_FiniteMDPLearning open MeasureTheory ProbabilityTheory
theorem BanditAlgorithm.jao_two_class_gadget_optimal_gain_ge {S A : ℕ}
(δ ε : ℝ) (hδ0 : 0 < δ) (hδ : δ ≤ 1 / 3) (hε0 : 0 < ε) (hεδ : ε ≤ δ)
(ρ : Fin S → ℝ) (up down : Fin S → Fin S) (nav : Fin S → Fin A → Fin S)
(sStar : Fin S) (bStar : Fin A) (M : FiniteMDP S A)
(hstar : ρ sStar = 0) (hfix : nav sStar bStar = sStar)
(hud : down (up sStar) = sStar)
(hrM : ∀ s b, M.r s b = ρ s)
(hrow0M : ∀ s b, ρ s = 0 →
ρ (up s) = 1 ∧ ρ (nav s b) = 0 ∧ up s ≠ nav s b ∧
(M.P s b (up s) : ℝ) = δ + (if (s, b) = (sStar, bStar) then ε else 0) ∧
(M.P s b (nav s b) : ℝ) = 1 - δ - (if (s, b) = (sStar, bStar) then ε else 0))
(hrow1M : ∀ s b, ρ s = 1 →
ρ (down s) = 0 ∧ down s ≠ s ∧
(M.P s b (down s) : ℝ) = δ ∧ (M.P s b s : ℝ) = 1 - δ) :
(δ + ε) / (2 * δ + ε) ≤ mdpOptimalGain M := by
sorry