A Function-Space Dichotomy for Compositional Learning: Exponential Sub-Optimality of the Neural Tangent Kernel
This paper establishes a function-space dichotomy demonstrating that the Neural Tangent Kernel suffers from exponential sub-optimality compared to finite-width neural networks on compositional tasks, a gap driven by the mismatch between the kernel's smoothness bias and the target's architectural complexity rather than a generic kernel-versus-network limitation.
Original paper licensed under CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/). This is an AI-generated explanation of the paper below. It is not written or endorsed by the authors. For technical accuracy, refer to the original paper. Read full disclaimer
The Big Picture: Why "Lazy" Learning Fails at Building Complex Things
Imagine you are trying to teach a computer to recognize patterns. For a long time, researchers have used a tool called the Neural Tangent Kernel (NTK) to predict how well a neural network will learn. Think of the NTK as a "lazy" teacher. This teacher is very good at smoothing out rough edges and learning gentle curves, but they refuse to change their mind or learn new tricks. They just stick to a very rigid, pre-set way of looking at the world.
The paper asks a simple question: When does this "lazy" teacher fail, and why?
The authors discovered that the lazy teacher fails spectacularly when the task involves composition—building something complex by stacking simple layers on top of each other (like a Russian nesting doll or a recipe with many steps). In these cases, a "rich" teacher (a standard neural network that actually learns and changes its weights) is exponentially better than the lazy one.
The Two Ways to Measure Difficulty
To explain this, the authors look at a target function (the pattern you want the computer to learn) through two different lenses:
The "Smoothness" Lens (Fourier Complexity):
Imagine the target is a musical note. If the note is a low, smooth hum, it's easy to describe. If it's a super-fast, jagged squeal that vibrates thousands of times per second, it's "complex" in terms of smoothness.- The NTK's view: The lazy teacher hates jagged, fast-vibrating sounds. To learn a high-frequency squeal, the NTK needs a massive amount of data (samples) to figure it out. It treats every wiggle as a huge obstacle.
The "Architectural" Lens (Architectural Complexity):
Now, imagine you want to build a machine that makes that same squealing sound.- The Network's view: A standard neural network is like a master builder. Even if the sound is a crazy, fast squeal, the builder might be able to create it by stacking just a few simple gears (layers) together. The "cost" to build it is low, even if the result looks chaotic.
The Conflict: The paper shows that for certain tasks, the "Smoothness" cost is astronomical, but the "Architectural" cost is tiny. The lazy teacher (NTK) sees the astronomical cost and gives up, while the smart builder (neural network) sees the tiny cost and builds it easily.
The Star Example: The "Sawtooth" Wave
The authors use a specific shape called the Sawtooth to prove their point. Imagine a triangle wave that goes up and down.
- Depth 1: One triangle. Easy.
- Depth 2: Two triangles inside one.
- Depth 10: A wave that zig-zags back and forth thousands of times.
The Catch:
- For the Network: You can build this crazy, zig-zagging wave by stacking just 10 simple layers. It's a cheap, efficient construction.
- For the NTK: To the lazy teacher, this wave looks like it has a frequency of (over 1,000). Because the NTK is biased toward smoothness, it thinks this is incredibly hard.
The Result:
The paper proves that to learn this 10-layer sawtooth:
- The Network needs a manageable number of examples (polynomial growth).
- The NTK needs a number of examples that grows exponentially (like ).
- In plain English: By the time the depth is just 12, the lazy teacher would need 10 million times more data than the smart builder to get the same result.
The "Lazy" vs. "Rich" Regime
The paper distinguishes between two ways neural networks learn:
- The Lazy Regime (NTK): The network is so wide and trained so gently that its internal settings barely move. It acts like a fixed mathematical formula (a kernel). It's great for smooth, simple things but terrible for complex, layered things.
- The Rich Regime (Standard Training): The network actually changes its internal weights. It learns features. This allows it to build complex structures efficiently, even if they look messy.
What the Experiments Showed
The authors didn't just do math; they ran experiments to confirm their theory:
- Smooth Targets: When they gave the computer a smooth, simple wave (like a gentle sine wave), the lazy teacher (NTK) and the smart builder performed almost identically. The NTK is fine here.
- Complex Targets (Parity): They tested a "sparse parity" problem (a logic puzzle involving multiplying specific numbers together).
- The NTK was stuck at the bottom, guessing randomly, no matter how much data they gave it.
- The Neural Network learned the pattern quickly, beating the NTK by a factor of 10,000 to 1,000,000.
The Takeaway
The paper concludes that the gap between neural networks and kernel methods isn't just about "kernels vs. networks." It's about mismatched tools.
- If your problem is smooth and simple, the "lazy" kernel is a great, efficient tool.
- If your problem is built from layers of composition (like deep hierarchies or complex logic), the "lazy" tool is the wrong instrument. It sees a mountain of difficulty where a smart builder sees a simple staircase.
The "exponential sub-optimality" mentioned in the title simply means that for these specific types of complex, layered problems, using the lazy kernel approach is not just slightly worse; it is catastrophically inefficient compared to letting the network actually learn.
Drowning in papers in your field?
Get daily digests of the most novel papers matching your research keywords — with technical summaries, in your language.