AI Research Scientist · Generalization Theory & Phenomena
Neural tangent kernel linearization regime
Neural tangent kernel linearization regime
NTK linearization regime
As width grows to infinity, a network’s output is well-approximated by its first-order Taylor expansion around initialization, so training reduces to linear regression in a fixed feature space defined by the NTK.Because training is just linear regression over a fixed feature space, gradient descent converges to one well-defined, predictable solution — the same guarantees classical kernel regression enjoys. That's the whole payoff: wide networks generalize predictably not because their optimization is mysterious, but because it secretly isn't nonlinear optimization at all.
- 1Write .WhyAt large width, weights barely move from initialization, so this first-order term already captures almost all the change.
- 2Treat as a fixed feature map instead of something that evolves.WhySince the gradient stays nearly constant during training, it acts like a static kernel feature vector, not a learned representation.
- 3Fit by gradient descent on these fixed features.WhyThis is exactly linear regression with the NTK as kernel, so the dynamics are convex and predictable.
Recall check from the same lesson
Because a network's training sits in the NTK linearization regime, that fact alone lets you compute the exact NTK formula for that specific architecture.
Review the explanation
Answer: False. The regime explains WHY wide-network training behaves like linear kernel regression and generalizes predictably — it does not hand you the specific kernel formula for an architecture. Deriving that formula, and finite-width correction theory, are separate topics beyond what this regime claims.
Sources
One sitting · 20–30 minutes
A focused session on your AI Research Scientist interview
LearnBench starts from what you already know — skip what you have, master what you’re missing.
Start now