提出随机矩阵模型解释神经网络中重尾现象的成因。
Models of Heavy-Tailed Mechanistic Universality
- 构建高温马尔琴科-帕斯图模型,揭示重尾谱密度的三重机制。
- 重尾行为由数据复杂相关性、训练温度低与特征向量熵减共同导致。
- 适用于理解模型性能、优化轨迹及训练相变等深层规律。
深度学习近期的理论与实证成功,如著名的神经网络缩放定律,伴随着一个显著观察:诸多关键对象常表现出重尾或幂律行为。特别是雅可比矩阵、海森矩阵和权重矩阵中重尾谱密度的普遍性,催生了‘重尾机制普适性’(HT-MU)概念。多项实证证据表明,重尾度量与模型性能存在稳健关联,暗示其可能是深度学习有效性的基础特征。本文提出一类通用的随机矩阵模型——高温马尔琴科-帕斯图(HTMP)系综,以探究训练神经网络中重尾行为的成因。该模型显示,上下尾部的幂律谱密度源于三个独立因素的耦合:数据中的复杂相关结构、训练过程中的低温效应以及特征向量熵降低,这些以隐式偏置形式存在于模型结构中,并可通过‘本征值排斥’参数调控。本文还讨论了该模型对其他重尾现象的启示,包括神经网络缩放定律、优化器轨迹及五加一阶段的神经网络训练过程。
原文摘要 · Abstract (English)
Recent theoretical and empirical successes in deep learning, including the celebrated neural scaling laws, are punctuated by the observation that many objects of interest tend to exhibit some form of heavy-tailed or power law behavior. In particular, the prevalence of heavy-tailed spectral densities in Jacobians, Hessians, and weight matrices has led to the introduction of the concept of heavy-tailed mechanistic universality (HT-MU). Multiple lines of empirical evidence suggest a robust correlation between heavy-tailed metrics and model performance, indicating that HT-MU may be a fundamental aspect of deep learning efficacy. Here, we propose a general family of random matrix models -- the high-temperature Marchenko-Pastur (HTMP) ensemble -- to explore attributes that give rise to heavy-tailed behavior in trained neural networks. Under this model, spectral densities with power laws on (upper and lower) tails arise through a combination of three independent factors (complex correlation structures in the data; reduced temperatures during training; and reduced eigenvector entropy), appearing as an implicit bias in the model structure, and they can be controlled with an "eigenvalue repulsion" parameter. Implications of our model on other appearances of heavy tails, including neural scaling laws, optimizer trajectories, and the five-plus-one phases of neural network training, are discussed.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。