从神经元激活看深度网络行为,揭示过参数化与泛化关系
Understanding Artificial Neural Network's Behavior from Neuron Activation Perspective
- 将神经元激活建模为随机过程,推导出激活数量与分布的数学表达式
- 发现损失函数随数据量呈幂律衰减,解释了过参数化下的泛化能力
- 理论可预测模型压缩性与参数效率,适合研究模型缩放规律者阅读
本文从神经元激活动态角度探索深度神经网络(DNN)的复杂行为。提出一个概率框架,将模型的神经元激活模式视为随机过程,揭示了神经网络缩放定律的理论依据,如过参数化及损失函数随数据集规模呈现幂律衰减的现象。通过推导关键数学关系,得出激活神经元数量呈 $N(1-(\frac{bN}{D+bN})^b)$ 形式,且神经元激活遵循幂律分布。基于此,阐明了在过参数化条件下仍能保持泛化能力的原因,并解释了在对数坐标下数据规模线性增长时损失曲线出现相变的现象。结合上述现象与激活的幂律分布,进一步推导出损失函数随数据规模增加而呈现幂律衰减。该分析弥合了经验观察与理论基础之间的差距,提出了可实验验证的参数效率与模型可压缩性预测。这些发现为理解神经网络缩放机制提供了基础,并指明优化DNN性能的新方向。
原文摘要 · Abstract (English)
This paper explores the intricate behavior of deep neural networks (DNNs) through the lens of neuron activation dynamics. We propose a probabilistic framework that can analyze models' neuron activation patterns as a stochastic process, uncovering theoretical insights into neural scaling laws, such as over-parameterization and the power-law decay of loss with respect to dataset size. By deriving key mathematical relationships, we present that the number of activated neurons increases in the form of $N(1-(\frac{bN}{D+bN})^b)$, and the neuron activation should follows power-law distribution. Based on these two mathematical results, we demonstrate how DNNs maintain generalization capabilities even under over-parameterization, and we elucidate the phase transition phenomenon observed in loss curves as dataset size plotted in log-axis (i.e. the data magnitude increases linearly). Moreover, by combining the above two phenomenons and the power-law distribution of neuron activation, we derived the power-law decay of neural network's loss function as the data size scale increases. Furthermore, our analysis bridges the gap between empirical observations and theoretical underpinnings, offering experimentally testable predictions regarding parameter efficiency and model compressibility. These findings provide a foundation for understanding neural network scaling and present new directions for optimizing DNN performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。