xlmroberta.rst 6.5 KB
Newer Older
Sylvain Gugger's avatar
Sylvain Gugger committed
1
2
3
4
5
6
7
8
9
10
11
12
.. 
    Copyright 2020 The HuggingFace Team. All rights reserved.

    Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
    the License. You may obtain a copy of the License at

        http://www.apache.org/licenses/LICENSE-2.0

    Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
    an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
    specific language governing permissions and limitations under the License.

Lysandre's avatar
Lysandre committed
13
XLM-RoBERTa
Sylvain Gugger's avatar
Sylvain Gugger committed
14
-----------------------------------------------------------------------------------------------------------------------
Lysandre's avatar
Lysandre committed
15

Sylvain Gugger's avatar
Sylvain Gugger committed
16
Overview
Sylvain Gugger's avatar
Sylvain Gugger committed
17
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Sylvain Gugger's avatar
Sylvain Gugger committed
18

Sylvain Gugger's avatar
Sylvain Gugger committed
19
20
21
22
23
The XLM-RoBERTa model was proposed in `Unsupervised Cross-lingual Representation Learning at Scale
<https://arxiv.org/abs/1911.02116>`__ by Alexis Conneau, Kartikay Khandelwal, Naman Goyal, Vishrav Chaudhary, Guillaume
Wenzek, Francisco Guzm谩n, Edouard Grave, Myle Ott, Luke Zettlemoyer and Veselin Stoyanov. It is based on Facebook's
RoBERTa model released in 2019. It is a large multi-lingual language model, trained on 2.5TB of filtered CommonCrawl
data.
Lysandre's avatar
Lysandre committed
24

Lysandre's avatar
Lysandre committed
25
The abstract from the paper is the following:
Lysandre's avatar
Lysandre committed
26

Sylvain Gugger's avatar
Sylvain Gugger committed
27
28
*This paper shows that pretraining multilingual language models at scale leads to significant performance gains for a
wide range of cross-lingual transfer tasks. We train a Transformer-based masked language model on one hundred
Lysandre's avatar
Lysandre committed
29
languages, using more than two terabytes of filtered CommonCrawl data. Our model, dubbed XLM-R, significantly
Sylvain Gugger's avatar
Sylvain Gugger committed
30
31
32
33
34
35
36
37
outperforms multilingual BERT (mBERT) on a variety of cross-lingual benchmarks, including +13.8% average accuracy on
XNLI, +12.3% average F1 score on MLQA, and +2.1% average F1 score on NER. XLM-R performs particularly well on
low-resource languages, improving 11.8% in XNLI accuracy for Swahili and 9.2% for Urdu over the previous XLM model. We
also present a detailed empirical evaluation of the key factors that are required to achieve these gains, including the
trade-offs between (1) positive transfer and capacity dilution and (2) the performance of high and low resource
languages at scale. Finally, we show, for the first time, the possibility of multilingual modeling without sacrificing
per-language performance; XLM-Ris very competitive with strong monolingual models on the GLUE and XNLI benchmarks. We
will make XLM-R code, data, and models publicly available.*
Lysandre's avatar
Lysandre committed
38

Lysandre's avatar
Lysandre committed
39
Tips:
Lysandre's avatar
Lysandre committed
40

Sylvain Gugger's avatar
Sylvain Gugger committed
41
42
- XLM-RoBERTa is a multilingual model trained on 100 different languages. Unlike some XLM multilingual models, it does
  not require :obj:`lang` tensors to understand which language is used, and should be able to determine the correct
Lysandre's avatar
Lysandre committed
43
  language from the input ids.
Sylvain Gugger's avatar
Sylvain Gugger committed
44
45
- This implementation is the same as RoBERTa. Refer to the :doc:`documentation of RoBERTa <roberta>` for usage examples
  as well as the information relative to the inputs and outputs.
Lysandre's avatar
Lysandre committed
46

Sylvain Gugger's avatar
Sylvain Gugger committed
47
The original code can be found `here <https://github.com/pytorch/fairseq/tree/master/examples/xlmr>`__.
48
49


Lysandre's avatar
Lysandre committed
50
XLMRobertaConfig
Sylvain Gugger's avatar
Sylvain Gugger committed
51
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
52
53
54
55
56
57

.. autoclass:: transformers.XLMRobertaConfig
    :members:


XLMRobertaTokenizer
Sylvain Gugger's avatar
Sylvain Gugger committed
58
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
59
60

.. autoclass:: transformers.XLMRobertaTokenizer
Lysandre Debut's avatar
Lysandre Debut committed
61
62
    :members: build_inputs_with_special_tokens, get_special_tokens_mask,
        create_token_type_ids_from_sequences, save_vocabulary
Lysandre's avatar
Lysandre committed
63
64
65


XLMRobertaModel
Sylvain Gugger's avatar
Sylvain Gugger committed
66
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
67
68

.. autoclass:: transformers.XLMRobertaModel
Sylvain Gugger's avatar
Sylvain Gugger committed
69
    :members: forward
Lysandre's avatar
Lysandre committed
70
71


72
XLMRobertaForCausalLM
Sylvain Gugger's avatar
Sylvain Gugger committed
73
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
74
75

.. autoclass:: transformers.XLMRobertaForCausalLM
Sylvain Gugger's avatar
Sylvain Gugger committed
76
    :members: forward
77
78


Lysandre's avatar
Lysandre committed
79
XLMRobertaForMaskedLM
Sylvain Gugger's avatar
Sylvain Gugger committed
80
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
81
82

.. autoclass:: transformers.XLMRobertaForMaskedLM
Sylvain Gugger's avatar
Sylvain Gugger committed
83
    :members: forward
Lysandre's avatar
Lysandre committed
84
85
86


XLMRobertaForSequenceClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
87
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
88
89

.. autoclass:: transformers.XLMRobertaForSequenceClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
90
    :members: forward
Lysandre's avatar
Lysandre committed
91
92
93


XLMRobertaForMultipleChoice
Sylvain Gugger's avatar
Sylvain Gugger committed
94
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
95
96

.. autoclass:: transformers.XLMRobertaForMultipleChoice
Sylvain Gugger's avatar
Sylvain Gugger committed
97
    :members: forward
Lysandre's avatar
Lysandre committed
98
99
100


XLMRobertaForTokenClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
101
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
102
103

.. autoclass:: transformers.XLMRobertaForTokenClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
104
    :members: forward
Lysandre's avatar
Lysandre committed
105

Lysandre's avatar
Lysandre committed
106

107
XLMRobertaForQuestionAnswering
Sylvain Gugger's avatar
Sylvain Gugger committed
108
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
109
110

.. autoclass:: transformers.XLMRobertaForQuestionAnswering
Sylvain Gugger's avatar
Sylvain Gugger committed
111
    :members: forward
112
113


Lysandre's avatar
Lysandre committed
114
TFXLMRobertaModel
Sylvain Gugger's avatar
Sylvain Gugger committed
115
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
116
117

.. autoclass:: transformers.TFXLMRobertaModel
Sylvain Gugger's avatar
Sylvain Gugger committed
118
    :members: call
Lysandre's avatar
Lysandre committed
119
120
121


TFXLMRobertaForMaskedLM
Sylvain Gugger's avatar
Sylvain Gugger committed
122
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
123
124

.. autoclass:: transformers.TFXLMRobertaForMaskedLM
Sylvain Gugger's avatar
Sylvain Gugger committed
125
    :members: call
Lysandre's avatar
Lysandre committed
126
127
128


TFXLMRobertaForSequenceClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
129
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
130
131

.. autoclass:: transformers.TFXLMRobertaForSequenceClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
132
    :members: call
Lysandre's avatar
Lysandre committed
133
134


Sylvain Gugger's avatar
Sylvain Gugger committed
135
TFXLMRobertaForMultipleChoice
Sylvain Gugger's avatar
Sylvain Gugger committed
136
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Sylvain Gugger's avatar
Sylvain Gugger committed
137
138

.. autoclass:: transformers.TFXLMRobertaForMultipleChoice
Sylvain Gugger's avatar
Sylvain Gugger committed
139
    :members: call
Sylvain Gugger's avatar
Sylvain Gugger committed
140
141


Lysandre's avatar
Lysandre committed
142
TFXLMRobertaForTokenClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
143
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
144
145

.. autoclass:: transformers.TFXLMRobertaForTokenClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
146
    :members: call
Sylvain Gugger's avatar
Sylvain Gugger committed
147
148
149


TFXLMRobertaForQuestionAnswering
Sylvain Gugger's avatar
Sylvain Gugger committed
150
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Sylvain Gugger's avatar
Sylvain Gugger committed
151
152

.. autoclass:: transformers.TFXLMRobertaForQuestionAnswering
Sylvain Gugger's avatar
Sylvain Gugger committed
153
    :members: call