神经网络能用低秩分解高效学习群运算,揭示了模型在有限宽度下实现泛化的能力。
Learning words in groups: fusion algebras, tensor ranks and grokking
- 将词运算建模为三阶张量,利用群的共轭表示进行低秩分解
- 网络通过梯度下降找到低秩近似解,实现类似Strassen的高效矩阵乘法
- 揭示了模型在有限容量下实现泛化的机制,适用于理解深层网络的归纳偏置
本文证明,只要宽度足够,一个带有标准激活函数的两层神经网络能够学习任意有限群中的词运算,并在此过程中表现出
原文摘要 · Abstract (English)
In this work, we demonstrate that a simple two-layer neural network with standard activation functions can learn an arbitrary word operation in any finite group, provided sufficient width is available and exhibits grokking while doing so. To explain the mechanism by which this is achieved, we reframe the problem as that of learning a particular $3$-tensor, which we show is typically of low rank. A key insight is that low-rank implementations of this tensor can be obtained by decomposing it along triplets of basic self-conjugate representations of the group and leveraging the fusion structure to rule out many components. Focusing on a phenomenologically similar but more tractable surrogate model, we show that the network is able to find such low-rank implementations (or approximations thereof), thereby using limited width to approximate the word-tensor in a generalizable way. In the case of the simple multiplication word, we further elucidate the form of these low-rank implementations, showing that the network effectively implements efficient matrix multiplication in the sense of Strassen. Our work also sheds light on the mechanism by which a network reaches such a solution under gradient descent.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。