通过群运算任务揭示深度学习如何处理序列结构计算。
Sequential Group Composition: A Window into the Mechanics of Deep Learning
- 用群元素序列的累积乘积任务研究网络学习机制。
- 两层网络需隐藏层宽指数级增长,而深层网络可线性/对数级优化。
- 适合研究神经网络对称性、结构化计算与架构效率的学者。
神经网络如何从序列训练中学会算术、几何和算法等结构化操作?我们引入了序列群组合任务:网络接收有限群元素的实向量编码序列,需预测其累积乘积。该任务具有顺序敏感性,无法被线性模型解决。我们的分析揭示了群结构、编码统计和序列长度对学习的影响。证明从零初始化的两层网络会逐个学习群的一个不可约表示,顺序由编码的傅里叶统计决定。要完美学习该任务,此类网络需隐藏层宽度指数级于序列长度 $k$。相比之下,我们构建了更深架构:循环网络可在 $k$ 步内顺序组合元素,多层网络则可在 $\ ext{log} k$ 层内并行组合相邻对。整体上,序列群组合任务为理解深度学习机制提供了可解析的窗口。
原文摘要 · Abstract (English)
How do neural networks trained over sequences acquire the ability to perform structured operations, such as arithmetic, geometric, and algorithmic computation? To gain insight into this question, we introduce the sequential group composition task. In this task, networks receive a sequence of elements from a finite group encoded in a real vector space and must predict their cumulative product. This task can be order-sensitive and cannot be solved by a linear model. Our analysis isolates the roles of the group structure, encoding statistics, and sequence length in shaping learning. We prove that two-layer networks from vanishing initialization learn this task one irreducible representation of the group at a time in an order determined by the Fourier statistics of the encoding. To perfectly learn the task, these networks require a hidden width exponential in the sequence length $k$. In contrast, we construct deeper architectures that exploit associativity to dramatically improve this scaling: recurrent neural networks can compose elements sequentially in $k$ steps, while multilayer networks can compose adjacent pairs in parallel in $\log k$ layers. Overall, the sequential group composition task offers a tractable window into the mechanics of deep learning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。