Phase diagram for two-layer relu neural networks at infinite-width limit

T Luo, ZQJ Xu, Z Ma, Y Zhang - Journal of Machine Learning Research, 2021 - jmlr.org
T Luo, ZQJ Xu, Z Ma, Y Zhang
Journal of Machine Learning Research, 2021jmlr.org
How neural network behaves during the training over different choices of hyperparameters
is an important question in the study of neural networks. In this work, inspired by the phase
diagram in statistical mechanics, we draw the phase diagram for the two-layer ReLU neural
network at the infinite-width limit for a complete characterization of its dynamical regimes
and their dependence on hyperparameters related to initialization. Through both
experimental and theoretical approaches, we identify three regimes in the phase diagram …
How neural network behaves during the training over different choices of hyperparameters is an important question in the study of neural networks. In this work, inspired by the phase diagram in statistical mechanics, we draw the phase diagram for the two-layer ReLU neural network at the infinite-width limit for a complete characterization of its dynamical regimes and their dependence on hyperparameters related to initialization. Through both experimental and theoretical approaches, we identify three regimes in the phase diagram, i.e., linear regime, critical regime and condensed regime, based on the relative change of input weights as the width approaches infinity, which tends to 0, O(1) and +∞, respectively. In the linear regime, NN training dynamics is approximately linear similar to a random feature model with an exponential loss decay. In the condensed regime, we demonstrate through experiments that active neurons are condensed at several discrete orientations. The critical regime serves as the boundary between above two regimes, which exhibits an intermediate nonlinear behavior with the mean-field model as a typical example. Overall, our phase diagram for the two-layer ReLU NN serves as a map for the future studies and is a first step towards a more systematical investigation of the training behavior and the implicit regularization of NNs of different structures.
jmlr.org
以上显示的是最相近的搜索结果。 查看全部搜索结果

Google学术搜索按钮

example.edu/paper.pdf
搜索
获取 PDF 文件
引用
References