Documentation

TauCeti.MeasureTheory.OptimalTransport.Disintegration

Disintegrating transport plans #

A coupling can be read in two equivalent ways: as a measure on a product with prescribed marginals, or as a source measure followed by a probability kernel whose averaged law is the target. This file connects the coupling API to Mathlib's composition-product and regular conditional-kernel APIs.

For a Markov kernel κ : Kernel X Y, the composition-product μ ⊗ₘ κ couples μ to the averaged law κ ∘ₘ μ. Conversely, when the target space is standard Borel, a finite coupling π reconstructs as μ ⊗ₘ π.condKernel, and its conditional kernel averages to the prescribed target. Any other finite kernel reconstructing the same plan agrees with π.condKernel μ-almost everywhere.

When the plan is concentrated on a set that meets almost every vertical line in at most one point, the conditional kernel is almost everywhere a Dirac mass, so the plan is determined by its source marginal and that set; this is the form in which uniqueness of contact partners becomes uniqueness of optimal plans.

Main statements #

This is Layer 0, item 3 of the optimal-transport roadmap. The construction and uniqueness theorem consume Mathlib's MeasureTheory.Measure.disintegrate and ProbabilityTheory.eq_condKernel_of_measure_eq_compProd; no disintegration is rebuilt here.

References #

The composition-product of a measure and a Markov kernel couples the measure to the law obtained by averaging the kernel against it. This is the unbundled reconstruction of a joint law from a marginal and a conditional probability kernel.

@[simp]

A composition-product has prescribed marginals μ and ν exactly when averaging its Markov kernel against μ gives ν.

A finite coupling is reconstructed by composing its source marginal with Mathlib's regular conditional kernel of the second coordinate given the first.

Averaging the conditional kernel of a finite coupling against its source marginal gives its target marginal.

A finite kernel agrees almost everywhere with the regular conditional kernel of a finite coupling exactly when its composition-product with the source marginal reconstructs that coupling. This is the coupling-facing uniqueness theorem for disintegration.

The conditional distribution of the second coordinate given the first reconstructs a finite coupling from its source marginal. This is the coordinate-map form of TauCeti.IsCoupling.compProd_condKernel.

The conditional distribution of the second coordinate given the first averages against the source marginal to the target marginal.

For a finite coupling, the conditional distribution of the second coordinate given the first is almost everywhere the same kernel as Mathlib's regular conditional kernel of the joint measure.

theorem TauCeti.IsCoupling.ae_exists_condKernel_eq_dirac {X : Type u} {Y : Type v} [MeasurableSpace X] [MeasurableSpace Y] {π : MeasureTheory.Measure (X × Y)} {μ : MeasureTheory.Measure X} {ν : MeasureTheory.Measure Y} [StandardBorelSpace Y] [Nonempty Y] [MeasureTheory.IsFiniteMeasure π] (hπ : IsCoupling π μ ν) {s : Set (X × Y)} (hs : ∀ᵐ (z : X × Y) ∂π, z ∈ s) (hsub : ∀ᵐ (x : X) ∂μ, {y : Y | (x, y) ∈ s}.Subsingleton) :
∀ᵐ (x : X) ∂μ, ∃ (y : Y), (x, y) ∈ s ∧ π.condKernel x = MeasureTheory.Measure.dirac y

A finite coupling concentrated on a set whose sections {y | (x, y) ∈ s} are μ-almost all subsingletons is deterministic: for μ-almost every x, its conditional kernel at x is the Dirac mass at the unique point of the section. No measurability of s is needed.

theorem TauCeti.IsCoupling.eq_of_ae_mem_of_subsingleton {X : Type u} {Y : Type v} [MeasurableSpace X] [MeasurableSpace Y] {π : MeasureTheory.Measure (X × Y)} {μ : MeasureTheory.Measure X} {ν : MeasureTheory.Measure Y} [StandardBorelSpace Y] [Nonempty Y] [MeasureTheory.IsFiniteMeasure π] {π' : MeasureTheory.Measure (X × Y)} {ν' : MeasureTheory.Measure Y} (hπ : IsCoupling π μ ν) (hπ' : IsCoupling π' μ ν') {s : Set (X × Y)} (hs : ∀ᵐ (z : X × Y) ∂π, z ∈ s) (hs' : ∀ᵐ (z : X × Y) ∂π', z ∈ s) (hsub : ∀ᵐ (x : X) ∂μ, {y : Y | (x, y) ∈ s}.Subsingleton) :
π = π'

Two finite couplings with the same source marginal, concentrated on a common set whose sections are μ-almost all subsingletons, are equal. In particular their target marginals agree.

@[simp]

The underlying measure of a bundled probability coupling is reconstructed from its source law and regular conditional kernel.

@[simp]

The regular conditional kernel of a bundled probability coupling averages to its prescribed target law.

A finite kernel is a version of the regular conditional kernel of a bundled probability coupling exactly when it reconstructs the coupling from its source law.

@[simp]

The coordinate conditional distribution of a bundled probability coupling reconstructs its underlying joint law.

@[simp]

The coordinate conditional distribution of a bundled probability coupling averages to its target law.

The coordinate conditional distribution and regular conditional kernel of a bundled probability coupling agree almost everywhere under its source law.

theorem TauCeti.Coupling.ae_exists_condKernel_eq_dirac {X : Type u} {Y : Type v} [MeasurableSpace X] [MeasurableSpace Y] {μ : MeasureTheory.ProbabilityMeasure X} {ν : MeasureTheory.ProbabilityMeasure Y} [StandardBorelSpace Y] [Nonempty Y] (π : Coupling μ ν) {s : Set (X × Y)} (hs : ∀ᵐ (z : X × Y) ∂↑↑π, z ∈ s) (hsub : ∀ᵐ (x : X) ∂↑μ, {y : Y | (x, y) ∈ s}.Subsingleton) :
∀ᵐ (x : X) ∂↑μ, ∃ (y : Y), (x, y) ∈ s ∧ (↑↑π).condKernel x = MeasureTheory.Measure.dirac y

A bundled coupling concentrated on a set with almost everywhere subsingleton sections has Dirac conditional laws at points of those sections.

theorem TauCeti.Coupling.eq_of_ae_mem_of_subsingleton {X : Type u} {Y : Type v} [MeasurableSpace X] [MeasurableSpace Y] {μ : MeasureTheory.ProbabilityMeasure X} {ν : MeasureTheory.ProbabilityMeasure Y} [StandardBorelSpace Y] [Nonempty Y] {ν' : MeasureTheory.ProbabilityMeasure Y} (π : Coupling μ ν) (π' : Coupling μ ν') {s : Set (X × Y)} (hs : ∀ᵐ (z : X × Y) ∂↑↑π, z ∈ s) (hs' : ∀ᵐ (z : X × Y) ∂↑↑π', z ∈ s) (hsub : ∀ᵐ (x : X) ∂↑μ, {y : Y | (x, y) ∈ s}.Subsingleton) :
↑↑π = ↑↑π'

Bundled couplings with the same source and almost everywhere the same uniquely determined partner in a set have equal underlying plan measures.