arXiv:2410.20107cs.LGcs.AI2024-10被引 1

揭示深度网络中隐藏表示相似性如何收敛到唯一稳定状态。

Emergence of Globally Attracting Fixed Points in Deep Neural Networks With Nonlinear Activations

  • 用赫米特多项式分析激活函数,推导出核映射的显式形式。
  • 非线性激活下,核序列全局收敛至唯一不动点,表征相似或正交表示。
  • 适用于带残差连接和归一化层的网络,为架构设计提供理论依据。

理解神经网络在各层对输入数据的变换机制,是揭示其学习与泛化能力的基础。尽管已有研究借助核方法分析神经网络,但对隐藏表示间相似性随层数演化的全局分析仍不充分。本文提出一个理论框架,用于分析核序列的演化,该序列衡量两个不同输入在隐藏层表示间的相似性。在均场假设下,我们证明核序列通过仅依赖于激活函数的核映射确定性演化。通过将激活函数展开为赫米特多项式并利用其代数性质,我们推导出核映射的显式表达式,并完全刻画其不动点。分析表明,对于非线性激活函数,核序列全局收敛至唯一不动点,该不动点对应于正交或相似的表示,具体取决于激活函数和网络结构。我们进一步将结果推广至带有残差连接和归一化层的网络,证实了类似的收敛行为。本工作为深度神经网络的隐式偏差提供了新见解,揭示了架构选择如何影响表示的跨层演化。

原文摘要 · Abstract (English)

Understanding how neural networks transform input data across layers is fundamental to unraveling their learning and generalization capabilities. Although prior work has used insights from kernel methods to study neural networks, a global analysis of how the similarity between hidden representations evolves across layers remains underexplored. In this paper, we introduce a theoretical framework for the evolution of the kernel sequence, which measures the similarity between the hidden representation for two different inputs. Operating under the mean-field regime, we show that the kernel sequence evolves deterministically via a kernel map, which only depends on the activation function. By expanding activation using Hermite polynomials and using their algebraic properties, we derive an explicit form for kernel map and fully characterize its fixed points. Our analysis reveals that for nonlinear activations, the kernel sequence converges globally to a unique fixed point, which can correspond to orthogonal or similar representations depending on the activation and network architecture. We further extend our results to networks with residual connections and normalization layers, demonstrating similar convergence behaviors. This work provides new insights into the implicit biases of deep neural networks and how architectural choices influence the evolution of representations across layers.

神经网络核方法表示演化不动点

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