FoundationsChecking saved investigationReading browser-local route memory before showing a continuation.

Foundation Lab

Neural Tangent Kernel & Infinite-Width Limits

NTK provides a mathematically clean limit where we can predict learning dynamics and generalization

Concept 6 of 100TheoryPhase 3
#6NTKTheory
key equation\Theta(x,x') = \nabla_\theta f_\theta(x)^\top \nabla_\theta f_\theta(x')

Selected Foundation Object

Keep the equation fixed; move through the evidence.

Concept 6 of 100NTKTheory / Phase 3: Optimization & generalization
Current question

Most expositions are algebraic; missing is a geometric animation showing how trajectories in function space under NTK differ from genuine feature learning

\Theta(x,x') = \nabla_\theta f_\theta(x)^\top \nabla_\theta f_\theta(x')
PredictionCommit before the demo.

Ask what should change when the equation is manipulated, then let the visualization test that expectation.

EvidenceCompare local witness and source.

Use the runnable panel, the key equation, and canonical papers as separate forms of evidence for the same object.

InvariantName what survives notation changes.

The useful learning product is the reusable mechanism you can carry into another model, paper, or engineering tradeoff.

Next moveContinue in the newer notebook.

This atlas page keeps the working demo; the domain notebook carries the fuller Intuition -> Math -> Code -> Demo sequence.

Why It Matters for Modern Models

  • NTK provides a mathematically clean limit where we can predict learning dynamics and generalization
  • Many mechanistic-interpretability arguments assume behavior "somewhere between" kernel-like and feature-learning regimes

What Tutorials Skip

What is still poorly explained in textbooks and papers:

  • Most expositions are algebraic; missing is a geometric animation showing how trajectories in function space under NTK differ from genuine feature learning

Interactive Visualization

Core Math (Optional Deep Dive)

If you want intuition first, start with the key equation and the visualization. Come back here for the full walkthrough.

Key Equation
Θ(x,x)=θfθ(x)θfθ(x)\Theta(x,x') = \nabla_\theta f_\theta(x)^\top \nabla_\theta f_\theta(x')

Define network fθ(x)f_\theta(x) with parameters θ\theta. The NTK is:

Θ(x,x)=θfθ(x)θfθ(x)\Theta(x,x') = \nabla_\theta f_\theta(x)^\top \nabla_\theta f_\theta(x')

In the infinite-width limit, this kernel becomes deterministic and remains constant during training. Training becomes:

tft(x)=iΘ(x,xi)(ft(xi),yi)f\partial_t f_t(x) = - \sum_{i} \Theta(x,x_i)\, \frac{\partial \ell(f_t(x_i), y_i)}{\partial f}

a linear ODE in function space, just like kernel regression.

Canonical Papers

Neural Tangent Kernel: Convergence and Generalization in Neural Networks

Jacot et al.2018NeurIPS
Read paper →

Connections

Next Moves

Choose the next question to carry this object forward.