|
--- |
|
language: en |
|
tags: |
|
- exbert |
|
license: mit |
|
datasets: |
|
- bookcorpus |
|
- wikipedia |
|
--- |
|
|
|
# Data2Vec-Text base model |
|
|
|
Pretrained model on English language using the *data2vec* objective. It was introduced in |
|
[this paper](https://arxiv.org/abs/2202.03555) and first released in |
|
[this repository](https://github.com/pytorch/fairseq/tree/main/examples/data2vec). This model is case-sensitive: it |
|
makes a difference between english and English. |
|
|
|
Disclaimer: The team releasing Data2Vec-Text did not write a model card for this model so this model card has been written by |
|
the Hugging Face team. |
|
|
|
## Pre-Training method |
|
|
|
![model image](https://raw.githubusercontent.com/patrickvonplaten/scientific_images/master/data2vec.png) |
|
|
|
For more information, please take a look at the [official paper](https://arxiv.org/abs/2202.03555). |
|
|
|
## Abstract |
|
|
|
*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 selfdistillation 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.* |
|
|
|
## Intended uses & limitations |
|
|
|
The model is intended to be fine-tuned on a downstream task. |
|
See the [model hub](https://huggingface.co/models?filter=data2vec-text) to look for fine-tuned versions on a task that |
|
interests you. |
|
|
|
Note that this model is primarily aimed at being fine-tuned on tasks that use the whole sentence (potentially masked) |
|
to make decisions, such as sequence classification, token classification or question answering. For tasks such as text |
|
generation you should look at model like GPT2. |
|
|
|
## Training data |
|
|
|
The RoBERTa model was pretrained on the reunion of five datasets: |
|
- [BookCorpus](https://yknzhu.wixsite.com/mbweb), a dataset consisting of 11,038 unpublished books; |
|
- [English Wikipedia](https://en.wikipedia.org/wiki/English_Wikipedia) (excluding lists, tables and headers) ; |
|
- [CC-News](https://commoncrawl.org/2016/10/news-dataset-available/), a dataset containing 63 millions English news |
|
articles crawled between September 2016 and February 2019. |
|
- [OpenWebText](https://github.com/jcpeterson/openwebtext), an opensource recreation of the WebText dataset used to |
|
train GPT-2, |
|
- [Stories](https://arxiv.org/abs/1806.02847) a dataset containing a subset of CommonCrawl data filtered to match the |
|
story-like style of Winograd schemas. |
|
|
|
Together theses datasets weight 160GB of text. |
|
|
|
### BibTeX entry and citation info |
|
|
|
```bibtex |
|
@misc{https://doi.org/10.48550/arxiv.2202.03555, |
|
doi = {10.48550/ARXIV.2202.03555}, |
|
url = {https://arxiv.org/abs/2202.03555}, |
|
author = {Baevski, Alexei and Hsu, Wei-Ning and Xu, Qiantong and Babu, Arun and Gu, Jiatao and Auli, Michael}, |
|
keywords = {Machine Learning (cs.LG), FOS: Computer and information sciences, FOS: Computer and information sciences}, |
|
title = {data2vec: A General Framework for Self-supervised Learning in Speech, Vision and Language}, |
|
publisher = {arXiv}, |
|
year = {2022}, |
|
copyright = {arXiv.org perpetual, non-exclusive license} |
|
} |
|
``` |