New proof unifies Dropout and Gradient Masking for deep ResNets
Dropout and Random Gradient Masking: two different methods, but at scale they become one.
Dropout and Random Gradient Masking (RaM) are two popular but seemingly distinct regularization methods in deep learning. Dropout randomly zeros out activations during the forward pass to prevent co-adaptation, while RaM leaves the forward pass untouched and instead randomly masks gradients during backpropagation. Because RaM introduces unbiased noise into parameter updates, standard explanations for dropout's effectiveness—such as penalization of weights or neuron co-adaptation avoidance—do not apply. This has left researchers puzzled about whether they are fundamentally different or secretly connected.
In a new arXiv paper, Javier Maass and Lénaïc Chizat prove that for ResNets in the large depth and width asymptotics (the complete feature learning regime), Dropout and RaM converge to the same limiting dynamics. Their analysis covers multiple variants, including the layerwise dropout used in stochastic-depth ResNets, and shows that all these methods collapse to the same limit as network scale increases. This theoretical equivalence provides a unifying framework: at sufficient scale, the way you inject randomness matters less than the fact that you inject it. For practitioners, the result means techniques can be chosen based on computational convenience rather than presumed differing effects, and future research can focus on a single effective regularization approach for large models.
- Dropout randomly masks activations in the forward pass; RaM masks gradients in the backward pass without altering forward activations.
- The two methods become asymptotically equivalent for ResNets in the large depth and width limit, in the complete feature learning regime.
- Several variants (including stochastic-depth dropout) collapse to the same limiting dynamics, suggesting a universal regularization behavior at scale.
Why It Matters
Unifies two major training techniques, simplifying theoretical understanding and potentially guiding future algorithm design.