Samplinglib
Lean gate not recorded for this source state main · 9bb7a7394ab0

SPHMC first Picard Gaussian input law

Fan Chen, Sinho Chewi, Jianfeng Lu and Matthew S. Zhang. Samplinglib supplies omitted measure-theoretic details; this is not the authors' original wording or a complete D.7 claim.

arXiv:2609.06906v1 · Algorithm 3.1 lines 282–300 and Lemma D.4 lines 2177–2181 · faithful paraphrase

Complete source statement (Samplinglib restatement)

Construct the actual independent first refresh noise and finite first-layer Picard Gaussian array, with all measurable and second-moment facts needed by the existing layerwise bound.

  • Incoming phase-state probability law with integrable energy bounded by M≥0.
  • h≥0, finite Picard index, finite-dimensional real inner-product Borel state space.

Input law

\[\mu=(\nu\otimes\gamma_d)\otimes\gamma_d^{\otimes J}.\]

Moments

\[\mathbb E_\mu[\|X-x_*\|^2+\|P_0\|^2]\le M+(1-e^{-h})d,\qquad\mathbb E_\mu\|G_j^{[0]}\|^2=d.\]

Read the formalized proofs

Each statement and proof below has its own closed Lean disclosure. Samplinglib parents, Mathlib calls and external mathematical sources are distinguished in each proof.

Samplinglib mathematical lesson

The actual independent Gaussian inputs for the first Picard layer

AutoSamplingTheory.ExampleCases.SmoothedPicardHMC.PicardInputLaw.partial_refresh_with_picard_innovations · theorem · Teaching coverage

Read this first

Mathematical statement

On the nested product of the incoming law, a fresh standard Gaussian, and a finite product of standard Gaussians, the refreshed X and P0 and every Picard innovation G_j are measurable and square-integrable with their exact consumer-ready moment bounds.

\[B_{\rm refresh}=M+(1-e^{-h})d,\qquad \mathbb E\|G_j^{[0]}\|^2=d.\]
Lean statement · partial_refresh_with_picard_innovations

The nested lets expose the exact probability law and random variables, while the conjunction returns every measurability and integrability premise required downstream.

Braces mark parameters Lean can infer; square brackets request structures such as a measurable space or probability measure. Named hypotheses are mathematical premises, not facts established by this declaration. Section parameters are described in the mathematical hypotheses above; the module link retains their exact source context.

theorem partial_refresh_with_picard_innovations
    (ν : Measure (E × E)) [IsProbabilityMeasure ν]
    (xstar : E) {h M : ℝ} (hh : 0 ≤ h) (hM : 0 ≤ M)
    (hstateI : Integrable (fun s : E × E => ‖s.1 - xstar‖ ^ 2 + ‖s.2‖ ^ 2) ν)
    (hstate : (∫ s : E × E, ‖s.1 - xstar‖ ^ 2 + ‖s.2‖ ^ 2 ∂ν) ≤ M) :
    let a := Real.exp (-h / 2)
    let sigma := Real.sqrt (1 - Real.exp (-h))
    let μRefresh := ν.prod (stdGaussian E)
    let γι := Measure.pi (fun _ : ι => stdGaussian E)
    let μ := μRefresh.prod γι
    let X : (((E × E) × E) × (ι → E)) → E := fun w => w.1.1.1
    let P0 : (((E × E) × E) × (ι → E)) → E :=
      fun w => a • w.1.1.2 + sigma • w.1.2
    let G : ι → (((E × E) × E) × (ι → E)) → E := fun j w => w.2 j
    let BRefresh := M + (1 - Real.exp (-h)) * (Module.finrank ℝ E : ℝ)
    0 ≤ BRefresh ∧
      Measurable X ∧ Measurable P0 ∧
      Integrable (fun w => ‖X w - xstar‖ ^ 2 + ‖P0 w‖ ^ 2) μ ∧
      (∫ w, ‖X w - xstar‖ ^ 2 + ‖P0 w‖ ^ 2 ∂μ) ≤ BRefresh ∧
      (∀ j, Measurable (G j)) ∧
      (∀ j, Integrable (fun w => ‖G j w‖ ^ 2) μ) ∧
      (∀ j, (∫ w, ‖G j w‖ ^ 2 ∂μ) = (Module.finrank ℝ E : ℝ))

Exact module and namespace context

Proof roadmap

  1. 1Build the probability space

    The nested product measure represents freshness of the first refresh Gaussian and the first Picard innovation array.

  2. 2Reuse the refresh calculation

    The previous theorem already proves measurability, integrability, and the corrected OU moment budget; product integration lifts it unchanged over the extra innovation array.

  3. 3Read each Gaussian coordinate

    Evaluation at j is measurable. Finite-product marginal integration returns the original standard Gaussian law, so every coordinate has exactly dimension d as its squared-norm moment.

  4. 4Feed the real variables to the next theorem

    The focused test passes the product-law coordinate projections to the two-layer Picard moment theorem, and separately checks a nondegenerate one-dimensional Gaussian coordinate moment.

Full assumptions and notation

All objects and hypotheses

  • The incoming phase-state law is a probability measure with an integrable combined square bounded by M≥0.
  • The phase length is nonnegative and the Picard index type is finite.
  • E is a complete finite-dimensional real inner-product Borel space.

Mathematical proof

1. Build the probability space

The nested product measure represents freshness of the first refresh Gaussian and the first Picard innovation array.

\[\mu=(\nu\otimes\gamma_d)\otimes\bigotimes_{j\in[J]}\gamma_d.\]
Corresponding Lean step

Measure.prod constructs the refresh factor and Measure.pi constructs the finite innovation array.

2. Reuse the refresh calculation

The previous theorem already proves measurability, integrability, and the corrected OU moment budget; product integration lifts it unchanged over the extra innovation array.

\[\mathbb E_\mu[\|X-x_*\|^2+\|P_0\|^2]\le B_{\rm refresh}.\]
Corresponding Lean step

Integrable.comp_fst and integral_prod transport the established result.

3. Read each Gaussian coordinate

Evaluation at j is measurable. Finite-product marginal integration returns the original standard Gaussian law, so every coordinate has exactly dimension d as its squared-norm moment.

\[G_j(w)=w_{\rm array}(j),\quad \mathbb E\|G_j\|^2=d.\]
Corresponding Lean step

integrable_comp_eval and integral_comp_eval are the pinned Mathlib APIs.

4. Feed the real variables to the next theorem

The focused test passes the product-law coordinate projections to the two-layer Picard moment theorem, and separately checks a nondegenerate one-dimensional Gaussian coordinate moment.

\[(X,P_0,(G_j)_j)\longrightarrow\text{PicardCenterMoment}.\]
Corresponding Lean step

Tests.SmoothedPicardHMCPartialMomentumRefresh checks symbolic API composition in dimension zero and the exact positive-dimensional Gaussian moment in E=R.

Lean verification

The exact proof begins folded immediately after the mathematical argument; the statement is folded beside the claim above.

Lean proof · partial_refresh_with_picard_innovations

ASTIS owns the OU refresh and Gaussian norm-square results. Mathlib owns finite product measures, coordinate marginals, and product integration.

Braces mark parameters Lean can infer; square brackets request structures such as a measurable space or probability measure. Named hypotheses are mathematical premises, not facts established by this declaration. Section parameters are described in the mathematical hypotheses above; the module link retains their exact source context.

Download module (.lean)
theorem partial_refresh_with_picard_innovations
    (ν : Measure (E × E)) [IsProbabilityMeasure ν]
    (xstar : E) {h M : ℝ} (hh : 0 ≤ h) (hM : 0 ≤ M)
    (hstateI : Integrable (fun s : E × E => ‖s.1 - xstar‖ ^ 2 + ‖s.2‖ ^ 2) ν)
    (hstate : (∫ s : E × E, ‖s.1 - xstar‖ ^ 2 + ‖s.2‖ ^ 2 ∂ν) ≤ M) :
    let a := Real.exp (-h / 2)
    let sigma := Real.sqrt (1 - Real.exp (-h))
    let μRefresh := ν.prod (stdGaussian E)
    let γι := Measure.pi (fun _ : ι => stdGaussian E)
    let μ := μRefresh.prod γι
    let X : (((E × E) × E) × (ι → E)) → E := fun w => w.1.1.1
    let P0 : (((E × E) × E) × (ι → E)) → E :=
      fun w => a • w.1.1.2 + sigma • w.1.2
    let G : ι → (((E × E) × E) × (ι → E)) → E := fun j w => w.2 j
    let BRefresh := M + (1 - Real.exp (-h)) * (Module.finrank ℝ E : ℝ)
    0 ≤ BRefresh ∧
      Measurable X ∧ Measurable P0 ∧
      Integrable (fun w => ‖X w - xstar‖ ^ 2 + ‖P0 w‖ ^ 2) μ ∧
      (∫ w, ‖X w - xstar‖ ^ 2 + ‖P0 w‖ ^ 2 ∂μ) ≤ BRefresh ∧
      (∀ j, Measurable (G j)) ∧
      (∀ j, Integrable (fun w => ‖G j w‖ ^ 2) μ) ∧
      (∀ j, (∫ w, ‖G j w‖ ^ 2 ∂μ) = (Module.finrank ℝ E : ℝ)) := by
  classical
  dsimp only
  obtain ⟨hX, hP0, hrefreshI, hrefreshEq, hrefresh⟩ :=
    PartialMomentumRefresh.partial_momentum_refresh_second_moment
      ν xstar hh hstateI hstate
  let μRefresh := ν.prod (stdGaussian E)
  let γι := Measure.pi (fun _ : ι => stdGaussian E)
  have hvar : 0 ≤ 1 - Real.exp (-h) :=
    sub_nonneg.mpr (Real.exp_le_one_iff.mpr (by linarith))
  have hB : 0 ≤ M + (1 - Real.exp (-h)) * (Module.finrank ℝ E : ℝ) :=
    add_nonneg hM (mul_nonneg hvar (Nat.cast_nonneg _))
  have hrefreshOuter : Integrable
      (fun w : ((E × E) × E) × (ι → E) =>
        ‖w.1.1.1 - xstar‖ ^ 2 +
          ‖Real.exp (-h / 2) • w.1.1.2 +
            Real.sqrt (1 - Real.exp (-h)) • w.1.2‖ ^ 2)
      (μRefresh.prod γι) := hrefreshI.comp_fst γι
  have hrefreshOuterInt :
      (∫ w : ((E × E) × E) × (ι → E),
          ‖w.1.1.1 - xstar‖ ^ 2 +
            ‖Real.exp (-h / 2) • w.1.1.2 +
              Real.sqrt (1 - Real.exp (-h)) • w.1.2‖ ^ 2
          ∂μRefresh.prod γι) ≤
        M + (1 - Real.exp (-h)) * (Module.finrank ℝ E : ℝ) := by
    rw [integral_prod _ hrefreshOuter]
    simpa [γι, μRefresh] using hrefresh
  obtain ⟨hgaussI, hgauss⟩ :=
    integrable_norm_sq_and_integral_stdGaussian (E := E)
  have hcoordI (j : ι) : Integrable (fun z : ι → E => ‖z j‖ ^ 2) γι :=
    by
      simpa [γι] using
        (integrable_comp_eval (X := fun _ : ι => E)
          (μ := fun _ : ι => stdGaussian E) (i := j) hgaussI)
  have hcoordOuterI (j : ι) : Integrable
      (fun w : ((E × E) × E) × (ι → E) => ‖w.2 j‖ ^ 2)
      (μRefresh.prod γι) := (hcoordI j).comp_snd μRefresh
  have hcoordOuter (j : ι) :
      (∫ w : ((E × E) × E) × (ι → E), ‖w.2 j‖ ^ 2
          ∂μRefresh.prod γι) = (Module.finrank ℝ E : ℝ) := by
    rw [integral_prod _ (hcoordOuterI j)]
    simp [μRefresh, γι,
      integral_comp_eval (X := fun _ : ι => E) (μ := fun _ : ι => stdGaussian E)
        (i := j) hgaussI.aestronglyMeasurable,
      hgauss]
  exact ⟨hB, by fun_prop, by fun_prop, hrefreshOuter, hrefreshOuterInt,
    fun _ => by fun_prop, hcoordOuterI, hcoordOuter⟩

end AutoSamplingTheory.ExampleCases.SmoothedPicardHMC.PicardInputLaw

Exact module and namespace context

Source assumptions versus formal assumptions

SourceActual LeanDifference kindWhy it matters
Algorithm 3.1 says all displayed Gaussian variables are independent.The first refresh and first innovation array are represented by nested product measures.sameProduct measures make the stated freshness semantic rather than an informal label.
Lemma D.4 uses Gaussian second moments without spelling out measurability or integrability.Every coordinate is proved measurable and square-integrable with exact moment finrank(E).source-implicitThese conditions are required before the expectations in D.7 are legal.
Algorithm 3.1 samples a second array and zeta_1 later in the phase.This deliberately scoped declaration constructs only the variables consumed by the already-formalized first two query-center layers.sameIt formalizes an exact initial segment of the algorithm; later phase outputs are explicitly separate rather than treated as implicit premises.

No repeated phase history, D.3/M_kappa adapter, second array, second refresh, quadrature construction, complete D.7, D.8 or theorem result.

Encoder–denoiser: source-reviewed · equivalent-after-elaboration

Detected semantic differences

  • assumptions: Nested product measures expose independence, and measurability/integrability legalize the moment calculations. — Independent source review classification: explicit-elaboration; nonblocking.
  • scopes: The declaration is an exact conditional initial-segment adapter, not all of Algorithm 3.1 or Lemma D.4. — Independent source review classification: scope-boundary; nonblocking.
  • conclusion: Tests separate a positive-dimensional Gaussian coordinate moment from zero-dimensional end-to-end Picard API composition. — Independent source review classification: validation-boundary; nonblocking.
  • objects: The source field is explicitly an ASTIS faithful paraphrase rather than a quotation. — Independent source review classification: provenance; nonblocking.

A generalization is not a source correction. Proposed missing conditions require separate independent repair review. No proposed repair silently changes the original theorem.

Scope and omitted-condition boundaries
  • This constructs the first refresh noise and first finite Picard innovation array. The incoming run-wide moment, second array, second refresh, quadrature data, numerical D.7 and D.8 remain separate.
Sources, dependencies, and reuse

Source and reuse

Samplinglib parents called

Mathlib API called (external library)

  • MeasureTheory.Measure.pi
  • MeasureTheory.integrable_comp_eval
  • MeasureTheory.integral_comp_eval
  • MeasureTheory.integral_prod

Mathematical sources

Samplinglib prose is not a quotation or a source-equivalence certificate. Definitions and aliases are explained as constructions, not counted as new mathematical proofs.

Which proof obligations and prerequisites are covered?

  • Formalized proof edge; source adapter/review separate Actual finite independent Gaussian array
  • Formalized proof edge; source adapter/review separate Coordinate measurability, integrability and exact moments
  • TODO — not closed by these contributions Transport-derived run-wide incoming state bound (remaining)
  • TODO — not closed by these contributions Actual Chebyshev–Lobatto nodes and weights (remaining)
  • TODO — not closed by these contributions Complete numerical D.7 (remaining)