Documentation

TauCeti.MeasureTheory.OptimalTransport.Wasserstein.Space

Finite-distance Wasserstein spaces #

The Wasserstein distance can be infinite, even between probability measures. This file isolates the parts on which it is finite and hence defines an ordinary pseudometric:

Every pseudoemetric and pseudometric structure below needs 1 ≤ p, since that is the range in which the Wasserstein triangle inequality holds; the abstract constructors take it as an explicit hypothesis and the instances read it off Fact (1 ≤ p). For such p, on a standard Borel extended pseudometric space with measurable ground distance, every anchored component carries the Wasserstein pseudoemetric, and hence the real-valued Wasserstein pseudometric, because distances inside a single component are finite. On an ordinary pseudometric space, finite moment about one point is equivalent to finite Wasserstein distance from the Dirac mass at any point. Consequently WassersteinSpace p X is canonically equivalent, after choosing a point x₀, to the component anchored at δ_[x₀], and it carries the same Wasserstein pseudometric without a basepoint appearing in its definition. More generally, every finite-moment probability measure anchors exactly the same carrier; this identification is measurable in both directions and is an isometry for the Wasserstein pseudometrics.

When moreover the ground space is Polish and its distance separates points, both pseudometrics are metrics. The measurable structures are the subtype structures inherited from ProbabilityMeasure; no second measurable-space instance is introduced. Identifying these measurable structures with the Borel sigma algebras of the Wasserstein metrics is a separate topological result.

Neither carrier exposes its subtype body: elements are built with TauCeti.WassersteinComponent.mk and TauCeti.WassersteinSpace.mk, and taken apart with toProbabilityMeasure together with TauCeti.WassersteinComponent.wassersteinEDist_anchor_ne_top and TauCeti.WassersteinSpace.hasFiniteMoment. The identification equivComponent leaves the underlying probability measure unchanged, hence is an isometry for the two Wasserstein pseudometrics and is measurable in both directions.

Main definitions #

References #

Probability measures at finite p-Wasserstein distance from the anchor μ₀.

This is a type synonym for a subtype, so it can carry the Wasserstein topology independently of the weak topology already present on ProbabilityMeasure. Its measurable structure is still the canonical subtype structure. The subtype body is not exposed: use TauCeti.WassersteinComponent.mk and TauCeti.WassersteinComponent.toProbabilityMeasure.

Instances For

    Probability measures on X with finite p-moment.

    As for TauCeti.WassersteinComponent, the subtype body is not exposed: use TauCeti.WassersteinSpace.mk and TauCeti.WassersteinSpace.toProbabilityMeasure.

    Instances For

      The element of the component anchored at μ₀ given by a probability measure at finite p-Wasserstein distance from μ₀.

      Instances For

        The probability measure underlying an element of a Wasserstein component.

        Instances For

          The inclusion of a Wasserstein component into probability measures is injective.

          theorem TauCeti.WassersteinComponent.ext {X : Type u} {p : ENNReal} [MeasurableSpace X] [PseudoEMetricSpace X] {μ₀ : MeasureTheory.ProbabilityMeasure X} {μ ν : WassersteinComponent p μ₀} (h : ↑μ = ↑ν) :
          μ = ν

          Two elements of a Wasserstein component are equal when their probability measures are equal.

          @[instance_reducible]

          A Wasserstein component inherits the Giry measurable structure from probability measures.

          The inclusion of a Wasserstein component into probability measures is measurable.

          theorem TauCeti.WassersteinComponent.measurable_mk {X : Type u} {p : ENNReal} [MeasurableSpace X] [PseudoEMetricSpace X] {μ₀ : MeasureTheory.ProbabilityMeasure X} {Y : Type u_1} [MeasurableSpace Y] {f : Y → MeasureTheory.ProbabilityMeasure X} (hf : Measurable f) (hmem : ∀ (y : Y), wassersteinEDist p ↑μ₀ ↑(f y) ≠ ⊤) :
          Measurable fun (y : Y) => mk (f y) ⋯

          A measurably varying family of probability measures in an anchored component gives a measurable map into that component.

          Every element of the component anchored at μ₀ is at finite Wasserstein distance from μ₀; together with TauCeti.WassersteinComponent.mk this characterises the elements of the component.

          @[instance_reducible]

          The Wasserstein extended distance equips every anchored finite-distance component with a pseudoemetric space structure whenever the ground extended distance is measurable and 1 ≤ p.

          Instances For
            @[instance_reducible]

            The canonical Wasserstein pseudoemetric on an anchored component when the ground extended distance is Borel measurable.

            @[simp]

            The extended distance on a Wasserstein component is the Wasserstein distance of the underlying probability measures.

            Any two laws in one anchored Wasserstein component are at finite distance from one another.

            @[instance_reducible]

            Every anchored finite-distance component carries the real-valued Wasserstein pseudometric.

            @[simp]

            The distance on a Wasserstein component is the real value of the Wasserstein extended distance.

            @[instance_reducible]

            On a Polish metric ground space, Wasserstein distance separates laws, so every anchored finite-distance component is a metric space.

            The element of WassersteinSpace p X given by a probability measure of finite p-moment.

            Instances For

              The probability measure underlying an element of WassersteinSpace p X.

              Instances For

                The inclusion of finite-moment laws into probability measures is injective.

                theorem TauCeti.WassersteinSpace.ext {X : Type u} {p : ENNReal} [MeasurableSpace X] [PseudoEMetricSpace X] {μ ν : WassersteinSpace p X} (h : ↑μ = ↑ν) :
                μ = ν

                Two finite-moment laws are equal when their probability measures are equal.

                theorem TauCeti.WassersteinSpace.ext_iff {X : Type u} {p : ENNReal} [MeasurableSpace X] [PseudoEMetricSpace X] {μ ν : WassersteinSpace p X} :
                μ = ν ↔ ↑μ = ↑ν
                @[instance_reducible]

                WassersteinSpace p X inherits the Giry measurable structure from probability measures.

                The inclusion of finite-moment laws into probability measures is measurable.

                theorem TauCeti.WassersteinSpace.measurable_mk {X : Type u} {p : ENNReal} [MeasurableSpace X] [PseudoEMetricSpace X] {Y : Type u_1} [MeasurableSpace Y] {f : Y → MeasureTheory.ProbabilityMeasure X} (hf : Measurable f) (hmem : ∀ (y : Y), HasFiniteMoment p ↑(f y)) :
                Measurable fun (y : Y) => mk (f y) ⋯

                A measurably varying family of finite-moment probability measures gives a measurable map into the Wasserstein space.

                Every law in WassersteinSpace p X has finite p-moment; together with TauCeti.WassersteinSpace.mk this characterises its elements.

                @[simp]

                The Wasserstein space over an empty ground space is empty: there is no probability measure.

                Choosing a basepoint identifies the finite-moment Wasserstein space with the finite-distance component anchored at its Dirac law. The underlying probability measure is unchanged.

                Instances For

                  The identification with a Dirac-anchored component is measurable: it does not change the underlying probability measure, and both sides carry the subtype measurable structure.

                  The inverse of the identification with a Dirac-anchored component is measurable.

                  A finite-moment anchor identifies the usual finite-moment Wasserstein space with its finite-distance component. The equivalence leaves the underlying probability measure unchanged, hence is measurable in both directions: both carriers inherit the measurable structure of ProbabilityMeasure.

                  Instances For
                    theorem TauCeti.WassersteinSpace.wassersteinEDist_ne_top {X : Type u} {p : ENNReal} [MeasurableSpace X] [PseudoMetricSpace X] (hd : Measurable fun (z : X × X) => edist z.1 z.2) (μ ν : WassersteinSpace p X) :
                    wassersteinEDist p ↑↑μ ↑↑ν ≠ ⊤

                    Two finite-moment laws are at finite Wasserstein distance: this is the general finite-moment anchor criterion with the source law as the anchor.

                    theorem TauCeti.WassersteinSpace.exists_coe_eq_smul_add_smul {X : Type u} {p : ENNReal} [MeasurableSpace X] [PseudoMetricSpace X] [OpensMeasurableSpace X] {a b : ENNReal} (hab : a + b = 1) (μ ν : WassersteinSpace p X) :
                    ∃ (ρ : WassersteinSpace p X), ↑↑ρ = a • ↑↑μ + b • ↑↑ν

                    P_p(X) is convex. For weights with a + b = 1, the mixture a • μ + b • ν of two laws of P_p(X) is again a law of P_p(X).

                    theorem TauCeti.WassersteinSpace.exists_coe_eq_cond {X : Type u} {p : ENNReal} [MeasurableSpace X] [PseudoMetricSpace X] [OpensMeasurableSpace X] {m : MeasureTheory.Measure X} {s : Set X} (hs : MeasurableSet s) (hb : Bornology.IsBounded s) (h₀ : m s ≠ 0) (hfin : m s ≠ ⊤) :
                    ∃ (μ : WassersteinSpace p X), ↑↑μ = m[|s] ∧ Bornology.IsBounded (↑↑μ).support ∧ (↑↑μ).AbsolutelyContinuous m

                    Uniform laws on bounded sets lie in P_p(X). The normalised restriction m[|s] of a measure m to a bounded measurable set s of positive, finite measure is a law of P_p(X). It is carried by s, so it has bounded support and is absolutely continuous with respect to m.

                    theorem TauCeti.WassersteinSpace.sum_mul_wassersteinEDist_rpow_ne_top {X : Type u} {p : ENNReal} [MeasurableSpace X] [PseudoMetricSpace X] [BorelSpace X] [SecondCountableTopology X] {J : Type u_1} [Fintype J] {w : J → NNReal} (μ : J → WassersteinSpace p X) (z : WassersteinSpace p X) :
                    ∑ j : J, ↑(w j) * wassersteinEDist p ↑↑z ↑↑(μ j) ^ p.toReal ≠ ⊤

                    The weighted p-Wasserstein cost against finitely many points of P_p(X) is finite at every point of P_p(X).

                    @[instance_reducible]

                    The Wasserstein extended distance equips finite-moment laws with a pseudoemetric space structure whenever the ground distance is measurable and 1 ≤ p.

                    Instances For
                      @[instance_reducible]

                      The canonical Wasserstein pseudoemetric on finite-moment laws when the ground distance is Borel measurable.

                      @[simp]

                      The extended distance on finite-moment laws is their Wasserstein distance.

                      @[instance_reducible]

                      Finite-moment laws carry the real-valued Wasserstein pseudometric.

                      @[simp]

                      The distance on finite-moment laws is the real value of the Wasserstein extended distance.

                      The identification with a Dirac-anchored component preserves the Wasserstein extended distance, since it does not change the underlying probability measure.

                      Choosing a basepoint identifies WassersteinSpace p X isometrically with the finite-distance component anchored at the Dirac law of that basepoint.

                      Choosing a basepoint gives an isometric equivalence from WassersteinSpace p X to the finite-distance component anchored at the Dirac law of that basepoint.

                      Instances For

                        The identification with a Dirac-anchored component preserves the real-valued Wasserstein distance.

                        The identification with a finite-moment anchored component preserves the Wasserstein extended distance.

                        The identification with a finite-moment anchored component is an isometry.

                        A finite-moment anchor gives an isometric equivalence from P_p(X) to its finite-distance component.

                        Instances For

                          The finite-moment-anchor identification preserves the real-valued Wasserstein distance.

                          @[instance_reducible]

                          On a Polish metric ground space, the Wasserstein pseudometric on finite-moment laws is a metric.

                          Laws with finite second moment have finite transport cost for ‖x - y‖ ^ 2 / 2: the cost is at most ‖x - y‖ ^ 2, whose transport cost is the square of their finite 2-Wasserstein distance.

                          theorem TauCeti.WassersteinSpace.dist_sq_le_integral {E : Type u_1} [NormedAddCommGroup E] [MeasurableSpace E] [BorelSpace E] [SecondCountableTopology E] [CompleteSpace E] {ν₀ ν₁ : WassersteinSpace 2 E} {π : MeasureTheory.Measure (E × E)} (hπ : IsCoupling π ↑↑ν₀ ↑↑ν₁) :
                          dist ν₀ ν₁ ^ 2 ≤ ∫ (z : E × E), ‖z.1 - z.2‖ ^ 2 ∂π

                          The squared quadratic Wasserstein distance between two laws of P₂(E) is at most the quadratic cost ∫ ‖x - y‖² dπ of any of their couplings π.

                          theorem TauCeti.WassersteinSpace.dist_sq_eq_integral {E : Type u_1} [NormedAddCommGroup E] [MeasurableSpace E] [BorelSpace E] [SecondCountableTopology E] [CompleteSpace E] {ν₀ ν₁ : WassersteinSpace 2 E} {π : MeasureTheory.Measure (E × E)} (hπ : IsOptimalCoupling (fun (z : E × E) => edist z.1 z.2 ^ 2) π ↑↑ν₀ ↑↑ν₁) :
                          dist ν₀ ν₁ ^ 2 = ∫ (z : E × E), ‖z.1 - z.2‖ ^ 2 ∂π

                          An optimal coupling π of two laws of P₂(E) for the quadratic cost realizes their squared quadratic Wasserstein distance as its quadratic cost: dist ν₀ ν₁ ^ 2 = ∫ ‖x - y‖² dπ.

                          theorem TauCeti.WassersteinSpace.isOptimalCoupling_of_integral_le_dist_sq {E : Type u_1} [NormedAddCommGroup E] [MeasurableSpace E] [BorelSpace E] [SecondCountableTopology E] [CompleteSpace E] {ν₀ ν₁ : WassersteinSpace 2 E} {π : MeasureTheory.Measure (E × E)} (hπ : IsCoupling π ↑↑ν₀ ↑↑ν₁) (h : ∫ (z : E × E), ‖z.1 - z.2‖ ^ 2 ∂π ≤ dist ν₀ ν₁ ^ 2) :
                          IsOptimalCoupling (fun (z : E × E) => edist z.1 z.2 ^ 2) π ↑↑ν₀ ↑↑ν₁

                          A coupling π of two laws of P₂(E) whose quadratic cost ∫ ‖x - y‖² dπ is at most their squared quadratic Wasserstein distance is an optimal coupling for the quadratic cost. This is the converse of TauCeti.WassersteinSpace.dist_sq_eq_integral.

                          theorem TauCeti.WassersteinSpace.norm_toLp_sub_eq_dist {E : Type u_1} [NormedAddCommGroup E] [MeasurableSpace E] [BorelSpace E] [SecondCountableTopology E] [CompleteSpace E] {ν₀ ν₁ : WassersteinSpace 2 E} {π : MeasureTheory.Measure (E × E)} (hπ : IsOptimalCoupling (fun (z : E × E) => edist z.1 z.2 ^ 2) π ↑↑ν₀ ↑↑ν₁) (hd : MeasureTheory.MemLp (fun (z : E × E) => z.2 - z.1) 2 π) :
                          ‖MeasureTheory.MemLp.toLp (fun (z : E × E) => z.2 - z.1) hd‖ = dist ν₀ ν₁

                          The L²(π) norm of the displacement (x, y) ↦ y - x of an optimal coupling π of two laws of P₂(E) for the quadratic cost is their quadratic Wasserstein distance.