JAO Section 6: the planted two-state gadget has optimal average reward at least
ProvedBanditAlgorithm.jao_two_state_gadget_optimal_gain_geThroughout, the MDP is the collapsed two-state gadget of JAO Figure 3, presented by its defining equations rather than through an auxiliary definition: state is with reward , state is with reward , the return probability is for every action , and the escape probability is for every action other than the planted action , for which it is . The gadget has diameter . The reference MDP is the same gadget with no planting.
Statement. For and , the optimal gain of the planted gadget satisfies .
This is the quantity JAO obtain on p. 1586 ("Calculating the stationary distribution, we find that the optimal average reward for the MDP is "). Only the lower bound is stated here, because that is the direction a regret lower bound needs: regret is , so a large is what makes the bound strong, and stating an inequality avoids having to prove that no policy does better.
Proof. The optimal gain is a supremum over initial states and over all policies, so it suffices to exhibit one policy from one state. Take the memoryless deterministic policy that plays the planted action in every state. It induces the two-state Markov chain with escape probability out of and return probability out of ; since , the chain is irreducible and aperiodic with stationary mass on . Because the reward is the indicator of , the expected reward at step started from equals with , so the Cesàro averages converge to from below and the defining the gain equals it.
import Mathlib.Data.Real.Sqrt import Mathlib.Analysis.SpecialFunctions.Log.Basic import Definitions.Def_UCRL2ConfidenceSets open MeasureTheory ProbabilityTheory
theorem BanditAlgorithm.jao_two_state_gadget_optimal_gain_ge
{m : ℕ} (δ ε : ℝ) (hδ0 : 0 < δ) (hδ : δ ≤ 1 / 3) (hε0 : 0 < ε) (hεδ : ε ≤ δ)
(a : Fin m) (M : FiniteMDP 2 m)
(hr0 : ∀ b, M.r 0 b = 0) (hr1 : ∀ b, M.r 1 b = 1)
(hP1 : ∀ b, (M.P 1 b 0 : ℝ) = δ)
(hP0 : ∀ b, (M.P 0 b 1 : ℝ) = δ + (if b = a then ε else 0)) :
(δ + ε) / (2 * δ + ε) ≤ mdpOptimalGain M := by
sorry