Prove2Me
Navigate
DiscoverFormalpediaBlogsUsersMomentumMy Missions+
Prove2Me
⌕
Log in
← Formalpedia

Proposition 1 — Gaussian Process Limit at Initialization

Proved
JGH.GaussianInitialization

by Minghui · Sep 26, 2026 · Mathlib c5ea003 (Lean v4.30.0)

machine-learningneural-tangent-kernelprobability

Mathematical statement

For all positive input and output dimensions, arbitrary hidden-layer count hhh, positive bias scale, and Lipschitz activation, the initialized output vector on every fixed finite input family converges in distribution:

(fθ,k(xi))i<N,k<q ⟹ N ⁣(0,[Σ(h+1)(xi,xj)δkk′](i,k),(j,k′)).\big(f_{\theta,k}(x_i)\big)_{i<N,k<q} \ \Longrightarrow\ \mathcal N\!\left(0, [\Sigma^{(h+1)}(x_i,x_j)\delta_{kk'}]_{(i,k),(j,k')}\right).(fθ,k​(xi​))i<N,k<q​ ⟹ N(0,[Σ(h+1)(xi​,xj​)δkk′​](i,k),(j,k′)​).

Widths tend to infinity in the sequential order defined below. The Gaussian outputs are independent across output coordinates, but generally correlated across inputs. Formalization note: direct source Proposition 1 in its full finite-dimensional-distribution formulation, rather than a claim about a topology on all functions on the input space.

Source: Arthur Jacot, Franck Gabriel, Clément Hongler, Neural Tangent Kernel: Convergence and Generalization in Neural Networks, NeurIPS 2018, arXiv:1806.07572v4, https://arxiv.org/abs/1806.07572v4; Section 4.1, PDF p. 5, Proposition 1; Appendix A.1, PDF p. 11 and PDF p. 12, Proposition 1 and its proof. Displays are unnumbered.

Notation and probability model

Let d,q≥1d,q\ge1d,q≥1 be the input and output dimensions, h≥0h\ge0h≥0 the number of hidden layers, L=h+1L=h+1L=h+1, β>0\beta>0β>0, and σ:R→R\sigma:\mathbb R\to\mathbb Rσ:R→R a Lipschitz activation with a nonnegative Lipschitz constant KKK. For widths n0=dn_0=dn0​=d, nL=qn_L=qnL​=q, and nℓ=wℓ−1+1n_\ell=w_{\ell-1}+1nℓ​=wℓ−1​+1 with wi∈Nw_i\in\mathbb Nwi​∈N, the probability space Ωw\Omega_wΩw​ is the finite real parameter space with every weight and bias coordinate independently N(0,1)\mathcal N(0,1)N(0,1). Its law is Pw\mathbb P_wPw​. The network has the recursion

zj(ℓ+1)(x)=1nℓ∑iWji(ℓ)ai(ℓ)(x)+βbj(ℓ),a(0)(x)=x,a(ℓ)(x)=σ(z(ℓ)(x)) (1≤ℓ≤h),z^{(\ell+1)}_j(x)=\frac{1}{\sqrt{n_\ell}}\sum_i W^{(\ell)}_{ji} a^{(\ell)}_i(x)+\beta b^{(\ell)}_j,\qquad a^{(0)}(x)=x,\quad a^{(\ell)}(x)=\sigma(z^{(\ell)}(x))\ (1\le\ell\le h),zj(ℓ+1)​(x)=nℓ​​1​i∑​Wji(ℓ)​ai(ℓ)​(x)+βbj(ℓ)​,a(0)(x)=x,a(ℓ)(x)=σ(z(ℓ)(x)) (1≤ℓ≤h),

with output fθ=z(L)f_\theta=z^{(L)}fθ​=z(L). The full kernel, including all weights and biases, is

Θkk′(L)(θ;x,y)=∑p∂θpfθ,k(x)∂θpfθ,k′(y).\Theta^{(L)}_{kk'}(\theta;x,y)=\sum_p \partial_{\theta_p}f_{\theta,k}(x)\partial_{\theta_p}f_{\theta,k'}(y).Θkk′(L)​(θ;x,y)=p∑​∂θp​​fθ,k​(x)∂θp​​fθ,k′​(y).

For a centered Gaussian pair (U,V)(U,V)(U,V) with covariance induced by Σ(ℓ)\Sigma^{(\ell)}Σ(ℓ) on (x,y)(x,y)(x,y), put

Σ(1)(x,y)=⟨x,y⟩/d+β2,Σ(ℓ+1)(x,y)=E[σ(U)σ(V)]+β2,\Sigma^{(1)}(x,y)=\langle x,y\rangle/d+\beta^2,\qquad \Sigma^{(\ell+1)}(x,y)=\mathbb E[\sigma(U)\sigma(V)]+\beta^2,Σ(1)(x,y)=⟨x,y⟩/d+β2,Σ(ℓ+1)(x,y)=E[σ(U)σ(V)]+β2, Σ˙(ℓ+1)(x,y)=E[σ′(U)σ′(V)],Θ∞(1)=Σ(1),Θ∞(ℓ+1)=Θ∞(ℓ)Σ˙(ℓ+1)+Σ(ℓ+1).\dot\Sigma^{(\ell+1)}(x,y)=\mathbb E[\sigma'(U)\sigma'(V)],\qquad \Theta_\infty^{(1)}=\Sigma^{(1)},\quad \Theta_\infty^{(\ell+1)}=\Theta_\infty^{(\ell)}\dot\Sigma^{(\ell+1)}+\Sigma^{(\ell+1)}.Σ˙(ℓ+1)(x,y)=E[σ′(U)σ′(V)],Θ∞(1)​=Σ(1),Θ∞(ℓ+1)​=Θ∞(ℓ)​Σ˙(ℓ+1)+Σ(ℓ+1).

All kernel products in the last expression are pointwise. Local index hhh in covarianceKernel and limitingNTK denotes paper depth h+1h+1h+1. The dataset X=(xi)i<NX=(x_i)_{i<N}X=(xi​)i<N​ is any fixed finite family; repetitions and N=0N=0N=0 are allowed. δkk′\delta_{kk'}δkk′​ is the Kronecker delta.

The limit takes n1n_1n1​ to infinity first and nhn_hnh​ last. More precisely, for any required error tolerance, the width condition is ∀eventuallywh−1⋯∀eventuallyw0\forall^{\mathrm{eventually}}w_{h-1}\cdots \forall^{\mathrm{eventually}}w_0∀eventuallywh−1​⋯∀eventuallyw0​; each inner threshold may depend on the fixed outer widths. For h=0h=0h=0 the filter is concentrated on the unique empty width vector, so the statements require the exact affine base case. This is not a simultaneous-width or whole-input-space uniform limit.

Formalization note: Gaussian measures are concrete Mathlib measures, including singular covariance. The covariance-validity milestone establishes their covariance interpretation; it is not a hypothesis of either convergence target. The activation assumption is only Lipschitz. Derivatives take Mathlib's zero value at points without derivatives, and proofs must justify the null exceptional set under positive Gaussian bias. Native convergence in distribution includes almost-everywhere measurability and weak convergence of probability laws. Primary source conventions: Jacot–Gabriel–Hongler, Section 2, PDF pp. 2–3; Section 4.1, PDF p. 5, Proposition 1, Theorem 1 and Remarks 2–3; Appendix A opening paragraphs, PDF p. 11, and Appendix A.1, PDF pp. 11–13. The relevant displays have no equation numbers.

Preamble
import Definitions.Def_JGH_NTK_Model
open MeasureTheory Filter
open scoped Topology NNReal
Formal statement
namespace JGH
theorem GaussianInitialization :
  ∀ (d q : ℕ), 0 < d → 0 < q →
    ∀ (σ : ℝ → ℝ) (K : ℝ≥0), LipschitzWith K σ →
      ∀ (β : ℝ), 0 < β → ∀ (h N : ℕ) (X : Fin N → Input d),
        TendstoInDistribution (initializedOutput (h := h) d q σ β X)
          (sequentialWidths h) id (fun w ↦ initialization (h + 1) (widths d q w))
          (outputGaussian d q σ β h X) := by sorry
end JGH
Source
Arthur Jacot, Franck Gabriel, Clément Hongler, Neural Tangent Kernel: Convergence and Generalization in Neural Networks, NeurIPS 2018, arXiv:1806.07572v4, https://arxiv.org/abs/1806.07572v4; Section 4.1, PDF p. 5, Proposition 1; Appendix A.1, PDF p. 11 and PDF p. 12, Proposition 1 and its proof. Displays are unnumbered.
Read-back

What the Lean code literally says, in plain math · gpt-6

For every positive pair of integers d,qd,qd,q, every function σ:R→R\sigma:\mathbb R\to\mathbb Rσ:R→R, every nonnegative real KKK satisfying ∣σ(s)−σ(t)∣≤K∣s−t∣|\sigma(s)-\sigma(t)|\le K|s-t|∣σ(s)−σ(t)∣≤K∣s−t∣ for all real s,ts,ts,t, every real β>0\beta>0β>0, every h,N∈Nh,N\in\mathbb Nh,N∈N, and every family X0,…,XN−1∈RdX_0,\ldots,X_{N-1}\in\mathbb R^dX0​,…,XN−1​∈Rd, the joint initialized output vector defined below is almost everywhere measurable at every width tuple and converges in distribution to the stated Gaussian-constructor law. A width tuple is w=(w0,…,wh−1)∈Nhw=(w_0,\ldots,w_{h-1})\in\mathbb N^hw=(w0​,…,wh−1​)∈Nh; put n0=dn_0=dn0​=d, nℓ=wℓ−1+1n_\ell=w_{\ell-1}+1nℓ​=wℓ−1​+1 for 1≤ℓ≤h1\le\ell\le h1≤ℓ≤h, and nh+1=qn_{h+1}=qnh+1​=q. For each tuple independently specify its probability space as all real parameters WjaℓW^\ell_{ja}Wjaℓ​ and bjℓb^\ell_jbjℓ​, with 0≤ℓ≤h0\le\ell\le h0≤ℓ≤h, 0≤j<nℓ+10\le j<n_{\ell+1}0≤j<nℓ+1​, and 0≤a<nℓ0\le a<n_\ell0≤a<nℓ​, endowed with the finite product law making every weight and every bias independent with distribution N(0,1)\mathcal N(0,1)N(0,1). For an input xxx, define z0(x)=xz^0(x)=xz0(x)=x, zj1(x)=d−1/2∑a=0d−1Wja0xa+βbj0z^1_j(x)=d^{-1/2}\sum_{a=0}^{d-1}W^0_{ja}x_a+\beta b^0_jzj1​(x)=d−1/2∑a=0d−1​Wja0​xa​+βbj0​, and zjℓ+1(x)=nℓ−1/2∑a=0nℓ−1Wjaℓσ(zaℓ(x))+βbjℓz^{\ell+1}_j(x)=n_\ell^{-1/2}\sum_{a=0}^{n_\ell-1}W^\ell_{ja}\sigma(z^\ell_a(x))+\beta b^\ell_jzjℓ+1​(x)=nℓ−1/2​∑a=0nℓ​−1​Wjaℓ​σ(zaℓ​(x))+βbjℓ​ for 1≤ℓ≤h1\le\ell\le h1≤ℓ≤h. Write fk(θ,x)=zkh+1(x)f_k(\theta,x)=z^{h+1}_k(x)fk​(θ,x)=zkh+1​(x), so the output is the final preactivation; the random vector is (fk(θ,Xi))0≤i<N, 0≤k<q∈RNq(f_k(\theta,X_i))_{0\le i<N,\ 0\le k<q}\in\mathbb R^{Nq}(fk​(θ,Xi​))0≤i<N, 0≤k<q​∈RNq, using the same parameter sample for all inputs and all outputs at a fixed width tuple. Define C0(x,y)=d−1∑a=0d−1xaya+β2C_0(x,y)=d^{-1}\sum_{a=0}^{d-1}x_a y_a+\beta^2C0​(x,y)=d−1∑a=0d−1​xa​ya​+β2 and Cr+1(x,y)=∫R2σ(u)σ(v) dG(Sr(x,y))(u,v)+β2C_{r+1}(x,y)=\int_{\mathbb R^2}\sigma(u)\sigma(v)\,dG(S_r(x,y))(u,v)+\beta^2Cr+1​(x,y)=∫R2​σ(u)σ(v)dG(Sr​(x,y))(u,v)+β2, with Sr(x,y)=(Cr(x,x)Cr(x,y)Cr(y,x)Cr(y,y))S_r(x,y)=\begin{pmatrix}C_r(x,x)&C_r(x,y)\\C_r(y,x)&C_r(y,y)\end{pmatrix}Sr​(x,y)=(Cr​(x,x)Cr​(y,x)​Cr​(x,y)Cr​(y,y)​). For a finite real square matrix SSS, G(S)G(S)G(S) is the distribution of S1/2ZS^{1/2}ZS1/2Z for a standard Gaussian vector ZZZ when SSS is symmetric positive semidefinite, and is the point mass at zero otherwise; the integral is zero under the total-integral convention if its integrand is nonintegrable. The target law on RNq\mathbb R^{Nq}RNq is G(A)G(A)G(A), where A(i,k),(j,k′)=1{k=k′}Ch(Xi,Xj)A_{(i,k),(j,k')}=\mathbf1_{\{k=k'\}}C_h(X_i,X_j)A(i,k),(j,k′)​=1{k=k′}​Ch​(Xi​,Xj​); the limit random variable is the identity map on this probability space and is almost everywhere measurable. Convergence means weak convergence of the output laws along the following nested-tail filter: when h≥1h\ge1h≥1, a property P(w)P(w)P(w) holds eventually precisely when ∃Mh−1 ∀wh−1≥Mh−1 ∃Mh−2 ∀wh−2≥Mh−2 ⋯ ∃M0 ∀w0≥M0, P(w)\exists M_{h-1}\ \forall w_{h-1}\ge M_{h-1}\ \exists M_{h-2}\ \forall w_{h-2}\ge M_{h-2}\ \cdots\ \exists M_0\ \forall w_0\ge M_0,\ P(w)∃Mh−1​ ∀wh−1​≥Mh−1​ ∃Mh−2​ ∀wh−2​≥Mh−2​ ⋯ ∃M0​ ∀w0​≥M0​, P(w), with the earlier-coordinate thresholds allowed to depend on the already fixed later coordinates; equivalently, the required approximation holds in that nested-tail sense for every bounded continuous test function and every positive accuracy. Thus the first hidden width occurs in the innermost tail and the last hidden width in the outermost tail; no common growth rate or coupling between different width probability spaces is specified. The claim holds jointly for every fixed finite input family and all qqq output coordinates. It includes K=0K=0K=0, repeated or zero inputs, and N=0N=0N=0, in which case the observed vector has no coordinates. For h=0h=0h=0 there is one empty width tuple and the network is the single affine layer above; the filter is concentrated at that tuple, so the convergence assertion requires its law to equal the target law exactly. No differentiability assumption on σ\sigmaσ is imposed; input and output dimensions and β\betaβ are strictly positive, and every hidden width is at least one.

Human review
  • Endorsed by Shuze Chen · Sep 26, 2026

    Confirmed by the moderator at approval.

  • Endorsed by Minghui · Sep 26, 2026

    Confirmed by the mission captain (proposal self-audit).

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