arXiv:2510.19382stat.MLcs.LG2025-10

提出去随机化框架,揭示神经网络权重的低秩结构成因。

A Derandomization Framework for Structure Discovery: Applications in Neural Networks and Beyond

  • 基于去随机化引理,从优化目标出发解释权重结构形成机制。
  • 在弱假设下证明任意深度网络训练至二阶驻点时仍出现低秩结构。
  • 适用于多种模型与优化方法,对结构发现和近似算法有启发意义。

理解神经网络中特征学习的动力学仍是重大挑战。此前研究(Mousavi-Hosseini 等,2023)分析了多指标师生设置,发现使用随机梯度下降(SGD)和强正则化训练的两层学生网络其第一层权重呈现低秩结构,该性质可降低泛化所需的样本复杂度。本文聚焦结构发现本身,在更弱假设下进行研究:允许任意规模与深度的神经网络、所有参数可训练、任意光滑损失函数、微弱正则化,以及任意能收敛到二阶驻点(SOSP)的优化方法(如扰动梯度下降,PGD)。核心是提出一个关键的去随机化引理:在温和条件下,优化函数 $\mathbb{E}_{\mathbf{x}} \left[g_θ(\mathbf{W}\mathbf{x} + \mathbf{b})\right]$ 会收敛至 $\mathbf{W} = \mathbf{0}$,这一根本性质直接解释了结构发现现象,并可立即应用于其他领域,如端到端的 MAXCUT 近似和 Johnson-Lindenstrauss 嵌入计算。

原文摘要 · Abstract (English)

Understanding the dynamics of feature learning in neural networks (NNs) remains a significant challenge. The work of (Mousavi-Hosseini et al., 2023) analyzes a multiple index teacher-student setting and shows that a two-layer student attains a low-rank structure in its first-layer weights when trained with stochastic gradient descent (SGD) and a strong regularizer. This structural property is known to reduce sample complexity of generalization. Indeed, in a second step, the same authors establish algorithm-specific learning guarantees under additional assumptions. In this paper, we focus exclusively on the structure discovery aspect and study it under weaker assumptions, more specifically: we allow (a) NNs of arbitrary size and depth, (b) with all parameters trainable, (c) under any smooth loss function, (d) tiny regularization, and (e) trained by any method that attains a second-order stationary point (SOSP), e.g.\ perturbed gradient descent (PGD). At the core of our approach is a key $\textit{derandomization}$ lemma, which states that optimizing the function $\mathbb{E}_{\mathbf{x}} \left[g_θ(\mathbf{W}\mathbf{x} + \mathbf{b})\right]$ converges to a point where $\mathbf{W} = \mathbf{0}$, under mild conditions. The fundamental nature of this lemma directly explains structure discovery and has immediate applications in other domains including an end-to-end approximation for MAXCUT, and computing Johnson-Lindenstrauss embeddings.

结构发现去随机化神经网络

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