arXiv · 2207.04932
Stochastic Gradient Descent and Anomaly of Variance-flatness Relation in Artificial Neural Networks
Abstract
Stochastic gradient descent (SGD), a widely used algorithm in deep-learning neural networks has attracted continuing studies for the theoretical principles behind its success. A recent work reports an anomaly (inverse) relation between the variance of neural weights and the landscape flatness of the loss function driven under SGD [Feng & Tu, PNAS 118, 0027 (2021)]. To investigate this seemingly violation of statistical physics principle, the properties of SGD near fixed points are analysed via a dynamic decomposition method. Our approach recovers the true "energy" function under which the universal Boltzmann distribution holds. It differs from the cost function in general and resolves the paradox raised by the the anomaly. The study bridges the gap between the classical statistical mechanics and the emerging discipline of artificial intelligence, with potential for better algorithms to the latter.
Explore related subjects
Keep this discovery
Xia Xiong, Yong-Cong Chen, Chunxiao Shi, Ping Ao. 2022-07-11. Stochastic Gradient Descent and Anomaly of Variance-flatness Relation in Artificial Neural Networks. https://doi.org/10.1088/0256-307x%2F40%2F8%2F080202
Cite the original work for its findings. Save a collection to share your selection of sources.