JAO Section 6: the composite MDP has diameter and its regret dominates that of the collapsed two-state MDP
OpenBanditAlgorithm.jao_composite_dominates_collapsedFix and , and set and . Then every learning algorithm for -state, -action MDPs induces an algorithm for the two-state, -action gadget such that, for every planting with and every gadget with those parameters, there is an MDP with states, actions and diameter at most whose expected regret under , from every initial state, is at least the expected regret of under from .
Here M is the 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 except the single planted action , for which it is . The gadget has diameter , so the conclusion is the of the paper with .
This is the construction-and-reduction half of Section 6 of Jaksch, Ortner and Auer (2010), stated so that it composes with the probabilistic core. The MDP consists of copies of the gadget, exactly one carrying the planted action, joined into a single communicating MDP: further actions per state, with deterministic transitions that do not leave the -states, induce an -ary tree on the -states, one action moving toward the root and toward the leaves, all with reward zero. Every state can reach the root in at most steps beyond the gadget's own , so the diameter is at most , which is at most because . The action budget is met because by the choice , and the state budget because .
The domination is JAO's observation (p. 1583) that one may analyse the simpler MDP in which all -states are identified. That MDP is a single gadget with actions; learning it is easier, since the learner may switch between copies at no cost, while its optimal average reward is unchanged. Hence regret on is at least regret on , and this holds from any initial state of , since starting away from the identified state only forces the learner to travel and so cannot reduce regret.
The quantifier order matters and is the one the composition needs: is produced from alone, before the planting is chosen, because the core must be applied to in order to select ; only then is the composite built.
import Mathlib.Data.Real.Sqrt import Mathlib.Analysis.SpecialFunctions.Log.Basic import Definitions.Def_FiniteMDPLearning open MeasureTheory ProbabilityTheory
theorem BanditAlgorithm.jao_composite_dominates_collapsed :
∀ S A : ℕ, ∀ D : ℝ, 10 ≤ S → 10 ≤ A →
20 * (Real.log S / Real.log A) ≤ D → 12 ≤ D →
∀ T : ℕ, ∀ π : MDPPolicy S A,
∃ π' : MDPPolicy 2 (S / 2 * ((A - 1) / 2)),
∀ (a : Fin (S / 2 * ((A - 1) / 2))) (ε : ℝ)
(M' : FiniteMDP 2 (S / 2 * ((A - 1) / 2))),
0 < ε → ε ≤ 4 / D →
(∀ b, M'.r 0 b = 0) → (∀ b, M'.r 1 b = 1) →
(∀ b, (M'.P 1 b 0 : ℝ) = 4 / D) →
(∀ b, (M'.P 0 b 1 : ℝ) = 4 / D + (if b = a then ε else 0)) →
∃ M : FiniteMDP S A,
mdpDiameterENN M ≤ ENNReal.ofReal D ∧
∀ s : Fin S,
(∫ h, mdpRegret M' T h ∂(mdpMeasure M' (mdpStateDirac 0) π' T)) ≤
∫ h, mdpRegret M T h ∂(mdpMeasure M (mdpStateDirac s) π T) := by
sorry