Spherical motion dynamics: learning dynamics of normalized neural network using SGD and weight decay

Sun, Jian, Wan, Ruosi, Zhang, Xiangyu and Zhu, Zhanxing (2021) Spherical motion dynamics: learning dynamics of normalized neural network using SGD and weight decay. In, Ranzato, M., Beygelzimer, A., Dauphin, Y., Liang, P.S. and Wortman Vaughan, J. (eds.) Advances in Neural Information Processing Systems 34. (Advances in Neural Information Processing Systems, 34) 35th Conference on Neural Information Processing Systems (06/12/21 - 14/12/21) Neural Information Processing Systems Foundation, pp. 6380-6391.

Abstract

In this paper, we comprehensively reveal the learning dynamics of normalized neural network using Stochastic Gradient Descent (with momentum) and Weight Decay (WD), named as Spherical Motion Dynamics (SMD). Most related works on this topic focus on studying “effective learning rate" using “equilibrium" assumption, i.e. assuming weight norm has converge to a fixed value. However, their discussion on why equilibrium can be reached is either absent or unjustified. To clarify the mechanism behind, our work directly explores the cause of equilibrium, which should be regarded as a special state of SMD. Specifically, 1) we introduce the assumptions that can lead to equilibrium state in SMD, and prove equilibrium can be reached in a linear rate regime; 2) we propose “angular update" as a substitute for effective learning rate to depict the state of SMD, and derive the theoretical value of angular update in equilibrium state; 3) we verify our assumptions and theoretical results on various large-scale computer vision tasks including ImageNet and MSCOCO with standard settings. Experiment results show our theoretical findings agree well with empirical observations. Furthermore, we provide intuitive interpretations, showing how the behavior of angular update in SMD affects the optimization of neural network, and yields unexpected phenomenon in practice. We believe our findings and theoretical results can deepen our understanding on current training techniques for deep neural network.

This record has no associated files available for download.

Contributors

Download statistics