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

Actual ideal RGO identification and normalization

ASTIS mathematical restatement and exposition of the ideal target construction used by Fan Chen, Sinho Chewi, Jianfeng Lu and Matthew S. Zhang, Smoothed Picard Hamiltonian Monte Carlo.

arXiv:2602.01338v1 · Appendix D.1 Eq18 ideal target identification; consumed by SPHMC Appendix A.4(2) · faithful paraphrase

Complete source statement (Samplinglib restatement)

exp(-f) is G-integrable with strictly positive real integral; exp(m) is q-integrable with strictly positive real integral; exp(-V) is canonical-volume-integrable with strictly positive real integral. The actual q.tilted(m) equals G.tilted(-f) and also equals volume.withDensity(x -> ofReal(exp(-V(x))/integral exp(-V) dvolume)).

  • E is a finite-dimensional real inner-product space with its Borel sigma-algebra and canonical volume. f:E->R is differentiable everywhere and its actual gradient is globally beta-Lipschitz, with beta>=0. Fix eta>0 satisfying beta*eta<1 and arbitrary x0,x_plus in E. No convexity or second differentiability is assumed.
  • Set g=grad f(x_plus), h=x0-eta*g, P=(z->sqrt(eta) z)_*stdGaussian(E), U=volume restricted to (0,1], nu=U product P, G=(z->x0+sqrt(eta) z)_*stdGaussian(E), and q=(z->h+sqrt(eta) z)_*stdGaussian(E). Thus time and auxiliary Gaussian inputs are independent.
  • For s=(r,z), theta=pi*r/2, gamma=h+sin(theta)(x-h)+cos(theta)z and v=(pi/2)(cos(theta)(x-h)-sin(theta)z). The actual untruncated estimator is W(x,s)=<v,g-grad f(gamma)>. Define m(x)=integral W(x,s) dnu(s) and V(x)=f(x)+norm(x-x0)^2/(2eta). These are definitions of actual functions and laws, not assumptions about their means or integrability.

Actual ideal RGO law

\[q.\mathrm{tilted}(m)=G.\mathrm{tilted}(-f)=\frac{e^{-V(x)}}{\int_Ee^{-V(y)}\,dy}\,dx,\qquad V(x)=f(x)+\frac{\|x-x_0\|^2}{2\eta}.\]

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

Actual ideal RGO identification and normalization

AutoSamplingTheory.ExampleCases.SmoothedPicardHMC.IdealRGOIdentification.ideal_rgo_identification · theorem · Teaching coverage

Read this first

Mathematical statement

exp(-f) is G-integrable with strictly positive real integral; exp(m) is q-integrable with strictly positive real integral; exp(-V) is canonical-volume-integrable with strictly positive real integral. The actual q.tilted(m) equals G.tilted(-f) and also equals volume.withDensity(x → ofReal(exp(-V(x))/integral exp(-V) dvolume)).

\[q.\mathrm{tilted}(m)=G.\mathrm{tilted}(-f)=\frac{e^{-V(x)}}{\int_Ee^{-V(y)}\,dy}\,dx,\qquad V(x)=f(x)+\frac{\|x-x_0\|^2}{2\eta}.\]
Lean statement · ideal_rgo_identification

Three actual exponential L1 and positive real integral pairs, plus shifted-Gaussian/ideal Gaussian and normalized-volume measure identities.

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 ideal_rgo_identification (f : E → ℝ) (hf : Differentiable ℝ f)
    (beta : NNReal) (hlip : LipschitzWith beta (gradient f))
    (eta : ℝ) (heta : 0 < eta) (hstep : (beta : ℝ)*eta < 1) (x0 xp : E) :
    let h := x0-eta • gradient f xp
    let P := (stdGaussian E).map (fun z => Real.sqrt eta • z)
    let nu := (volume.restrict (Set.Ioc (0 : ℝ) 1)).prod P
    let G := (stdGaussian E).map (fun z => x0 + Real.sqrt eta • z)
    let q := (stdGaussian E).map (fun z => h + Real.sqrt eta • z)
    let m := fun x => ∫ s, estimator f h xp x s ∂nu
    let V := fun x => f x + ‖x-x0‖^2/(2*eta)
    Integrable (fun x => Real.exp (-f x)) G ∧
      0 < ∫ x, Real.exp (-f x) ∂G ∧
      Integrable (fun x => Real.exp (m x)) q ∧
      0 < ∫ x, Real.exp (m x) ∂q ∧
      Integrable (fun x => Real.exp (-V x)) volume ∧
      0 < ∫ x, Real.exp (-V x) ∂volume ∧
      q.tilted m = G.tilted (fun x => -f x) ∧
      q.tilted m = (volume : Measure E).withDensity (fun x => ENNReal.ofReal
        (Real.exp (-V x) / ∫ y, Real.exp (-V y) ∂volume))

Exact module and namespace context

Proof roadmap

  1. 1Obtain the sharp lower Taylor estimate from first derivatives

    For fixed u, consider H(t)=f(x0+t u)-t<grad f(x0),u>+(beta/2)t^2 norm(u)^2 on [0,1]. Differentiate the actual potential along this path. The Lipschitz gradient bound and Cauchy-Schwarz make H prime nonnegative. Monotonicity therefore gives H(1)≥H(0). This retains the precise beta/2 coefficient without a Hessian or convexity assumption.

  2. 2Prove a coercive envelope and actual volume integrability

    Set k=1/eta-beta>0. Adding the proximal quadratic to the lower Taylor bound gives a quadratic with coefficient k/2. Young inequality absorbs the linear gradient term, leaving coefficient k/4. Exponentiation bounds the actual weight by an integrable translated Gaussian envelope. Continuity supplies measurability. The actual canonical volume is nonzero, so the integral of this everywhere positive integrable exponential is strictly positive.

  3. 3Normalize under the actual Gaussian input

    Reuse the public scaled Gaussian density and prove the translation adapter by pushforward composition and volume translation invariance. Multiplying this density by exp(-f) gives c_eta exp(-V), where c_eta is the strictly positive Gaussian normalization constant. The withDensity integrability equivalence transfers the established volume L1 to the actual Gaussian G. Its probability property and integrability imply a strictly positive real integral. Applying the same density identity to the tilt gives the explicit normalized volume law.

  4. 4Use the actual gradient-arc mean with one common constant

    Instantiate the proved GradientArcMean result at h=x0-eta grad f(x_plus). It already establishes the auxiliary integrability needed for its Fubini argument. Expanding the local estimator verifies that it is exactly the same sine/cosine arc and gradient difference. Its potential expectation defines C before x is chosen, and gives the actual mean identity for every x. No affine mean formula is assumed in place of the random estimator.

  5. 5Cancel the actual translated Gaussian likelihood

    The public Gaussian likelihood theorem gives q=G.withDensity(L) in the correct direction. Substituting h-x0=-eta g yields log L=-<g,x-x0>-eta norm(g)^2/2. Multiplication by the actual exponential mean leaves A exp(-f), where A is positive and independent of x. This uses the actual proposal measure, not an abstract density selected to satisfy the conclusion.

  6. 6Transfer integrability before cancelling normalizers

    The weighted integrability equivalence and the previous product identity prove exp(m) is q-integrable. Integration gives Z_q=A Z_G, hence Z_q>0. Only then expand the definition of exponential tilt and compose the two withDensity operations. The same positive A cancels in numerator and denominator. Combine this law equality with the established normalized volume identity and all three positive integrability statements.

Full assumptions and notation

All objects and hypotheses

  • E is a finite-dimensional real inner-product space with its Borel sigma-algebra and canonical volume. f:E→R is differentiable everywhere and its actual gradient is globally beta-Lipschitz, with beta≥0. Fix eta>0 satisfying beta*eta<1 and arbitrary x0,x_plus in E. No convexity or second differentiability is assumed.
  • Set g=grad f(x_plus), h=x0-eta*g, P=(z→sqrt(eta) z)_*stdGaussian(E), U=volume restricted to (0,1], nu=U product P, G=(z→x0+sqrt(eta) z)_*stdGaussian(E), and q=(z→h+sqrt(eta) z)_*stdGaussian(E). Thus time and auxiliary Gaussian inputs are independent.
  • For s=(r,z), theta=pi*r/2, gamma=h+sin(theta)(x-h)+cos(theta)z and v=(pi/2)(cos(theta)(x-h)-sin(theta)z). The actual untruncated estimator is W(x,s)=<v,g-grad f(gamma)>. Define m(x)=integral W(x,s) dnu(s) and V(x)=f(x)+norm(x-x0)^2/(2eta). These are definitions of actual functions and laws, not assumptions about their means or integrability.

Mathematical proof

1. Obtain the sharp lower Taylor estimate from first derivatives

For fixed u, consider H(t)=f(x0+t u)-t<grad f(x0),u>+(beta/2)t^2 norm(u)^2 on [0,1]. Differentiate the actual potential along this path. The Lipschitz gradient bound and Cauchy-Schwarz make H prime nonnegative. Monotonicity therefore gives H(1)≥H(0). This retains the precise beta/2 coefficient without a Hessian or convexity assumption.

\[f(x_0+u)\ge f(x_0)+\langle\nabla f(x_0),u\rangle-\frac\beta2\|u\|^2.\]
Corresponding Lean step

smooth_lower; monotoneOn_of_deriv_nonneg

2. Prove a coercive envelope and actual volume integrability

Set k=1/eta-beta>0. Adding the proximal quadratic to the lower Taylor bound gives a quadratic with coefficient k/2. Young inequality absorbs the linear gradient term, leaving coefficient k/4. Exponentiation bounds the actual weight by an integrable translated Gaussian envelope. Continuity supplies measurability. The actual canonical volume is nonzero, so the integral of this everywhere positive integrable exponential is strictly positive.

\[V(x_0+u)\ge f(x_0)-\frac{\|\nabla f(x_0)\|^2}{k}+\frac{k}{4}\|u\|^2,\qquad k=\eta^{-1}-\beta>0.\]
Corresponding Lean step

regularized_lower; ideal_volume_integrable; AutoSamplingTheory.TechnicalLemmas.Analysis.Integrability.integrable_exp_neg_add_mul_norm_sub_sq; MeasureTheory.integral_exp_pos

3. Normalize under the actual Gaussian input

Reuse the public scaled Gaussian density and prove the translation adapter by pushforward composition and volume translation invariance. Multiplying this density by exp(-f) gives c_eta exp(-V), where c_eta is the strictly positive Gaussian normalization constant. The withDensity integrability equivalence transfers the established volume L1 to the actual Gaussian G. Its probability property and integrability imply a strictly positive real integral. Applying the same density identity to the tilt gives the explicit normalized volume law.

\[G(dx)=c_\eta e^{-\|x-x_0\|^2/(2\eta)}dx,\qquad c_\eta=(\sqrt{2\pi\eta})^{-\dim E}>0.\]
Corresponding Lean step

translated_density; ideal_gaussian_integrable; ideal_volume_law; AutoSamplingTheory.TechnicalLemmas.Measure.IsotropicGaussianDensity.map_sqrt_smul_stdGaussian_eq_withDensity

4. Use the actual gradient-arc mean with one common constant

Instantiate the proved GradientArcMean result at h=x0-eta grad f(x_plus). It already establishes the auxiliary integrability needed for its Fubini argument. Expanding the local estimator verifies that it is exactly the same sine/cosine arc and gradient difference. Its potential expectation defines C before x is chosen, and gives the actual mean identity for every x. No affine mean formula is assumed in place of the random estimator.

\[m(x)=\langle g,x\rangle-f(x)+C,\qquad C=\mathbb E_{Z\sim P}f(h+Z)-\langle g,h\rangle.\]
Corresponding Lean step

estimator; actual_mean_tilt; AutoSamplingTheory.ExampleCases.SmoothedPicardHMC.GradientArcMean.gradient_arc_mean

5. Cancel the actual translated Gaussian likelihood

The public Gaussian likelihood theorem gives q=G.withDensity(L) in the correct direction. Substituting h-x0=-eta g yields log L=-<g,x-x0>-eta norm(g)^2/2. Multiplication by the actual exponential mean leaves A exp(-f), where A is positive and independent of x. This uses the actual proposal measure, not an abstract density selected to satisfy the conclusion.

\[L(x)e^{m(x)}=A e^{-f(x)},\qquad A=e^{C+\langle g,x_0\rangle-\eta\|g\|^2/2}>0.\]
Corresponding Lean step

affine_gaussian_tilt; AutoSamplingTheory.TechnicalLemmas.Measure.GaussianLikelihood.translated_gaussian_likelihood

6. Transfer integrability before cancelling normalizers

The weighted integrability equivalence and the previous product identity prove exp(m) is q-integrable. Integration gives Z_q=A Z_G, hence Z_q>0. Only then expand the definition of exponential tilt and compose the two withDensity operations. The same positive A cancels in numerator and denominator. Combine this law equality with the established normalized volume identity and all three positive integrability statements.

\[Z_q=\int e^m\,dq=A\int e^{-f}\,dG=A Z_G>0,\qquad \frac{L e^m}{Z_q}=\frac{e^{-f}}{Z_G}.\]
Corresponding Lean step

tilt_change_density; ideal_rgo_identification; MeasureTheory.integrable_withDensity_iff_integrable_smul'; MeasureTheory.integral_withDensity_eq_integral_toReal_smul; MeasureTheory.withDensity_mul

Lean verification

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

Lean proof · ideal_rgo_identification

Sharp first-derivative Taylor bound, Gaussian envelope, actual density transfer, actual parent mean, positive likelihood factor and normalizer cancellation.

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 ideal_rgo_identification (f : E → ℝ) (hf : Differentiable ℝ f)
    (beta : NNReal) (hlip : LipschitzWith beta (gradient f))
    (eta : ℝ) (heta : 0 < eta) (hstep : (beta : ℝ)*eta < 1) (x0 xp : E) :
    let h := x0-eta • gradient f xp
    let P := (stdGaussian E).map (fun z => Real.sqrt eta • z)
    let nu := (volume.restrict (Set.Ioc (0 : ℝ) 1)).prod P
    let G := (stdGaussian E).map (fun z => x0 + Real.sqrt eta • z)
    let q := (stdGaussian E).map (fun z => h + Real.sqrt eta • z)
    let m := fun x => ∫ s, estimator f h xp x s ∂nu
    let V := fun x => f x + ‖x-x0‖^2/(2*eta)
    Integrable (fun x => Real.exp (-f x)) G ∧
      0 < ∫ x, Real.exp (-f x) ∂G ∧
      Integrable (fun x => Real.exp (m x)) q ∧
      0 < ∫ x, Real.exp (m x) ∂q ∧
      Integrable (fun x => Real.exp (-V x)) volume ∧
      0 < ∫ x, Real.exp (-V x) ∂volume ∧
      q.tilted m = G.tilted (fun x => -f x) ∧
      q.tilted m = (volume : Measure E).withDensity (fun x => ENNReal.ofReal
        (Real.exp (-V x) / ∫ y, Real.exp (-V y) ∂volume)) := by
  have hg := ideal_gaussian_integrable f hf beta hlip eta heta hstep x0
  have hm := actual_mean_tilt f hf beta hlip eta heta hstep x0 xp
  have hv := ideal_volume_integrable f hf beta hlip eta heta hstep x0
  exact ⟨hg.1, hg.2, hm.1, hm.2.1, hv.1, hv.2, hm.2.2,
    hm.2.2.trans (ideal_volume_law f hf beta hlip eta heta hstep x0)⟩

end AutoSamplingTheory.ExampleCases.SmoothedPicardHMC.IdealRGOIdentification

Exact module and namespace context

Source assumptions versus formal assumptions

SourceActual LeanDifference kindWhy it matters
Euclidean smooth setting in positive dimension.Finite-dimensional real inner-product Borel space with canonical volume, including dimension zero.generalizationCoordinate-free Gaussian and inner-product results apply, without division by dimension.
The smoothness constant is conventionally positive.beta is a nonnegative real parameter (NNReal); beta=0 is allowed.generalizationNo division by beta is used.
Claim2 uses a stronger small-step condition together with ell>=2 for clipping accuracy.eta>0 and beta*eta<1 suffice for this ideal target identification and normalization.generalizationThe source smooth Claim2 range implies beta*eta<=1/(8ell)<=1/16. The ideal-only contract does not assert the clipping accuracy conclusion. Strict inequality matters: a negative quadratic can make V constant at beta*eta=1 in positive dimension.
A reference-point residual bound is used for downstream clipping control.x_plus is arbitrary for ideal mean identification.generalizationThe likelihood cancels the affine gradient term for every reference point; the residual condition is retained for later clipping estimates.
Uniform time is sampled on [0,1].U is volume restricted to Ioc(0,1), independently multiplied by the actual auxiliary Gaussian.source-implicitThe endpoints are Lebesgue-null. The true path and auxiliary product are retained.
The ideal RGO is the normalized density proportional to exp(-V).All three required exponential weights are proved integrable with strictly positive real normalizers before identifying the measures.source-implicitNo assumed integrability or mean identity replaces the actual Gaussian/gradient definitions; no totalized zero denominator supplies a fake equality.
SPHMC A.4(2) uses the ideal construction in a terminal approximate implementation.Fixed-parameter ideal target identification only; no clipped Renyi, joint adaptive kernel, initialization or cost theorem.sameThis is the selected dependency, not the complete implementation or full paper theorem.

This identifies the actual unclipped-mean ideal RGO and proves its positive finite normalization. beta*eta<1 is sufficient and contains the source Claim2 small-step range, not equivalent to all its conditions. Arbitrary reference x_plus, beta0, zero dimension and coordinate-free spaces are disclosed extensions. The public theorem has eight conclusions: three L1 statements, three positive real integrals and two measure equalities. Probability of the resulting law follows, but no separate IsProbabilityMeasure conjunct or joint adaptive-parameter kernel is returned. Clipped-vs-ideal Renyi accuracy, initialization, implementation and query costs remain separate. Neither full companion paper is complete.

Encoder–denoiser: accepted · domain-mismatch

Detected semantic differences

  • domains: Coordinate-free canonical volume and zero dimension supported. — Independent formal reviewer accepted: Coordinate-free canonical volume and zero dimension supported.
  • assumptions: NNReal beta includes zero; no division by beta. — Independent formal reviewer accepted: NNReal beta includes zero; no division by beta.
  • assumptions: betaeta<1 sufficient broader range; strictness matters for negative quadratic at equality in positive dimension. — Independent formal reviewer accepted: betaeta<1 sufficient broader range; strictness matters for negative quadratic at equality in positive dimension.
  • quantifiers: Arbitrary xp preserves source center; ideal likelihood identity needs no residual bound. — Independent formal reviewer accepted: Arbitrary xp preserves source center; ideal likelihood identity needs no residual bound.
  • objects: Ioc uniform differs by null endpoints; true independent product retained. — Independent formal reviewer accepted: Ioc uniform differs by null endpoints; true independent product retained.
  • conclusion: Exponential L1/positive integrals proved before measure identity, no zero-denominator fallback. — Independent formal reviewer accepted: Exponential L1/positive integrals proved before measure identity, no zero-denominator fallback.
  • scopes: Selected ideal dependency only; terminal approximate implementation remains separate. — Independent formal reviewer accepted: Selected ideal dependency only; terminal approximate implementation remains separate.

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 identifies the actual unclipped-mean ideal RGO and proves its positive finite normalization. beta*eta<1 is sufficient and contains the source Claim2 small-step range, not equivalent to all its conditions. Arbitrary reference x_plus, beta0, zero dimension and coordinate-free spaces are disclosed extensions. The public theorem has eight conclusions: three L1 statements, three positive real integrals and two measure equalities. Probability of the resulting law follows, but no separate IsProbabilityMeasure conjunct or joint adaptive-parameter kernel is returned. Clipped-vs-ideal Renyi accuracy, initialization, implementation and query costs remain separate. Neither full companion paper is complete.
Sources, dependencies, and reuse

Source and reuse

Samplinglib parents called

Mathlib API called (external library)

  • monotoneOn_of_deriv_nonneg
  • MeasureTheory.integral_exp_pos
  • MeasureTheory.integrable_withDensity_iff_integrable_smul'
  • MeasureTheory.integral_withDensity_eq_integral_toReal_smul
  • MeasureTheory.withDensity_mul

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 unclipped-mean target identification with positive finite normalization