Repository navigation
feat: the chain rule for the divergence at a stopping time, with equality - #288
Open
RemyDegenne wants to merge 1 commit into
Conversation
…lity Prove that the chain rule at a stopping time is an equality when the hitting time is almost surely finite under both algorithm-environment pairs, for the series, integral and bandit (pull count) forms, through the chain rule at a bounded stopping time for the histories stopped at a fixed time. Co-Authored-By: Claude Code <noreply@github.com> Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
This branch has not been deployed
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
This PR proves that the chain rule for the Kullback–Leibler divergence at a stopping time is an equality when the hitting time of the stopping rule is almost surely finite under both algorithm-environment pairs. It targets Identification, Layer 2.1, "Divergence decomposition at a stopping time": the first bullet's "with equality when τ is also almost surely finite under the second pair" for any two pairs
(alg, env),(alg', env'), and the third bullet's equalityKL = ∑ a, E_ν[N_a(τ)] KL(ν_a, ν'_a)for one algorithm against two stationary bandits. LMLExplore already had the inequalities (Learning.klDiv_stoppedHistMeasure_le_tsum,_le_lintegral,_le). The equalities are stated in the same shape, withstoppedHistMeasureandnotStoppedSet. After this PR, Layer 2.1 lacks only the joint law of the stopped history and the output for two arbitrary pairs; LMLExplore states it only for one algorithm against two bandits (klDiv_histOutputMeasure_le).Roadmap: Identification
The proof of the converse goes through the chain rule at a bounded stopping time, which holds with no finiteness assumption.
Learning.IsAlgEnvSeq.klDiv_map_stoppedProcess_sigmaHistorysays that for any two algorithm-environment sequences, on any probability spaces, the divergence between the laws of the histories stopped at a fixed timeM, the histories of the firstmin M τrounds, equals∑ t < M, KL(μ_t ⊗ₘ stepKernel alg env t ‖ μ_t ⊗ₘ stepKernel alg' env' t). Hereμ_tis the law of thet-round history restricted tonotStoppedSet S t. It is proved by induction onM. The laws of the histories stopped atMand atM + 1split, according to whether roundMis played, into parts with disjoint supports (LML'sklDiv_add_add_of_measure_eq_zero), and Mathlib'sklDiv_compProd_eq_addhandles the step of roundM. Whenτis almost surely finite under both pairs, the history stopped atMis the truncationtruncHist Mof the stopped history. Data processing then bounds every partial sum by the divergence of the stopped histories, which gives the converse ofklDiv_stoppedHistMeasure_le_tsum.New declarations:
LMLExplore/SequentialLearning/DivergenceDecomposition.lean:Learning.IsAlgEnvSeq.klDiv_map_stoppedProcess_sigmaHistory,Learning.klDiv_stoppedHistMeasure_eq_tsum,Learning.klDiv_stoppedHistMeasure_eq_lintegralandLearning.klDiv_stoppedHistMeasure_eq. The existing_le_lintegraland_lenow share with them two private series identities instead of repeating the computation. Their statements are unchanged.LMLExplore/SequentialLearning/StoppedHistory.leanadds the histories stopped at a fixed time:Learning.truncHist, built on Mathlib'sFin.take, withtruncHist_def,measurable_truncHist,truncHist_sigmaHistoryandtruncHist_stoppedValue_sigmaHistory;measurable_stoppedProcess_sigmaHistory,map_stoppedProcess_sigmaHistory_eq_map_truncHistandstoppedProcess_sigmaHistory_zero;stoppedProcess_sigmaHistory_{of,succ_of}_mem_notStoppedSet,stoppedProcess_sigmaHistory_succ_of_notMem_notStoppedSetandstoppedProcess_sigmaHistory_notMem_image_notStoppedSet;map_stoppedProcess_sigmaHistory_eq_addandmap_stoppedProcess_sigmaHistory_succ_eq_add, the support lemmasmap_restrict_stoppedProcess_sigmaHistory_apply_image_notStoppedSetandmap_stoppedProcess_sigmaHistory_apply_compl_fst_le, andIsAlgEnvSeq.map_restrict_history_succ.LMLExplore/Probability/HasCondDistrib.lean:ProbabilityTheory.HasCondDistrib.restrict_preimage, which says a conditional law givenXsurvives restricting the measure to an eventX ⁻¹' s.No public declaration is removed or renamed.
This is a port of item R19 of the paper project RemyDegenne/Essakine2026Tight, at commit
73a8c16c42ec1c41f8d38c5d2af232f36c2cff15, under the Apache 2.0 license. It adaptsForLMLExplore/InformationTheory/KLStoppedPrefix.lean,ForLMLExplore/Probability/Process/SigmaPrefix.leanandForLMLExplore/SequentialLearning/DivergenceDecomposition.lean, andHasCondDistrib.restrict_preimagefromForLMLExplore/Probability/CondDistrib.lean(item R4, the prerequisite that R19 uses). The source proves the chain rule for a general process along its prefix filtration, through newfinPrefix/sigmaPrefix/prefixFiltration/piFinSuccProdAPIs. This port specializes the argument to LMLExplore'ssigmaHistory,notStoppedSetand LML'shistory_succ/finSuccProd, so it does not add a parallel API for prefixes of histories. It gets the converse inequality from the existingklDiv_stoppedHistMeasure_le_tsumand data processing through the truncation, so it does not need monotone convergence. It also states the result for two arbitrary pairs(alg, env),(alg', env'), as Layer 2.1 asks, where the source has one algorithm against two environments. The files credit the source. Mathematical references: [KCG16, Lemma 1], [GMS19, Section 2], already cited inDivergenceDecomposition.lean.Paper: RemyDegenne/Essakine2026Tight
Item: R19
🤖 Prepared with Claude Code