arXiv · 2304.04784
Criticality versus uniformity in deep neural networks
Abstract
Deep feedforward networks initialized along the edge of chaos exhibit exponentially superior training ability as quantified by maximum trainable depth. In this work, we explore the effect of saturation of the tanh activation function along the edge of chaos. In particular, we determine the line of uniformity in phase space along which the post-activation distribution has maximum entropy. This line intersects the edge of chaos, and indicates the regime beyond which saturation of the activation function begins to impede training efficiency. Our results suggest that initialization along the edge of chaos is a necessary but not sufficient condition for optimal trainability.
Explore related subjects
Keep this discovery
Aleksandar Bukva, Jurriaan de Gier, Kevin T. Grosvenor, Ro Jefferson, Koenraad Schalm, Eliot Schwander. 2023-04-10. Criticality versus uniformity in deep neural networks. https://arxiv.org/abs/2304.04784
Cite the original work for its findings. Save a collection to share your selection of sources.