Loading summary...
11 min read (68% time saved)
Traditional machine learning theory posits a U-shaped test error curve where increasing model complexity beyond an optimal point leads to overfitting and worse performance; however, modern deep learning exhibits a "double descent" phenomenon where test error can decrease again in highly overparameterized models, challenging this long-held belief and suggesting a more nuanced understanding of generalization.
The video introduces the conventional understanding of machine learning model performance, highlighting that foundational AI books present a U-shaped test set error curve. This curve indicates that as model size increases, training error decreases, but test error initially decreases to a minimum before rising sharply due to overfitting. The concept is illustrated with polynomial curve fitting, showing how higher-order polynomials can perfectly fit noisy training data but perform poorly on unseen test data, leading to the characteristic U-shape.
In 2012, AlexNet, a large neural network, successfully used regularization techniques like data augmentation, dropout, and weight decay to combat overfitting. These methods were considered crucial for large models operating in the overfitting region of the bias-variance curve. The prevailing belief was that without regularization, such models would dramatically overfit, and regularization pushed them back towards optimal generalization. A key implication was that lower training error in the overfitting regime was causally linked to higher test error.
In 2016, a Google Brain paper challenged the understanding of generalization in deep learning. Experiments showed that deep models could perfectly memorize randomized labels from datasets like ImageNet and CIFAR, even with regularization, yet still generalize well when trained on correct labels. Furthermore, for more modern deep architectures like Inception v3, regularization was found to be less critical, making only modest improvements to test set performance without significantly impacting training error. This contradicted the idea that regularization primarily worked by moving models away from overfitting the training data.
This section is a sponsored message for KiwiCo, highlighting their hands-on project kits for children. The presenter shares personal anecdotes about his children engaging with KiwiCo products, emphasizing their educational value in promoting spatial reasoning and abstract thinking. He encourages viewers to use a special code for discounts on their first crates.
In 2018, Myle Belulin's team proposed the "double descent" hypothesis, suggesting that the traditional U-shaped bias-variance curve might not be the full picture. They hypothesized that if model size continued to increase beyond the overfitting regime, test set error could actually decrease again, leading to a W-shaped curve. Small-scale demonstrations on the MNIST dataset using random Fourier feature models showed this phenomenon, where test performance dramatically improved in a new regime as model size increased further.
In 2019, a Harvard and OpenAI team definitively confirmed the double descent phenomenon across various deep learning architectures, including transformers, and on both vision and language datasets. Crucially, they observed double descent not only as a function of model size but also of training time. This implies that stopping training when test error initially rises, a common practice, might prevent models from reaching a second, lower error minimum. Adding label noise to datasets made the double descent curve more pronounced, suggesting its relevance for noisier, real-world data.
Double descent can also be demonstrated with simple polynomial curve fitting. While a second-order polynomial provides a good fit, increasing the order to three or four leads to overfitting and increased test error, reaching the "interpolation threshold" where the model perfectly fits the training data. However, moving to a fifth-order polynomial, with more parameters than data points, allows for an infinite number of perfect fits. The solver, by choosing the solution with the smallest sum of squared coefficients (lowest L2 norm), selects a smoother curve that actually reduces test set error, initiating the double descent. This behavior continues with even higher-order polynomials.
The explanation for double descent lies in the model's flexibility and the training algorithm's ability to find solutions. At the interpolation threshold, the model has just enough capacity to perfectly fit the training data, making it highly susceptible to noise because there's only one unique fit. In contrast, for overparameterized models (beyond the interpolation threshold), there are many possible interpolating solutions. Training algorithms like stochastic gradient descent (SGD) can find smoother, less chaotic, and lower-norm solutions among these possibilities, which generalize better to new data. This increased flexibility allows the model to absorb noise while maintaining good performance on the underlying distribution.
The video addresses whether traditional machine learning books, which feature the U-shaped curve, are now obsolete. Trevor Hasty, a co-author of "The Elements of Statistical Learning," has acknowledged double descent, co-authoring a paper on it and updating his book to include a section on the phenomenon. Hasty and his co-authors argue that double descent doesn't contradict the bias-variance tradeoff but rather highlights that the measure of model complexity (or flexibility) changes after the interpolation threshold. Beyond this point, the lowest-norm solutions chosen by solvers are often simpler than the single, contorted solution at the threshold.
The bias-variance tradeoff is explained in detail using polynomial curve fitting. Bias refers to the difference between the average fit and the true underlying function, while variance measures the variability of fits across different data samples. For a second-order fit, variance is the largest error component. A first-order fit has high bias, as it cannot capture the underlying function. Third-order fits have low bias but enormous variance, making test error dominated by variability. This classic U-shaped region demonstrates the tradeoff: increasing complexity reduces bias but increases variance. However, beyond the interpolation threshold, the solver's ability to choose smoother, lower-norm solutions reduces the overall variance of the fits, leading to the double descent behavior. While error can still be decomposed into bias and variance, their tradeoff is no longer the primary driver of test error changes in this regime.
The presenter reflects on how the U-shaped curve deeply influenced his understanding of machine learning for years, emphasizing how easily prominent theories can be overgeneralized. He notes that double descent behavior is not universal and depends on factors like data noise and a model's "inductive bias" – how it handles overparameterized cases. Deep models, despite their capacity for catastrophic overfitting, exhibit a "friendly inductive bias" that allows them to generalize remarkably well. He concludes that deep learning theory is still evolving, comparing the shift from the bias-variance tradeoff to double descent as moving from Newtonian physics to Einstein's general relativity, expressing excitement for future theoretical developments.
The presenter announces his new book, "The Welch Labs Illustrated Guide to AI," which is available for preorder. He highlights its visual-first approach with hundreds of figures, including full-page spreads explaining complex concepts like loss landscapes. The book includes supporting Python code, a GitHub repository, and exercises with solutions. It covers fundamentals like perceptrons and backpropagation, as well as cutting-edge topics such as neural scaling laws, mechanistic interpretability, and AI image generation. The book is designed for self-study, AI courses, or as a coffee table book, and is available for US pre-orders with international shipping on a waitlist.
"The takeaway for a whole generation of machine learning practitioners was that we must carefully limit the power of our models to match the complexity of our data to avoid overfitting." — Stephen Welch
"But it's also wrong." — Stephen Welch
"Understanding Deep Learning Requires Rethinking Generalization." — Stephen Welch (referencing Google Brain paper title)
"So deep models even with regularization in place are perfectly capable of just memorizing their training data." — Stephen Welch
"If you were training one of these models and assumed a classical bias variance trade-off behavior, you would likely stop training long before you saw the double descent." — Stephen Welch
"Our worst generalizing fit is precisely at the interpolation threshold where our model exactly fits our training data for the first time." — Stephen Welch
"So, as Hasty and his co-authors state, technically the bias variance trade-off theory does not mean that our curve has to be U-shaped and the shape of our curve will depend on how we measure the flexibility or capacity of our models." — Stephen Welch
"It's just so easy to accept and in this case overgeneralize theories like this, especially when they fit so nicely into a simple mental image." — Stephen Welch
"If the bias variance trade-off is like Newtonian physics, it feels like we're getting glimpses of Einstein's general relativity with double descent." — Stephen Welch
Summarize another video
Press ⌘K to quickly paste a new URL
Anthropic
10 min read
No Priors: AI, Machine Learning, Tech, & Startups
9 min read
Sebastian Raschka
6 min read
No Priors: AI, Machine Learning, Tech, & Startups
7 min read
![He Co-Invented the Transformer. Now: Continuous Thought Machines [Llion Jones / Luke Darlow]](/_next/image?url=https%3A%2F%2Fimg.youtube.com%2Fvi%2FDtePicx_kFY%2Fmaxresdefault.jpg&w=3840&q=75)
Machine Learning Street Talk
9 min read
AI News & Strategy Daily | Nate B Jones
6 min read