data2vec:语音、视觉和语言自监督学习的通用框架

data2vec: A General Framework for Self-supervised Learning in Speech, Vision and Language

阿列克谢·巴耶夫斯基 Alexei Baevski · Meta AI · 2022-02-07 · arXiv:2202.03555 ↗ · 被引 1180

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

摘要 · Abstract

虽然自监督学习的总体思路在不同模态间是相同的,但实际的算法和目标却因针对单一模态开发而差异很大。为了更接近通用的自监督学习,我们提出了 data2vec,这是一个在语音、自然语言处理或计算机视觉中使用相同学习方法的框架。其核心思想是在自蒸馏设置下,使用标准 Transformer 架构,基于输入的掩码视图来预测完整输入数据的潜在表示。与预测模态特定的目标(如单词、视觉标记或人类语音单元)不同,这些目标本质上是局部的,而 data2vec 预测的是包含整个输入信息的上下文相关潜在表示。在语音识别、图像分类和自然语言理解等主要基准上的实验表明,该方法达到了新的最优性能或与主流方法相当的表现。

While the general idea of self-supervised learning is identical across modalities, the actual algorithms and objectives differ widely because they were developed with a single modality in mind. To get us closer to general self-supervised learning, we present data2vec, a framework that uses the same learning method for either speech, NLP or computer vision. The core idea is to predict latent representations of the full input data based on a masked view of the input in a self-distillation setup using a standard Transformer architecture. Instead of predicting modality-specific targets such as words, visual tokens or units of human speech which are local in nature, data2vec predicts contextualized latent representations that contain information from the entire input. Experiments on the major benchmarks of speech recognition, image classification, and natural language understanding demonstrate a new state of the art or competitive performance to predominant approaches.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 19)

阅读逐段中英对照全文 →