arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2309.11955cs.CVcs.LGeess.IV

面向自监督学习的 Forward-Forward 算法研究

A Study of Forward-Forward Algorithm for Self-Supervised Learning

University of Würzburg(维尔茨堡大学)

Jonas Brenig, Radu Timofte

AI总结:

本文首次在四种数据集和三种自监督任务上比较 forward-forward 与反向传播,发现二者训练性能相近,但前者迁移性能显著较差。

摘要:

自监督表征学习在过去几年中取得了显著进展,近期一些方法已经能够在没有标签的情况下学习有用的图像表征。这些方法采用事实上的标准方法——反向传播进行训练。最近,Geoffrey Hinton 提出了 forward-forward 算法作为一种替代训练方法。该算法利用两次前向传播,并为每一层设置独立的损失函数,从而在不使用反向传播的情况下训练网络。在本研究中,我们首次研究了 forward-forward 与反向传播在自监督表征学习中的性能表现,并对所学表征空间给出了深入分析。我们的基准测试使用了四个标准数据集,即 MNIST、F-MNIST、SVHN 和 CIFAR-10,以及三种常用的自监督表征学习技术,即旋转、翻转和拼图。我们的主要发现是:尽管 forward-forward 算法在(自)监督训练期间的表现与反向传播相当,但在所有研究设置中,其迁移性能都明显落后。这可能由多种因素共同造成,包括为每一层设置损失函数,以及在 forward-forward 范式中实现监督训练的方式。与反向传播相比,forward-forward 算法更关注边界,并丢弃了部分对决策而言不必要的信息,而这会损害表征学习目标。要使面向自监督学习的 forward-forward 策略稳定下来,并使其能够在 Geoffrey Hinton 所展示的数据集和配置之外发挥作用,还需要进一步调查和研究。

英文摘要:

Self-supervised representation learning has seen remarkable progress in the last few years, with some of the recent methods being able to learn useful image representations without labels. These methods are trained using backpropagation, the de facto standard. Recently, Geoffrey Hinton proposed the forward-forward algorithm as an alternative training method. It utilizes two forward passes and a separate loss function for each layer to train the network without backpropagation. In this study, for the first time, we study the performance of forward-forward vs. backpropagation for self-supervised representation learning and provide insights into the learned representation spaces. Our benchmark employs four standard datasets, namely MNIST, F-MNIST, SVHN and CIFAR-10, and three commonly used self-supervised representation learning techniques, namely rotation, flip and jigsaw. Our main finding is that while the forward-forward algorithm performs comparably to backpropagation during (self-)supervised training, the transfer performance is significantly lagging behind in all the studied settings. This may be caused by a combination of factors, including having a loss function for each layer and the way the supervised training is realized in the forward-forward paradigm. In comparison to backpropagation, the forward-forward algorithm focuses more on the boundaries and drops part of the information unnecessary for making decisions which harms the representation learning goal. Further investigation and research are necessary to stabilize the forward-forward strategy for self-supervised learning, to work beyond the datasets and configurations demonstrated by Geoffrey Hinton.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑