中文

预测编码网络能量景观中的严格鞍点吗?

机器学习 2024-11-12 v2 人工智能 神经与进化计算 机器学习

摘要

预测编码(PC)是一种基于能量的学习算法,通过迭代推断网络活动后再更新权重来执行。最近的工作表明,PC相对于反向传播具有更少的学习步骤优势,归功于其推断程序。然而,这些优势并不总是被观察到,PC推断对学习的影响也不太被理论上理解。本文研究了PC能量景观在网络活动平衡状态下的几何结构。对于深层线性网络,我们首先表明,平衡能量仅是一个权重相关比例缩放的均方误差损失。随后,我们证明了包括原点在内的许多高度退化(非严格)鞍点在平衡能量中变得更容易逃离(严格)。我们的理论通过线性和非线性网络上的实验得到验证。基于这些结果和其他结果,我们推测平衡能量的所有鞍点都是严格的。总体而言,这项工作表明PC推断使损失景观更良好,能够抵抗梯度消失,同时也突显了将PC扩展到更深模型的根本挑战。

关键词

引用

@article{arxiv.2408.11979,
  title  = {Only Strict Saddles in the Energy Landscape of Predictive Coding Networks?},
  author = {Francesco Innocenti and El Mehdi Achour and Ryan Singh and Christopher L. Buckley},
  journal= {arXiv preprint arXiv:2408.11979},
  year   = {2024}
}

备注

35 pages, 12 figures