AI & ML
Finite-width shift of the edge of chaos in deep neural networks
arXiv stat.ML0 views
arXiv:2611.06582v1 Announce Type: cross Abstract: Deep neural networks train best when they are initialised at the edge of chaos, the critical line in the plane of weight and bias variances on which the distance between two inputs neither grows nor shrinks with depth. The line is known for infinitely wide networks; at finite width it moves, and how it moves has been disputed. We treat the logarithm of the distance between two inputs through a network of width $N$ as a random walk and compute its drift and variance to order $1/N$. The edge of chaos moves to a larger weight variance, by $\Delta\sigma_w^2=-a_1/(N\,\partial_{\sigma_w^2}\ln\chi_1)$, about $8/N$ for $\tanh$ networks at a bias variance of $0.05$, confirmed by simulations of the exact finite-width recursion. The shift describes a typical network; the average over many networks is dominated by rare networks in which the distance grows, and its critical point moves the other way, which explains why calculations and simulations have disagreed. Backpropagation follows the same random walk, so the shift applies to gradients unchanged. Along the edge of chaos, the information about the similarity of two inputs is constant until the compression over the depth exceeds about one nat, which fixes where to sit on the line. The results give an initialisation rule, Crinit, that places a network on its finite-width edge of chaos without a search. On MNIST and Fashion-MNIST at width $64$ it is more accurate than the standard rules at their usual settings, by a small margin in shallow networks and a large one in deep networks, where most standard rules fail; it trains in fewer steps, and it matches or beats the standard rules with a tuned gain.
Source:https://arxiv.org/abs/2611.06582PreviousUnderstanding the Out-of-Distribution Generalization of Chain-of-Thought Reasoning in LLMsNextRegularization vs. Convolution in Kernel-Based Hypothesis Testing
Related posts
A Framework for Context Enrichment and Selective Robustness in LLM Social Agents
0arXiv cs.MA
Airlock: Bounding the Multi-Agent Blast Radius via Selective Online Containment
0arXiv cs.MA
Mechanical Engineering Skills: An Open Library of Agent Skills for Design, Analysis and Manufacturing
0arXiv cs.MA
Curvature-Calibrated Quasi-Bayesian Updating for Moment-Restricted Models
0arXiv stat.TH
Ordinal Credit Assignment: Multi-Step Reinforcement Learning from Comparisons Alone
0arXiv stat.TH
Optimal estimation of generalized causal effects in cluster-randomized trials with multiple outcomes
0arXiv stat.TH