JAO equation (34): the reward collected in the planted gadget is at most
ProvedBanditAlgorithm.jao_two_state_gadget_reward_le_reference_plus_planted_playsThroughout, 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 every horizon and every policy , writing for the number of rounds spent in (which, because the reward is the indicator of , is exactly the total reward collected) and for the number of rounds in which the planted action is played in state ,
Here is the expectation under the planted gadget and that under the reference gadget , both run against the same policy from , and .
This is equation (34) of JAO (p. 1583), in the form in which the rest of the argument consumes it. The proof has three steps. First, a one-step conditioning identity: summing over and splitting on the previous state gives
whence . Second, playing the planted action can only reduce the probability of remaining in , so ; this is a coupling between the two gadgets under the common policy. Third, identically, so .
import Mathlib.Data.Real.Sqrt import Mathlib.Analysis.SpecialFunctions.Log.Basic import Definitions.Def_UCRL2ConfidenceSets open MeasureTheory ProbabilityTheory
theorem BanditAlgorithm.jao_two_state_gadget_reward_le_reference_plus_planted_plays
{m : ℕ} (δ ε : ℝ) (hδ0 : 0 < δ) (hδ : δ ≤ 1 / 3) (hε0 : 0 < ε) (hεδ : ε ≤ δ)
(a : Fin m) (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))
(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 : ℝ) = δ)
(T : ℕ) (π : MDPPolicy 2 m) :
(∫ h, mdpTrajectoryReward M h ∂(mdpMeasure M (mdpStateDirac 0) π T))
≤ (T : ℝ)
- (∫ h, mdpTrajectoryReward M₀ h ∂(mdpMeasure M₀ (mdpStateDirac 0) π T))
+ (ε / δ)
* ∫ h, (mdpVisitCount h T 0 a : ℝ) ∂(mdpMeasure M (mdpStateDirac 0) π T) := by
sorry