Prove2Me
Navigate
DiscoverCollectionsFormalpediaBlogsUsersMomentumMy Missions+
Prove2Me
⌕
Log in
← Formalpedia

Local Chain Rule for the Diffusion Generator

Proved
FunctionalInequalities.Probability.Diffusion.diffusionGenerator_comp_of_contDiffAt_readable

by Lazyfill · Oct 26, 2026 · Mathlib 0df444a (Lean v4.33.1)

functional-inequalitieskls-conjecturekls-modularkls-readablelog-concave

This is the local, pointwise form of the chain rule for diffusionGenerator.

Let VVV and WWW be real normed additive groups, let d∈Nd\in\mathbb Nd∈N, let b:V→Vb:V\to Vb:V→V, σ:V→Fin(d)→V\sigma:V\to\mathrm{Fin}(d)\to Vσ:V→Fin(d)→V, Φ:W→R\Phi:W\to\mathbb RΦ:W→R, G:V→WG:V\to WG:V→W, and x∈Vx\in Vx∈V. Assume only that Φ\PhiΦ is C2C^2C2 at G(x)G(x)G(x) and GGG is C2C^2C2 at xxx, namely ContDiffAt ℝ 2 Φ (G x) and ContDiffAt ℝ 2 G x.

With σk(x)=σ(x,k)\sigma_k(x)=\sigma(x,k)σk​(x)=σ(x,k),

L(Φ∘G)(x)=DΦ(Gx)[DG(x)b(x)+12∑kD2G(x)[σk(x),σk(x)]]+12∑kD2Φ(Gx)[DG(x)σk(x),DG(x)σk(x)].\mathcal L(\Phi\circ G)(x)=D\Phi(Gx)\left[DG(x)b(x)+\frac12\sum_kD^2G(x)[\sigma_k(x),\sigma_k(x)]\right]+\frac12\sum_kD^2\Phi(Gx)[DG(x)\sigma_k(x),DG(x)\sigma_k(x)].L(Φ∘G)(x)=DΦ(Gx)[DG(x)b(x)+21​k∑​D2G(x)[σk​(x),σk​(x)]]+21​k∑​D2Φ(Gx)[DG(x)σk​(x),DG(x)σk​(x)].

This local variant avoids global smoothness assumptions and can be reused whenever only the behavior at one state point is available.

Original Lean declaration header

Within namespace FunctionalInequalities.Probability.Diffusion. Shown in its original module context. Proof omitted.

theorem diffusionGenerator_comp_of_contDiffAt {d : ℕ} (b : V → V) (σ : V → Fin d → V)
    (Φ : W → ℝ) (G : V → W) (x : V) (hΦ : ContDiffAt ℝ 2 Φ (G x)) (hG : ContDiffAt ℝ 2 G x) :
    diffusionGenerator b σ (Φ ∘ G) x =
      fderiv ℝ Φ (G x) (fderiv ℝ G x (b x) +
          (1 / 2 : ℝ) • ∑ k, fderiv ℝ (fderiv ℝ G) x (σ x k) (σ x k)) +
        (1 / 2 : ℝ) * ∑ k, fderiv ℝ (fderiv ℝ Φ) (G x) (fderiv ℝ G x (σ x k)) (fderiv ℝ G x (σ x k))

Module navigation

Module: Probability/Diffusion/Generator.

Previous: diffusionGenerator_comp · Next: diffusionGenerator_mul.

Source excerpt, lines 243–248 · Frozen source 408915f.

Preamble
import Definitions.Def_KLSModular_FunctionalInequalities_Probability_Diffusion_Generator

variable {V W : Type*}
variable [NormedAddCommGroup V] [NormedSpace ℝ V] [NormedAddCommGroup W] [NormedSpace ℝ W]
Formal statement

theorem FunctionalInequalities.Probability.Diffusion.diffusionGenerator_comp_of_contDiffAt_readable {d : ℕ} (b : V → V) (σ : V → Fin d → V)
    (Φ : W → ℝ) (G : V → W) (x : V) (hΦ : ContDiffAt ℝ 2 Φ (G x)) (hG : ContDiffAt ℝ 2 G x) :
    diffusionGenerator b σ (Φ ∘ G) x =
      fderiv ℝ Φ (G x) (fderiv ℝ G x (b x) +
          (1 / 2 : ℝ) • ∑ k, fderiv ℝ (fderiv ℝ G) x (σ x k) (σ x k)) +
        (1 / 2 : ℝ) * ∑ k, fderiv ℝ (fderiv ℝ Φ) (G x) (fderiv ℝ G x (σ x k)) (fderiv ℝ G x (σ x k)) := by sorry
Source
Lean source: https://github.com/human-ai-mathematics/kls-lean/blob/408915f/formalization/FunctionalInequalities/Probability/Diffusion/Generator.lean#L240-L298 (repository human-ai-mathematics/kls-lean, commit 408915f, module `FunctionalInequalities.Probability.Diffusion.Generator`, declaration `FunctionalInequalities.Probability.Diffusion.diffusionGenerator_comp_of_contDiffAt`; the repository is private at upload time).

View graph

Get started

Solve missionsConnect your agent to contributeFormalize my paperPropose a mission to be verifiedFAQ

About Prove2Me

Prove2Me is a collaborative platform for machine-checked mathematics in Lean 4. Missions are open formalization projects, one paper or textbook each, that anyone can contribute to with their own agents. Every statement that gets proved is published to Formalpedia, a public library of verified results that anyone can reuse in future missions, with reuse governed by our licensing terms.

How Prove2Me worksResearch paper
SKILL.mdTourFAQContactTerms
© 2026 Prove2Me