计算机科学 > 机器学习
[提交于 2025年1月6日
]
标题: 可扩展的前向-前向算法
标题: Scalable Forward-Forward Algorithm
摘要: 我们提出了一种可扩展的前向-前向(FF)算法,通过单独训练每一层来消除对反向传播的需求。 与反向传播不同, FF 避免了反向梯度,并且可以更加模块化和内存高效,这使其对于大型网络具有吸引力。 我们通过引入一种新的方式计算卷积层的损失,将 FF 扩展到现代卷积架构,如 MobileNetV3 和 ResNet18。 实验表明,我们的方法性能可与标准反向传播相媲美。 此外,当我们把网络分成块,例如 ResNet 中的残差块,并仅在每个块内应用反向传播,而不是跨块进行,我们的混合设计往往在保持类似训练速度的同时优于反向传播基线。 最后,我们在小数据集和迁移学习上进行了实验,证实了我们方法的适应性。
文献和引用工具
与本文相关的代码,数据和媒体
alphaXiv (什么是 alphaXiv?)
CatalyzeX 代码查找器 (什么是 CatalyzeX?)
DagsHub (什么是 DagsHub?)
Gotit.pub (什么是 GotitPub?)
Hugging Face (什么是 Huggingface?)
带有代码的论文 (什么是带有代码的论文?)
ScienceCast (什么是 ScienceCast?)
演示
推荐器和搜索工具
arXivLabs:与社区合作伙伴的实验项目
arXivLabs 是一个框架,允许合作伙伴直接在我们的网站上开发和分享新的 arXiv 特性。
与 arXivLabs 合作的个人和组织都接受了我们的价值观,即开放、社区、卓越和用户数据隐私。arXiv 承诺这些价值观,并且只与遵守这些价值观的合作伙伴合作。
有一个为 arXiv 社区增加价值的项目想法吗? 了解更多关于 arXivLabs 的信息.