Efficient Self-supervised Learning with Contextualized Target Representations for Vision, Speech and Language
Authors: Alexei Baevski, Arun Babu, Wei-Ning Hsu, Michael Auli
AI-generated Key Points
- The paper proposes an approach to increase the training efficiency of data2vec 2.0, a learning objective that generalizes across several modalities.
- The proposed method does not encode masked tokens and uses a fast convolutional decoder while amortizing the effort to build teacher representations.
- The authors leverage rich contextualized target representations introduced in data2vec to enable a fast self-supervised learner.
- Experiments show that data2vec 2.0 matches the accuracy of Masked Autoencoders in 16.4x lower pre-training time on ImageNet-1K image classification.
- On Librispeech speech recognition tasks it performs as well as wav2vec 2.0 in 10.6x less time and matches a retrained RoBERTa model in half the time on GLUE natural language understanding tasks.
- Trading some speed for accuracy results in ImageNet-1K top-1 accuracy of 86.8% with a ViT-L model trained for 150 epochs using data2vec 2.0.
- Overall, this paper presents an efficient self-supervised learning approach that can be applied across multiple modalities without requiring significant computational resources or encoding masked tokens while achieving comparable or better performance than existing methods in significantly less pre-training time.
Abstract: Current self-supervised learning algorithms are often modality-specific and require large amounts of computational resources. To address these issues, we increase the training efficiency of data2vec, a learning objective that generalizes across several modalities. We do not encode masked tokens, use a fast convolutional decoder and amortize the effort to build teacher representations. data2vec 2.0 benefits from the rich contextualized target representations introduced in data2vec which enable a fast self-supervised learner. Experiments on ImageNet-1K image classification show that data2vec 2.0 matches the accuracy of Masked Autoencoders in 16.4x lower pre-training time, on Librispeech speech recognition it performs as well as wav2vec 2.0 in 10.6x less time, and on GLUE natural language understanding it matches a retrained RoBERTa model in half the time. Trading some speed for accuracy results in ImageNet-1K top-1 accuracy of 86.8\% with a ViT-L model trained for 150 epochs.
Tools for this paper
Ask questions about this paper to our AI assistant
You can also chat with multiple papers at once here.
⚠The license of the paper does not allow us to build upon its content and the AI assistant only knows about the paper metadata rather than the full article.
The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.
⚠The license of this specific paper does not allow us to build upon its content and the summarizing tools will be run using the paper metadata rather than the full article. However, it still does a good job, and you can also try our tools on papers with more open licenses.
Similar papers summarized with our AI tools
Navigate through even more similar papers through a
tree representationLook for similar papers (in beta version)
We compare the title and abstract with papers indexed in our database. For arXiv papers, we search within the same category; for DOI and other sources, we search across sources and subjects. You can rerun the search as our database grows.
Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.