中文

前馈贡献替代反向传播

机器学习 2021-06-11 v1 神经与进化计算

摘要

深度学习在多个领域的卓越表现源于误差反向传播(BP)的使用。然而,若干研究表明,真实大脑中不可能执行 BP。此外,BP 仍是内存占用与速度方面一个重要且未解决的瓶颈。我们提出了一种简单新颖的算法——前馈贡献算法(Front-Contribution algorithm),作为 BP 的紧凑替代方案。所有权重相对于最终层权重的贡献在训练开始前计算,且所有贡献被附加到最终层权重上,即有效的最终层权重是其自身的非线性函数。我们的算法本质上折叠了网络,免除了对非最终层所有权重进行更新的必要。这种参数缩减带来了更低的内存占用与更高的训练速度。我们表明,与若干近期提出的近似 BP 的算法不同,我们的算法产生与 BP 完全相同的输出。我们的初步实验证明了所提算法的有效性。我们的工作为有效利用这些目前未被充分探索的“前馈贡献”奠定了基础,并旨在启发下一代训练算法。

关键词

引用

@article{arxiv.2106.05569,
  title  = {Front Contribution instead of Back Propagation},
  author = {Swaroop Mishra and Anjana Arunkumar},
  journal= {arXiv preprint arXiv:2106.05569},
  year   = {2021}
}

备注

NeurIPS 2020 - Beyond Backpropagation Workshop