自举你的潜在表示:一种新的自监督学习方法

Bootstrap your own latent: A new approach to self-supervised Learning

让-巴蒂斯特·格里尔 Jean-Bastien Grill · Google DeepMind · 2020-06-13 · arXiv:2006.07733 ↗ · 被引 8693

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

我们提出了自举你的潜在表示(BYOL),一种新的自监督图像表示学习方法。BYOL 依赖于两个神经网络,分别称为在线网络和目标网络,它们相互交互和学习。从一张图像的一个增强视图出发,我们训练在线网络预测同一图像在另一个不同增强视图下的目标网络表示。同时,我们使用在线网络的慢速移动平均来更新目标网络。虽然最先进的方法依赖于负样本对,但 BYOL 在没有它们的情况下达到了新的最先进水平。BYOL 在使用 ResNet-50 架构进行线性评估时,在 ImageNet 上达到了 74.3% 的 top-1 分类准确率,使用更大的 ResNet 时达到 79.6%。我们展示了 BYOL 在迁移学习和半监督基准测试上表现与当前最先进水平相当或更好。我们的实现和预训练模型已在 GitHub 上提供。

We introduce Bootstrap Your Own Latent (BYOL), a new approach to self-supervised image representation learning. BYOL relies on two neural networks, referred to as online and target networks, that interact and learn from each other. From an augmented view of an image, we train the online network to predict the target network representation of the same image under a different augmented view. At the same time, we update the target network with a slow-moving average of the online network. While state-of-the art methods rely on negative pairs, BYOL achieves a new state of the art without them. BYOL reaches $74.3\%$ top-1 classification accuracy on ImageNet using a linear evaluation with a ResNet-50 architecture and $79.6\%$ with a larger ResNet. We show that BYOL performs on par or better than the current state of the art on both transfer and semi-supervised benchmarks. Our implementation and pretrained models are given on GitHub.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 12)

阅读逐段中英对照全文 →