arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.07697cs.LGcs.AIcs.NE

预测编码网络的无限宽度和深度极限

On the Infinite Width and Depth Limits of Predictive Coding Networks

Brain Network Dynamics Unit, University of Oxford, UK(牛津大学脑网络动力学单位);UM6P College of Computing, Rabat, Morocco(拉巴特大学计算学院)

Francesco Innocenti, El Mehdi Achour, Rafal Bogacz

AI总结:

本文研究预测编码网络在无限宽度和深度极限下的参数化与梯度行为,证明其与反向传播等价的条件。

摘要:

预测编码(PC)是标准反向传播(BP)的一种生物合理替代方案,它在更新权重之前通过最小化关于网络活动的能量函数来工作。最近的工作通过利用一些受BP启发的重新参数化,提高了深度PC网络(PCN)的训练稳定性。然而,这些方法的完全可扩展性和理论基础仍不清楚。为了解决这一空白,我们研究了PCN的无限宽度和深度极限。对于线性残差网络,我们表明PC的宽度和深度稳定的特征学习参数化集合与BP完全相同。此外,在这些参数化中的任何一种下,当模型宽度远大于深度时,具有平衡活动的PC能量收敛到二次BP损失,导致PC计算与BP相同的梯度。实验表明,只要达到活动平衡,非线性模型(包括卷积网络和transformer)也收敛到BP。总体而言,这项工作限制了与PC可扩展的参数化类型,同时展示了如何通过仅局部更新在比深度宽得多的网络(如大脑)中有效实现BP。

英文摘要:

Predictive coding (PC) is a biologically plausible alternative to standard backpropagation (BP) that minimises an energy function with respect to network activities before updating weights. Recent work has improved the training stability of deep PC networks (PCNs) by leveraging some BP-inspired reparameterisations, but the scalability and theoretical basis of these methods remain unclear. To address this gap, we study the infinite width and depth limits of PCNs. For linear networks, we derive stable and "non-lazy" parameterisations when scaling both the model width and depth, revealing that the output of standard PCNs explodes with width during training. Moreover, under stable parameterisations, we show that the gradients computed by PC at activity equilibrium converge to the BP gradients for networks that are much wider than deep ($depth/width\to0$). Experiments show high gradient alignment between PC and BP at large width for different nonlinear models, including convolutional networks and transformers. Overall, this work constrains the parameterisations that are scalable with PC, while suggesting how BP could be implemented using only local updates in much wider than deep networks like the brain.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑