Instructions to use facebook/data2vec-text-base with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use facebook/data2vec-text-base with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="facebook/data2vec-text-base")# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("facebook/data2vec-text-base") model = AutoModel.from_pretrained("facebook/data2vec-text-base") - Inference
- Notebooks
- Google Colab
- Kaggle
| 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 | |
|  | |
| 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} | |
| } | |
| ``` |