arXiv:2510.23323cs.LGcs.NE2025-10

提出新型神经网络训练方法,让脑启发算法可扩展到百层以上。

Towards Scaling Deep Neural Networks with Predictive Coding: Theory and Practice

  • 用优化理论解释预测编码如何近似使用二阶信息进行学习。
  • 证明预测编码能克服梯度消失问题,使训练更稳定高效。
  • 设计新参数化方法μPC,实现百层网络稳定训练,适合硬件协同设计研究者。

反向传播(BP)是当前深度神经网络训练的标准算法,但能耗高且不符合大脑运作机制。本文研究一种更具生物合理性的替代方法——预测编码(PC)。不同于BP,预测编码网络(PCNs)通过迭代平衡神经元活动来完成推理,再进行学习或权重更新。已有研究表明,该迭代推理过程具有加速训练等优势,但这些效果尚未稳定出现,其推理与学习动态仍不清晰,深层PCNs也难以实际训练。本文从优化理论出发,取得关键进展:首先,揭示了预测编码的学习动态本质上是一种利用二阶信息的近似信赖域方法,尽管仅采用一阶局部更新;其次,进一步证明预测编码原则上可利用任意高阶信息,使得前馈网络中其学习的有效损失曲面远比均方误差损失更平滑、对梯度消失更鲁棒;第三,基于对推理动态的分析,提出全新参数化方式μPC,首次实现无需大量调参即可稳定训练100+层网络,在简单任务上表现媲美主流方法。本研究显著深化了对预测编码网络推理与学习动态的理解,同时强调未来需关注硬件协同设计,以使预测编码在大规模应用中挑战反向传播。

原文摘要 · Abstract (English)

Backpropagation (BP) is the standard algorithm for training the deep neural networks that power modern artificial intelligence including large language models. However, BP is energy inefficient and unlikely to be implemented by the brain. This thesis studies an alternative, potentially more efficient brain-inspired algorithm called predictive coding (PC). Unlike BP, PC networks (PCNs) perform inference by iterative equilibration of neuron activities before learning or weight updates. Recent work has suggested that this iterative inference procedure provides a range of benefits over BP, such as faster training. However, these advantages have not been consistently observed, the inference and learning dynamics of PCNs are still poorly understood, and deep PCNs remain practically untrainable. Here, we make significant progress towards scaling PCNs by taking a theoretical approach grounded in optimisation theory. First, we show that the learning dynamics of PC can be understood as an approximate trust-region method using second-order information, despite explicitly using only first-order local updates. Second, going beyond this approximation, we show that PC can in principle make use of arbitrarily higher-order information, such that for feedforward networks the effective landscape on which PC learns is far more benign and robust to vanishing gradients than the (mean squared error) loss landscape. Third, motivated by a study of the inference dynamics of PCNs, we propose a new parameterisation called "$μ$PC", which for the first time allows stable training of 100+ layer networks with little tuning and competitive performance on simple tasks. Overall, this thesis significantly advances our fundamental understanding of the inference and learning dynamics of PCNs, while highlighting the need for future research to focus on hardware co-design if PC is to compete with BP at scale.

神经网络预测编码训练优化可扩展性

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