|
--- |
|
license: apache-2.0 |
|
language: |
|
- multilingual |
|
- af |
|
- am |
|
- ar |
|
- az |
|
- be |
|
- bg |
|
- bn |
|
- ca |
|
- ceb |
|
- co |
|
- cs |
|
- cy |
|
- da |
|
- de |
|
- el |
|
- en |
|
- eo |
|
- es |
|
- et |
|
- eu |
|
- fa |
|
- fi |
|
- fil |
|
- fr |
|
- fy |
|
- ga |
|
- gd |
|
- gl |
|
- gu |
|
- ha |
|
- haw |
|
- hi |
|
- hmn |
|
- ht |
|
- hu |
|
- hy |
|
- ig |
|
- is |
|
- it |
|
- iw |
|
- ja |
|
- jv |
|
- ka |
|
- kk |
|
- km |
|
- kn |
|
- ko |
|
- ku |
|
- ky |
|
- la |
|
- lb |
|
- lo |
|
- lt |
|
- lv |
|
- mg |
|
- mi |
|
- mk |
|
- ml |
|
- mn |
|
- mr |
|
- ms |
|
- mt |
|
- my |
|
- ne |
|
- nl |
|
- no |
|
- ny |
|
- pa |
|
- pl |
|
- ps |
|
- pt |
|
- ro |
|
- ru |
|
- sd |
|
- si |
|
- sk |
|
- sl |
|
- sm |
|
- sn |
|
- so |
|
- sq |
|
- sr |
|
- st |
|
- su |
|
- sv |
|
- sw |
|
- ta |
|
- te |
|
- tg |
|
- th |
|
- tr |
|
- uk |
|
- und |
|
- ur |
|
- uz |
|
- vi |
|
- xh |
|
- yi |
|
- yo |
|
- zh |
|
- zu |
|
datasets: |
|
- mc4 |
|
- bigscience/xP3 |
|
--- |
|
|
|
Multilingual Text-to-Text Transfer Transformer Zero (MT0) |
|
Version 1. / 28 October 2022 |
|
|
|
--- |
|
|
|
# Models |
|
|
|
mT5 is pretrained on the [mC4](https://www.tensorflow.org/datasets/catalog/c4#c4multilingual) corpus, covering 101 languages: |
|
|
|
Afrikaans, Albanian, Amharic, Arabic, Armenian, Azerbaijani, Basque, Belarusian, Bengali, Bulgarian, Burmese, Catalan, Cebuano, Chichewa, Chinese, Corsican, Czech, Danish, Dutch, English, Esperanto, Estonian, Filipino, Finnish, French, Galician, Georgian, German, Greek, Gujarati, Haitian Creole, Hausa, Hawaiian, Hebrew, Hindi, Hmong, Hungarian, Icelandic, Igbo, Indonesian, Irish, Italian, Japanese, Javanese, Kannada, Kazakh, Khmer, Korean, Kurdish, Kyrgyz, Lao, Latin, Latvian, Lithuanian, Luxembourgish, Macedonian, Malagasy, Malay, Malayalam, Maltese, Maori, Marathi, Mongolian, Nepali, Norwegian, Pashto, Persian, Polish, Portuguese, Punjabi, Romanian, Russian, Samoan, Scottish Gaelic, Serbian, Shona, Sindhi, Sinhala, Slovak, Slovenian, Somali, Sotho, Spanish, Sundanese, Swahili, Swedish, Tajik, Tamil, Telugu, Thai, Turkish, Ukrainian, Urdu, Uzbek, Vietnamese, Welsh, West Frisian, Xhosa, Yiddish, Yoruba, Zulu. |
|
|
|
mt5 was then finetuned on: |
|
- [xP3](https://huggingface.co/bigscience/xP3) to obtain [mt0-small](https://huggingface.co/bigscience/mt0-small)/[mt0-base](https://huggingface.co/bigscience/mt0-base)/[mt0-large](https://huggingface.co/bigscience/mt0-large)/[mt0-xl](https://huggingface.co/bigscience/mt0-xl)/[mt0-xxl](https://huggingface.co/bigscience/mt0-xxl) |
|
- [P3](https://huggingface.co/bigscience/P3) to obtain [mt0-p3-xxl](https://huggingface.co/bigscience/mt0-p3-xxl) |
|
- [xP3mt](https://huggingface.co/bigscience/xP3mt) to obtain [mt0-mt-xxl](https://huggingface.co/bigscience/mt5-mt-xxl) |
|
|
|
## Model Flavors |
|
|
|
Multilingual model capable of following user instructions in a variety of languages. Together with our paper [TODO: LINK], we release the following models: |
|
|
|
---- |
|
- [mt0-small](https://huggingface.co/bigscience/mt0-small): 300M parameters multitask finetuned version of [mt5-small](https://huggingface.co/google/mt5-small) on [xP3](https://huggingface.co/bigscience/xP3) |
|
- [mt0-base](https://huggingface.co/bigscience/mt0-base): 580M parameters multitask finetuned version of [mt5-base](https://huggingface.co/google/mt5-base) on [xP3](https://huggingface.co/bigscience/xP3) |
|
- [mt0-large](https://huggingface.co/bigscience/mt0-large): 1.2B parameters multitask finetuned version of [mt5-large](https://huggingface.co/google/mt5-large) on [xP3](https://huggingface.co/bigscience/xP3) |
|
- [mt0-xl](https://huggingface.co/bigscience/mt0-xl): 3.7B parameters multitask finetuned version of [mt5-xl](https://huggingface.co/google/mt5-xl) on [xP3](https://huggingface.co/bigscience/xP3) |
|
- [mt0-xxl](https://huggingface.co/bigscience/mt0-xxl): 13B parameters multitask finetuned version of [mt5-xxl](https://huggingface.co/google/mt5-xxl) on [xP3](https://huggingface.co/bigscience/xP3) |
|
---- |
|
- [mt0-p3-xxl](https://huggingface.co/bigscience/mt0-p3-xxl): 13B parameters multitask finetuned version of [mt5-xxl](https://huggingface.co/google/mt5-xxl) on [P3](https://huggingface.co/bigscience/P3) |
|
- [mt0-mt-xxl](https://huggingface.co/bigscience/mt5-mt-xxl): 13B parameters multitask finetuned version of [mt5-xxl](https://huggingface.co/google/mt5-xxl) on [xP3mt](https://huggingface.co/bigscience/xP3mt) |
|
|
|
## Basics |
|
*This section provides information about the model type, version, license, funders, release date, developers, and contact information.* |
|
*It is useful for anyone who wants to reference the model.* |
|
|
|
<details> |
|
<summary>Click to expand</summary> |
|
|
|
*All collaborators are either volunteers or have an agreement with their employer. (Further breakdown of participants forthcoming.)* |
|
|
|
**Model Type:** Transformer-based Language Model |
|
|
|
**Checkpoints format:** `transformers` |
|
|
|
**Version:** 1.0.0 |
|
|
|
**Languages:** Multiple; see [training data](#training-data) |
|
|
|
**License:** Apache 2.0 |
|
|
|
**Release Date Estimate:** Friday, 28.October.2022 |
|
|
|
**Send Questions to:** niklas@huggingface.co |
|
|
|
**Funded by:** |
|
* The French government. |
|
* Hugging Face ([website](https://huggingface.co)). |
|
|
|
</details> |
|
|
|
|
|
## Technical Specifications |
|
*This section includes details about the model objective and architecture, and the compute infrastructure.* |
|
*It is useful for people interested in model development.* |
|
|
|
<details> |
|
<summary>Click to expand</summary> |
|
|
|
### Model Architecture and Objective |
|
|
|
* Same architecture as [mt5](https://arxiv.org/abs/2010.11934) |
|
|
|
* Encoder-decoder architecture |
|
|
|
**Objective Function:** Cross Entropy with mean reduction on target tokens (see [API documentation](https://pytorch.org/docs/stable/generated/torch.nn.CrossEntropyLoss.html#torch.nn.CrossEntropyLoss)). |
|
|
|
### Compute infrastructure |
|
|
|
Models were finetuned on [TPUv4](https://cloud.google.com/tpu/docs/system-architecture-tpu-vm#tpu_v4): |
|
- `mt0-small` was finetuned on TPUv4-64 |
|
- `mt0-base` was finetuned on TPUv4-64 |
|
- `mt0-large` was finetuned on TPUv4-64 |
|
- `mt0-xl` was finetuned on TPUv4-128 |
|
- `mt0-xxl` was finetuned on TPUv4-256 |
|
- `mt0-mt-xxl` was finetuned on TPUv4-256 |
|
- `mt0-p3-xxl` was finetuned on TPUv4-256 |
|
|
|
#### Software |
|
|
|
* T5X([Github link](https://github.com/google-research/t5x), [paper](https://arxiv.org/abs/2203.17189)) |
|
|
|
</details> |
|
|
|
--- |
|
|
|
# Training |
|
*This section provides information about the training data, the speed and size of training elements, and the environmental impact of training.* |
|
*It is useful for people who want to learn more about the model inputs and training footprint.* |
|
|
|
<details> |
|
<summary>Click to expand</summary> |
|
|
|
## Training Data |
|
*This section provides a high-level overview of the training data. It is relevant for anyone who wants to know the basics of what the model is learning.* |
|
|
|
It was pretrained on mC4 and then finetuned on xP3, P3 or xP3mt. |
|
|
|
### Languages |
|
|
|
// TODO @thomasw21: Copy list from mt5 |
|
|
|
## Speeds, Sizes, Times |
|
|
|
// TODO @adarob: Maybe we can push tensorboard on this repo as well |
|
|
|
- Training logs: |
|
|
|
- Checkpoint size: 51.7GB (Bf16 weights) |
|
|
|
- Number of epochs: 1 |
|
|
|
- Precision: bfloat16 |
|
|
|
|
|
## Environmental Impact |
|
|
|
The evaluation supercomputer, [Jean Zay](http://www.idris.fr/eng/jean-zay/), uses mostly nuclear energy. The heat generated by it is reused for heating campus housing. |
|
|
|
</details> |
|
|
|
--- |
|
|
|
# Uses |
|
|
|
*This section addresses questions around how the model is intended to be used, discusses the foreseeable users of the model (including those affected by the model), and describes uses that are considered out of scope or misuse of the model.* |
|
*It is useful for anyone considering using the model or who is affected by the model.* |
|
|
|
<details> |
|
<summary>Click to expand</summary> |
|
|
|
## How to use |
|
|
|
This model can be easily used and deployed using HuggingFace's ecosystem. This needs `transformers` and `accelerate` installed. The model can be downloaded as follows: |
|
|
|
```python |
|
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer |
|
|
|
checkpoint = "..." # "checkpoint_1006000" for example |
|
model_name = "bigscience/mt0-xxl" |
|
model = AutoModelForSeq2SeqLM.from_pretrained(model_name, revision=checkpoint, torch_dtype="auto", device_map="auto") |
|
tokenizer = AutoTokenizer.from_pretrained(model_name, revision=checkpoint) |
|
|
|
inputs = tokenizer.encode("Commentaire: C'est la meilleure crêpière que j'ai jamais eu. Je l'adore.\nCe commentaire est-il positif ou négatif?", return_tensors="pt") |
|
outputs = model.generate(inputs) |
|
print(tokenizer.decode(outputs[0])) |
|
``` |
|
|
|
## Intended Use |
|
|
|
This model is being created in order to enable public research on large language models (LLMs). LLMs are intended to be used for language generation or as a pretrained base model that can be further finetuned for specific tasks. Use cases below are not exhaustive. |
|
|
|
### Direct Use |
|
|
|
- Text generation |
|
|
|
- Exploring characteristics of language generated by a language model |
|
|
|
- Examples: Cloze tests, counterfactuals, generations with reframings |
|
|
|
### Downstream Use |
|
|
|
- Tasks that leverage language models include: Information Extraction, Question Answering, Summarization |
|
|
|
</details> |
|
|
|
--- |
|
|
|
# Risks and Limitations |
|
*This section identifies foreseeable harms and misunderstandings.* |
|
|
|
<details> |
|
<summary>Click to expand</summary> |
|
|
|
Model may: |
|
|
|
- Overrepresent some viewpoints and underrepresent others |
|
|
|
- Contain stereotypes |
|
|
|
- Contain [personal information](#personal-data-and-information) |
|
|
|
- Generate: |
|
|
|
- Hateful, abusive, or violent language |
|
|
|
- Discriminatory or prejudicial language |
|
|
|
- Content that may not be appropriate for all settings, including sexual content |
|
|
|
- Make errors, including producing incorrect information as if it were factual |
|
|
|
- Generate irrelevant or repetitive outputs |
|
|
|
- Induce users into attributing human traits to it, such as sentience or consciousness |
|
|
|
</details> |
|
|
|
--- |
|
|
|
# Evaluation |
|
*This section describes the evaluation protocols and provides the results.* |
|
|
|
|
|
<details> |
|
<summary>Click to expand</summary> |
|
|
|
## Results |
|
*Results are based on the [Metrics](#metrics).* |
|
|
|
**Train-time Evaluation:** |
|
|
|
// TODO @adarob: Pending if we can get access to tensorboard |
|
|
|
**Zero-shot evaluations:** |
|
|
|
// TODO @niklas |
|
|
|
|
|
</details> |
|
|
|
--- |
|
|
|
# Recommendations |
|
|
|
*This section provides information on warnings and potential mitigations.* |
|
|
|
<details> |
|
<summary>Click to expand</summary> |
|
|
|
- Indirect users should be made aware when the content they're working with is created by the LLM. |
|
|
|
- Users should be aware of [Risks and Limitations](#risks-and-limitations), and include an appropriate age disclaimer or blocking interface as necessary. |
|
|
|
- Models trained or finetuned downstream of MT0 should include an updated Model Card. |
|
|
|
- Users of the model should provide mechanisms for those affected to provide feedback, such as an email address for comments. |
|
|
|
</details> |
|
|
|
--- |
|
|
|
# Glossary and Calculations |
|
|
|
*This section defines common terms and how metrics are calculated.* |
|
<details> |
|
<summary>Click to expand</summary> |
|
|
|
- <a name="loss">**Loss:**</a> A calculation of the difference between what the model has learned and what the data shows ("groundtruth"). The lower the loss, the better. The training process aims to minimize the loss. |
|
|
|
- <a name="perplexity">**Perplexity:**</a> This is based on what the model estimates the probability of new data is. The lower the perplexity, the better. If the model is 100% correct at predicting the next token it will see, then the perplexity is 1. Mathematically this is calculated using entropy. |
|
|
|
- <a name="high-stakes">**High-stakes settings:**</a> Such as those identified as "high-risk AI systems" and "unacceptable risk AI systems" in the European Union's proposed [Artificial Intelligence (AI) Act](https://artificialintelligenceact.eu/annexes/). |
|
|
|
- <a name="critical-decisions">**Critical decisions:**</a> Such as those defined in [the United States' proposed Algorithmic Accountability Act](https://www.congress.gov/117/bills/s3572/BILLS-117s3572is.pdf). |
|
|
|
- <a name="human-rights">**Human rights:**</a> Includes those rights defined in the [Universal Declaration of Human Rights](https://www.un.org/sites/un2.un.org/files/2021/03/udhr.pdf). |
|
|
|
- <a name="personal-data-and-information">**Personal Data and Personal Information:**</a> Personal data and information is defined in multiple data protection regulations, such as "[personal data](https://gdpr-info.eu/issues/personal-data/)" in the [European Union's General Data Protection Regulation](https://gdpr-info.eu); and "personal information" in the Republic of South Africa's [Protection of Personal Information Act](https://www.gov.za/sites/default/files/gcis_document/201409/3706726-11act4of2013popi.pdf), The People's Republic of China's [Personal information protection law](http://en.npc.gov.cn.cdurl.cn/2021-12/29/c_694559.htm). |
|
|
|
- <a name="sensitive-characteristics">**Sensitive characteristics:**</a> This includes specifically protected categories in human rights (see [UHDR, Article 2](https://www.un.org/sites/un2.un.org/files/2021/03/udhr.pdf)) and personal information regulation (see GDPR, [Article 9; Protection of Personal Information Act, Chapter 1](https://www.gov.za/sites/default/files/gcis_document/201409/3706726-11act4of2013popi.pdf)) |
|
|
|
- <a name="deception">**Deception:**</a> Doing something to intentionally mislead individuals to believe something that is false, such as by creating deadbots or chatbots on social media posing as real people, or generating text documents without making consumers aware that the text is machine generated. |
|
|
|
</details> |
|
|
|
--- |
|
|
|
# More Information |
|
*This section provides links to writing on dataset creation, technical specifications, lessons learned, and initial results.* |
|
|
|
<details> |
|
<summary>Click to expand</summary> |
|
|
|
## Intermediate checkpoints |
|
|
|
For academic (or any) usage, we published the intermediate checkpoints, corresponding to the model state at each 1000 steps. There are available as branches in this repository. You can use them using `transformers`: |
|
|
|
```python |
|
from transformers import AutoModel |
|
|
|
checkpoint = "..." # "checkpoint_1006000" for example |
|
model = AutoModel.from_pretrained("bigscience/mt0-xxl", revision=checkpoint, torch_dtype="auto", device_map="auto") |
|
``` |
|
|
|
## Dataset Creation |
|
|
|
// TODO @niklas: Point to the arxiv paper |
|
|
|
## Original checkpoints |
|
|
|
The checkpoints in this repo correspond to the HuggingFace Transformers format. We'll provide T5X checkpoints as well. |
|
|
|
# Citing MT0 |
|
|
|
Please use the following bibtex entry to cite T0: |
|
```bibtex |
|
TODO @niklas |
|
``` |
|
|
