arXiv:2410.07451cs.LGphysics.comp-ph2024-10被引 3

通过神经正切核谱熵变化,揭示网络学习动态与深层学习的开启机制。

Collective variables of neural networks: empirical time evolution and scaling laws

  • 用NTK谱熵和迹分析网络表示演化过程。
  • 小网络以熵减实现信息压缩,大网络熵增形成新结构。
  • 发现深度学习本质是结构生成,适合研究模型训练机制者阅读。

本文提出一种理解神经网络学习动态与缩放规律的新方法。通过分析经验神经正切核(NTK)谱的熵与迹,揭示了网络所学表示及其随架构缩放的优化路径。实验覆盖多种复杂架构,包括Transformer、自编码器、图神经网络及强化学习模型。研究发现,训练过程中存在两种普遍机制:一是小网络中熵减少的信息压缩;二是大网络中熵增加的结构形成,即表示突破初始化先验并生成丰富特征。由于后者在深度网络中的普遍性及其对特征表达的灵活性,我们主张其作为深度学习阶段开启的标志。

原文摘要 · Abstract (English)

This work presents a novel means for understanding learning dynamics and scaling relations in neural networks. We show that certain measures on the spectrum of the empirical neural tangent kernel, specifically entropy and trace, yield insight into the representations learned by a neural network and how these can be improved through architecture scaling. These results are demonstrated first on test cases before being shown on more complex networks, including transformers, auto-encoders, graph neural networks, and reinforcement learning studies. In testing on a wide range of architectures, we highlight the universal nature of training dynamics and further discuss how it can be used to understand the mechanisms behind learning in neural networks. We identify two such dominant mechanisms present throughout machine learning training. The first, information compression, is seen through a reduction in the entropy of the NTK spectrum during training, and occurs predominantly in small neural networks. The second, coined structure formation, is seen through an increasing entropy and thus, the creation of structure in the neural network representations beyond the prior established by the network at initialization. Due to the ubiquity of the latter in deep neural network architectures and its flexibility in the creation of feature-rich representations, we argue that this form of evolution of the network's entropy be considered the onset of a deep learning regime.

神经网络学习动态深度学习结构形成

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。