揭示预测编码比反向传播更高效的原因
Understanding Sample Efficiency in Predictive Coding

- 用目标对齐度量化学习效率,分析深度线性网络
- 预测编码在深窄网络中效率显著优于反向传播
- 为神经网络参数设置提供理论指导,适合模型优化研究者
预测编码(PC)是大脑学习的重要理论。近期研究表明,PC在小规模实验中比反向传播(BP)更具样本效率,但其原理仍不清晰。本文提出“目标对齐度”作为衡量学习效率的指标,反映网络输出变化与预测误差的一致性。通过推导深度线性网络中的解析表达式并实证验证,发现PC在深层、狭窄及预训练网络中表现更优。进一步推导出保证最优对齐的精确条件,并在全训练轨迹上验证其有效性,即使部分假设被违反,优势依然存在。该研究为此前观察到的PC更高效率提供了机制解释,可指导其参数配置以实现更高效学习。
原文摘要 · Abstract (English)
Predictive Coding (PC) is an influential account of cortical learning. Much of recent work has focused on comparing PC to Backpropagation (BP) to find whether PC offers any advantages. Small scale experiments show that PC enables learning that is more sample efficient and effective in many contexts, though a thorough theoretical understanding of the phenomena remains elusive. To address this, we quantify the efficiency of learning in BP and PC through a metric called ``target alignment'', which measures how closely the change in the output of the network is aligned to the output prediction error. We then derive and empirically validate analytical expressions for target alignment in Deep Linear Networks. We show that learning in PC is more efficient than BP, which is especially pronounced in deep, narrow and pre-trained networks. We also derive exact conditions for guaranteed optimal target alignment in PC and validate our findings through experiments. We study full training trajectories of linear and non-linear models, and find the predicted benefits of PC persist in practice even when some assumptions are violated. Overall, this work provides a mechanistic understanding of the higher learning efficiency observed for PC over BP in previous works, and can guide how PC should be parametrised to learn most effectively.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。