LeanMachineLearning

Learning.klDiv_compProd_compProd_compProd_prodMkLeft_eq_klDiv_comp_compProd🔗

Lemma

From the authors

The divergence of one step of an observation/policy/reward decomposition, in composition-product form: the observation kernel o and the policy π are shared and the reward kernels κ, η (which ignore the history and the observation) differ, so the divergence is the conditional divergence of the reward kernels given the played action, whose law is π ∘ₘ (μ ⊗ₘ o).

Types
  • α : Type u_6mα : MeasurableSpace αA measurable space is a space equipped with a σ-algebra.
  • β : Type u_7mβ : MeasurableSpace β
  • γ : Type u_8mγ : MeasurableSpace γ
  • δ : Type u_9mδ : MeasurableSpace δ
Given
  • μ : MeasureTheory.Measure αA measure is defined to be an outer measure that is countably additive on measurable sets, with the additional assumption that the outer measure is the canonical extension of the restricted measure.MeasureTheory.IsFiniteMeasure μA measure μ is called finite if μ univ < ∞.
  • o : ProbabilityTheory.Kernel α βA kernel from a measurable space α to another measurable space β is a measurable function κ : α → Measure β.ProbabilityTheory.IsMarkovKernel oA kernel is a Markov kernel if every measure in its image is a probability measure.
  • π : ProbabilityTheory.Kernel × β) γProbabilityTheory.IsMarkovKernel π
  • κ : ProbabilityTheory.Kernel γ δProbabilityTheory.IsFiniteKernel κA kernel is finite if every measure in its image is finite, with a uniform bound.
  • η : ProbabilityTheory.Kernel γ δProbabilityTheory.IsFiniteKernel η
Then
InformationTheory.klDiv (μ.compProd (o.compProd (π.compProd (ProbabilityTheory.Kernel.prodMkLeft × β) κ))))
    (μ.compProd (o.compProd (π.compProd (ProbabilityTheory.Kernel.prodMkLeft × β) η)))) =
  InformationTheory.klDiv (((μ.compProd o).bind ⇑π).compProd κ) (((μ.compProd o).bind ⇑π).compProd η)
Code
lemma klDiv_compProd_compProd_compProd_prodMkLeft_eq_klDiv_comp_compProd (μ : Measure α)
    [IsFiniteMeasure μ] (o : Kernel α β) [IsMarkovKernel o] (π : Kernel (α × β) γ)
    [IsMarkovKernel π] (κ η : Kernel γ δ) [IsFiniteKernel κ] [IsFiniteKernel η] :
    klDiv (μ ⊗ₘ (o ⊗ₖ (π ⊗ₖ κ.prodMkLeft (α × β))))
        (μ ⊗ₘ (o ⊗ₖ (π ⊗ₖ η.prodMkLeft (α × β)))) =
      klDiv ((π ∘ₘ (μ ⊗ₘ o)) ⊗ₘ κ) ((π ∘ₘ (μ ⊗ₘ o)) ⊗ₘ η)
Proof
by
  rw [← klDiv_map_measurableEquiv _ _ MeasurableEquiv.prodAssoc.symm, Measure.compProd_assoc,
    Measure.compProd_assoc]
  exact klDiv_compProd_compProd_prodMkLeft_eq_klDiv_comp_compProd _ _ _ _

Meaning last changed in v4.34.0-rc2-76-g565f652 (2026-09-10).

Self-contained, with its dependencies inlined and proofs replaced by sorry: download the raw file · open it in the Lean web editor.

Dependency graph

Nothing to draw. Its statement rests on no other declaration in this project, and names nothing from a package left unaudited — so the graph is this declaration alone. That is the answer, not a missing picture.

Audit surface: 0 project declarations, 17 external constants

✓ Proved: no sorry anywhere in its closure

This is the tool's own reading of one build's recorded axioms, and it is not robust against an author who wants it to pass. Checking meant to be relied on should go through Comparator, which replays the proof through the kernel from an export against an explicit list of permitted axioms.