# Grokfast: Accelerated Grokking by Amplifying Slow Gradients

2405.20233

117

0

## Abstract

One puzzling artifact in machine learning dubbed grokking is where delayed generalization is achieved tenfolds of iterations after near perfect overfitting to the training data. Focusing on the long delay itself on behalf of machine learning practitioners, our goal is to accelerate generalization of a model under grokking phenomenon. By regarding a series of gradients of a parameter over training iterations as a random signal over time, we can spectrally decompose the parameter trajectories under gradient descent into two components: the fast-varying, overfitting-yielding component and the slow-varying, generalization-inducing component. This analysis allows us to accelerate the grokking phenomenon more than $times 50$ with only a few lines of code that amplifies the slow-varying components of gradients. The experiments show that our algorithm applies to diverse tasks involving images, languages, and graphs, enabling practical availability of this peculiar artifact of sudden generalization. Our code is available at https://github.com/ironjr/grokfast.

Get summaries of the top AI research delivered straight to your inbox:

## Overview

- The paper "Grokfast: Accelerated Grokking by Amplifying Slow Gradients" explores a technique to speed up the "grokking" process in deep neural networks.
- Grokking refers to the phenomenon where a neural network suddenly achieves high performance on a task after an initial period of slow learning.
- The authors propose a method called "Grokfast" that amplifies the low-frequency components of the stochastic gradients during training to accelerate grokking.

## Plain English Explanation

The paper discusses a challenge in training deep neural networks, which is the phenomenon of "grokking." Grokking as transition from lazy to rich Grokking is when a neural network suddenly starts performing very well on a task after a long period of slow progress.

The authors of this paper propose a technique called "Grokfast" to speed up this grokking process. Rationale from frequency perspective: grokking, training neural The key idea is to amplify the low-frequency components of the stochastic gradients used to train the network. Stochastic gradients are the small updates made to the network's parameters during training.

By boosting the low-frequency gradients, the network is able to more quickly find the "right" set of parameters that lead to high performance on the task. This is analogous to tuning a radio - you need to find the right frequency to get a clear signal, and amplifying the low frequencies helps you home in on that sweet spot faster.

The authors demonstrate through experiments that their Grokfast method can significantly accelerate the grokking process compared to standard training approaches. Deep grokking: would deep neural networks generalize This has important implications for making deep learning systems more sample-efficient and practical, especially for real-world applications.

## Technical Explanation

The core idea behind the "Grokfast" method proposed in this paper is to amplify the low-frequency components of the stochastic gradients used to train the deep neural network. Dichotomy: early late phase implicit biases can

The authors hypothesize that the low-frequency gradients are important for the "grokking" phenomenon, where the network suddenly achieves high performance after an initial period of slow progress. By selectively boosting these low-frequency gradients, they are able to accelerate the grokking process.

Specifically, the Grokfast method applies a frequency-dependent scaling to the stochastic gradients during training. Higher scaling factors are applied to the low-frequency components, while the high-frequency gradients are left unchanged. This creates a gradient signal that is biased towards the lower frequencies.

The authors evaluate their Grokfast method on a range of benchmark tasks and demonstrate significant improvements in the rate of grokking compared to standard training approaches. Progress measures for grokking on real-world datasets They analyze the learned representations and show that the Grokfast method leads to networks that converge to better minima in the optimization landscape.

## Critical Analysis

The Grokfast paper presents an intriguing approach to accelerating the grokking phenomenon in deep neural networks. The authors provide a compelling rationale for why amplifying low-frequency gradients could be beneficial, and their experimental results seem to support this hypothesis.

One potential limitation of the work is the reliance on carefully tuned hyperparameters to control the frequency-dependent scaling. The authors acknowledge that the optimal scaling factors may vary across different tasks and architectures, which could make the method less straightforward to apply in practice.

Additionally, while the authors demonstrate improvements on benchmark tasks, it's unclear how well the Grokfast method would generalize to more complex, real-world datasets. Progress measures for grokking on real-world datasets Further research would be needed to assess the broader applicability of this technique.

Another area for potential investigation is the relationship between the Grokfast method and other techniques that aim to improve the optimization dynamics of deep neural networks, such as deep grokking: would deep neural networks generalize or dichotomy: early late phase implicit biases can. Understanding how these different approaches interact could lead to more robust and effective training strategies.

Overall, the Grokfast paper presents a novel and promising direction for accelerating the grokking process in deep learning. While further research is needed to fully understand the implications and limitations of this approach, the authors have made a valuable contribution to the ongoing efforts to improve the training and generalization of deep neural networks.

## Conclusion

The paper "Grokfast: Accelerated Grokking by Amplifying Slow Gradients" introduces a novel technique to speed up the "grokking" phenomenon in deep neural networks. By selectively amplifying the low-frequency components of the stochastic gradients during training, the authors are able to significantly accelerate the process by which a network suddenly achieves high performance on a task.

This work has important implications for making deep learning systems more sample-efficient and practical, particularly for real-world applications where rapid learning is crucial. The authors' insights into the role of low-frequency gradients in the grokking process contribute to our fundamental understanding of deep neural network optimization and generalization.

While further research is needed to fully explore the limitations and broader applicability of the Grokfast method, this paper represents an exciting step forward in the quest to unlock the full potential of deep learning.

This summary was produced with help from an AI and may contain inaccuracies - check out the links to read the original source documents!

## Related Papers

### New!Deep Networks Always Grok and Here is Why

Ahmed Imtiaz Humayun, Randall Balestriero, Richard Baraniuk

0

0

Grokking, or delayed generalization, is a phenomenon where generalization in a deep neural network (DNN) occurs long after achieving near zero training error. Previous studies have reported the occurrence of grokking in specific controlled settings, such as DNNs initialized with large-norm parameters or transformers trained on algorithmic datasets. We demonstrate that grokking is actually much more widespread and materializes in a wide range of practical settings, such as training of a convolutional neural network (CNN) on CIFAR10 or a Resnet on Imagenette. We introduce the new concept of delayed robustness, whereby a DNN groks adversarial examples and becomes robust, long after interpolation and/or generalization. We develop an analytical explanation for the emergence of both delayed generalization and delayed robustness based on the local complexity of a DNN's input-output mapping. Our local complexity measures the density of so-called linear regions (aka, spline partition regions) that tile the DNN input space and serves as a utile progress measure for training. We provide the first evidence that, for classification problems, the linear regions undergo a phase transition during training whereafter they migrate away from the training samples (making the DNN mapping smoother there) and towards the decision boundary (making the DNN mapping less smooth there). Grokking occurs post phase transition as a robust partition of the input space thanks to the linearization of the DNN mapping around the training points. Website: https://bit.ly/grok-adversarial

6/10/2024

### Deep Grokking: Would Deep Neural Networks Generalize Better?

Simin Fan, Razvan Pascanu, Martin Jaggi

0

0

Recent research on the grokking phenomenon has illuminated the intricacies of neural networks' training dynamics and their generalization behaviors. Grokking refers to a sharp rise of the network's generalization accuracy on the test set, which occurs long after an extended overfitting phase, during which the network perfectly fits the training set. While the existing research primarily focus on shallow networks such as 2-layer MLP and 1-layer Transformer, we explore grokking on deep networks (e.g. 12-layer MLP). We empirically replicate the phenomenon and find that deep neural networks can be more susceptible to grokking than its shallower counterparts. Meanwhile, we observe an intriguing multi-stage generalization phenomenon when increase the depth of the MLP model where the test accuracy exhibits a secondary surge, which is scarcely seen on shallow models. We further uncover compelling correspondences between the decreasing of feature ranks and the phase transition from overfitting to the generalization stage during grokking. Additionally, we find that the multi-stage generalization phenomenon often aligns with a double-descent pattern in feature ranks. These observations suggest that internal feature rank could serve as a more promising indicator of the model's generalization behavior compared to the weight-norm. We believe our work is the first one to dive into grokking in deep neural networks, and investigate the relationship of feature rank and generalization performance.

5/31/2024

### Grokking as the Transition from Lazy to Rich Training Dynamics

Tanishq Kumar, Blake Bordelon, Samuel J. Gershman, Cengiz Pehlevan

0

0

We propose that the grokking phenomenon, where the train loss of a neural network decreases much earlier than its test loss, can arise due to a neural network transitioning from lazy training dynamics to a rich, feature learning regime. To illustrate this mechanism, we study the simple setting of vanilla gradient descent on a polynomial regression problem with a two layer neural network which exhibits grokking without regularization in a way that cannot be explained by existing theories. We identify sufficient statistics for the test loss of such a network, and tracking these over training reveals that grokking arises in this setting when the network first attempts to fit a kernel regression solution with its initial features, followed by late-time feature learning where a generalizing solution is identified after train loss is already low. We find that the key determinants of grokking are the rate of feature learning -- which can be controlled precisely by parameters that scale the network output -- and the alignment of the initial features with the target function $y(x)$. We argue this delayed generalization arises when (1) the top eigenvectors of the initial neural tangent kernel and the task labels $y(x)$ are misaligned, but (2) the dataset size is large enough so that it is possible for the network to generalize eventually, but not so large that train loss perfectly tracks test loss at all epochs, and (3) the network begins training in the lazy regime so does not learn features immediately. We conclude with evidence that this transition from lazy (linear model) to rich training (feature learning) can control grokking in more general settings, like on MNIST, one-layer Transformers, and student-teacher networks.

4/12/2024

### A rationale from frequency perspective for grokking in training neural network

Zhangchen Zhou, Yaoyu Zhang, Zhi-Qin John Xu

0

0

Grokking is the phenomenon where neural networks NNs initially fit the training data and later generalize to the test data during training. In this paper, we empirically provide a frequency perspective to explain the emergence of this phenomenon in NNs. The core insight is that the networks initially learn the less salient frequency components present in the test data. We observe this phenomenon across both synthetic and real datasets, offering a novel viewpoint for elucidating the grokking phenomenon by characterizing it through the lens of frequency dynamics during the training process. Our empirical frequency-based analysis sheds new light on understanding the grokking phenomenon and its underlying mechanisms.

5/29/2024