Understanding the Neural Tangent Kernel – Rajat's Blog – A blog about machine learning and math.
A flurry of recent papers in theoretical deep learning tackles the common theme of analyzing neural networks in the infinite-width limit. At first, this limit may seem impractical and even pointless to study. However, it turns out that neural networks in this regime simplify to linear models with a kernel called the neural tangent kernel. Gradient descent is therefore very simple to study. While this may seem promising at first, empirical results show that neural networks in this regime perform worse than practical over-parameterized networks. Nevertheless, this still provides theoretical insight into some aspects of neural network training, so it is worth studying. Additionally, this kernel regime can occur under a more general criterion which depends on the scale of the model, and does not require infinite width per se. In this post, I’ll present a simple and intuitive introduction to this theory that leads to a proof of convergence of gradient descent to 0 training loss. I’ll make u
Understanding the Neural Tangent Kernel This gif depicts the training dynamics of a neural network. Find out how by reading the rest of this post. A flurry of recent papers in theoretical deep learning tackles the common theme of analyzing neural networks in the infinite-width limit. At first, this limit may seem impractical and even pointless to study. However, it turns out that neural networks in this regime simplify to linear models with a kernel called the neural tangent kernel . Gradient descent is therefore very simple to study. While this may seem promising at first, empirical results s
Explore this link on the map →related reading
- Understanding the Neural Tangent Kernel – EigenTaleseigentales.com
- Some Math behind Neural Tangent Kernel | Lil'Loglilianweng.github.io
- Neural tangent kernel - Wikipediaen.wikipedia.org
- Infinite Limits of Neural Networks - Kempner Institutekempnerinstitute.harvard.edu
- [1806.07572] Neural Tangent Kernel: Convergence and Generalization in Neural Networksarxiv.org
- [2604.21691] There Will Be a Scientific Theory of Deep Learningarxiv.org
- [2605.01172] A Theory of Generalization in Deep Learningarxiv.org
- [2304.03408] Dynamics of Finite Width Kernel and Prediction Fluctuations in Mean Field Neural Networksarxiv.org
- Greg Yang | Professional pagethegregyang.com
- Maybe I was too harsh on deep learning theory (three days ago) — LessWronglesswrong.com
- A Theory of Deep Learning | Elements of a Vector Spaceelonlit.com
- Neural network training makes beautiful fractals | Jascha’s blogsohl-dickstein.github.io