JAO Theorem 5 per initial state, small-diameter regime
OpenBanditAlgorithm.mdp_regret_lower_bound_jao_per_initial_state_small_diameterThe regime , i.e. , of the per-initial-state form of JAO Theorem 5, which the main construction of Section 6 excludes.
JAO handle it in footnote 11 (p. 1582) with a different instance. In this regime the hypothesis forces , so there are plenty of spare actions: one uses of them to connect all states directly, producing an MDP of diameter , and sets up an independent two-state gadget in each state with the remaining actions, exactly one state carrying a better action. The bandit lower bound of Auer et al. (2002b) then gives regret , which suffices because here, so and agree up to a constant.
Note that rewards in this formalisation are a deterministic function of the state-action pair, so the per-state bandit must be built from transition probabilities, as in Figure 3, rather than from stochastic rewards.
On the quantifier. JAO Theorem 5 reads "there is an MDP with states, actions, and diameter , such that for any initial state the expected regret ... is ", i.e. . Section 6 proves something weaker: on p. 1583 it says "we will consider the simpler MDP where all -states are identified. We set this state to be the initial state", and the argument never leaves that state. The two readings are genuinely different, and JAO's construction cannot support the stronger one: let read the initial state, let be its copy, and never leave copy , running a minimax-optimal bandit algorithm over that copy's gadget actions. Staying is possible, since the gadget actions at lead to or stay and returns; and the global optimal gain is attainable inside a single copy. So from , faces exactly the two-state gadget with arms and suffers only , a factor below . Whichever copy carries the planted action, the initial state in that copy is the cheap one.
Nor is it a matter of bookkeeping: the argument needs a plantable action that the reference algorithm plays rarely, and averaging over the actions gives for a fixed start but only for one action good at all starts simultaneously, which turns into .
This node therefore states the per-initial-state form, : the hard MDP may depend on the initial state, which costs nothing, since the construction is simply relabelled so that plays the role of . That is the standard reading of "for any initial state there is a hard MDP", it is exactly what Section 6 establishes, and it is what the upper bound of the same paper is compared against. Note that the stronger form is not known to be false: it quantifies existentially over MDPs, and refuting it would mean showing that every MDP with these parameters admits a cheap initial state. Camping is cheap only because JAO's optimal gain lives inside one copy; an MDP whose optimal gain needs correct actions at many states, or one that mixes in steps, could plausibly satisfy it. Only the route through this construction is closed.
import Mathlib.Data.Real.Sqrt import Mathlib.Analysis.SpecialFunctions.Log.Basic import Definitions.Def_FiniteMDPLearning open MeasureTheory ProbabilityTheory
theorem BanditAlgorithm.mdp_regret_lower_bound_jao_per_initial_state_small_diameter :
∃ C : ℝ, 0 < C ∧
∀ S A T : ℕ, ∀ D : ℝ, 10 ≤ S → 10 ≤ A →
20 * (Real.log S / Real.log A) ≤ D → D * S * A ≤ (T : ℝ) → D < 12 →
∀ π : MDPPolicy S A, ∀ s : Fin S,
∃ M : FiniteMDP S A,
mdpDiameterENN M ≤ ENNReal.ofReal D ∧
C * Real.sqrt (D * S * A * T) ≤
∫ h, mdpRegret M T h ∂(mdpMeasure M (mdpStateDirac s) π T) := by
sorry