arXiv:2410.01779cs.LGcs.AI2024-10NeurIPS被引 4

用代数方法从部分解构造神经网络全局最优解,突破非线性瓶颈。

Composing Global Solutions to Reasoning Tasks via Algebraic Objects in Neural Nets

  • 利用半环代数结构,将部分解通过加乘运算组合成全局最优解。
  • 梯度下降获得的解中约95%与理论构造完全一致,验证方法有效性。
  • 适合研究神经网络优化机制或对代数推理感兴趣的读者。

我们证明了在阿贝尔群(如模加法)推理任务上,使用二次激活函数和 $L_2$ 损失的两层神经网络,其解空间具有丰富的代数结构。该结构使得即使在高度非线性情况下,也能通过分析方法从仅满足部分损失的局部解,构造出全局最优解。我们提出框架 CoGS(Compose Global Solutions),发现不同隐藏节点数下的权重空间具备半环结构,且目标损失由环同态形式的“求和势”构成,支持通过环加法和乘法组合局部解。实验表明,约95%的梯度下降解与理论构造完全匹配。尽管全局解仅需少量隐藏节点,但分析显示过参数化可渐近解耦训练动态,利于优化。此外,在权重衰减下,训练动态更偏好简单解,因此高阶全局解(如完美记忆)不被青睐。代码已开源:https://github.com/facebookresearch/luckmatters/tree/yuandong3/ssl/real-dataset。

原文摘要 · Abstract (English)

We prove rich algebraic structures of the solution space for 2-layer neural networks with quadratic activation and $L_2$ loss, trained on reasoning tasks in Abelian group (e.g., modular addition). Such a rich structure enables \emph{analytical} construction of global optimal solutions from partial solutions that only satisfy part of the loss, despite its high nonlinearity. We coin the framework as CoGS (\emph{\underline{Co}mposing \underline{G}lobal \underline{S}olutions}). Specifically, we show that the weight space over different numbers of hidden nodes of the 2-layer network is equipped with a semi-ring algebraic structure, and the loss function to be optimized consists of \emph{sum potentials}, which are ring homomorphisms, allowing partial solutions to be composed into global ones by ring addition and multiplication. Our experiments show that around $95\%$ of the solutions obtained by gradient descent match exactly our theoretical constructions. Although the global solutions constructed only required a small number of hidden nodes, our analysis on gradient dynamics shows that overparameterization asymptotically decouples training dynamics and is beneficial. We further show that training dynamics favors simpler solutions under weight decay, and thus high-order global solutions such as perfect memorization are unfavorable. The code is open sourced at https://github.com/facebookresearch/luckmatters/tree/yuandong3/ssl/real-dataset.

神经网络代数结构优化理论解空间

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