[2201.02177] Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets
In this paper we propose to study generalization of neural networks on small algorithmically generated datasets. In this setting, questions about data efficiency, memorization, generalization, and speed of learning can be studied in great detail. In some situations we show that neural networks learn through a process of "grokking" a pattern in the data, improving generalization performance from random chance level to perfect generalization, and that this improvement in generalization can happen well past the point of overfitting. We also study generalization as a function of dataset size and find that smaller datasets require increasing amounts of optimization for generalization. We argue that these datasets provide a fertile ground for studying a poorly understood aspect of deep learning: generalization of overparametrized neural networks beyond memorization of the finite training dataset.
In this paper we propose to study generalization of neural networks on small algorithmically generated datasets. In this setting, questions about data efficiency, memorization, generalization, and speed of learning can be studied in great detail. In some situations we show that neural networks learn through a process of "grokking" a pattern in the data, improving generalization performance from random chance level to perfect generalization, and that this improvement in generalization can happen well past the point of overfitting. We also study generalization as a function of dataset size and f
Explore this link on the map →related reading
- Clare Lyle | What's grokking good for?clarelyle.com
- Just Ask for Generalization | Eric Jangevjang.com
- The Scaling Hypothesis · Gwern.netgwern.net
- A Theory of Deep Learning | Elements of a Vector Spaceelonlit.com
- arxiv.org/pdf/1805.08522arxiv.org
- [2503.02113] Deep Learning is Not So Mysterious or Differentarxiv.org
- Human-like Neural Nets by Catapulting · Gwern.netgwern.net
- A Recipe for Training Neural Networkskarpathy.github.io
- Zipfian grokking | Jasper Gilleyjagilley.github.io
- The Little Book of Deep Learningfleuret.org
- Do Machine Learning Models Memorize or Generalize?pair.withgoogle.com
- Maybe I was too harsh on deep learning theory (three days ago) — LessWronglesswrong.com