bloom-560m / README.md

Younes Belkada

Update date on Model Card

78c24b3 over 2 years ago

17.4 kB

	---
	license: other
	---

	# <span style="color:red"><b>WARNING:</b> This is an <b>intermediary checkpoint</b>. It is not fully trained yet. You might want to use [Bloom-1B3](https://huggingface.co/bigscience/bloom-1b3) if you want a model that has completed training.</span>

	# <p>BLOOM LM<br/> _BigScience Large Open-source Open-access Multilingual Language Model_ <br/>Model Card</p>
	![BigScience Logo](https://assets.website-files.com/6139f3cdcbbff3a68486761d/613cd8997b270da063e230c5_Tekengebied%201-p-500.png)

	Version 1.0 / 24.May.2022

	## Table of Contents
	1. [Model Details](#model-details)
	2. [Uses](#uses)
	3. [Training Data](#training-data)
	4. [Risks and Limitations](#risks-and-limitations)
	5. [Evaluation](#evaluation)
	6. [Recommendations](#recommendations)
	7. [Glossary and Calculations](#glossary-and-calculations)
	8. [Model Card Authors](#model-card-authors)

	## Model Details

	### Basics
	This section provides information for anyone who wants to know about the model.
	<details>
	<summary>Click to expand</summary> <br/>

	Developed by: [BigScience](https://bigscience.huggingface.co)
	* All collaborators are either volunteers or have an agreement with their employer. [Further breakdown of participants forthcoming.]

	Model Type: Transformer-based Language Model

	Version: 1.0.0

	Languages: Multiple; see [training data](#training-data).

	License: [RAIL License v1.0](https://docs.google.com/document/d/10NMjEKjxR7mrZ5CvugGBVaF6nPEgNxFBIbkH7z5HB-0/edit#)

	Released: [Forthcoming]

	Send questions to: bigscience-contact@googlegroups.com

	Cite as: [BigScience Workshop](https://bigscience.huggingface.co), BigScience Language Open-source Open-access Multilingual (BLOOM). International, May 2021-May 2022.

	Funded by: The French government, [Hugging Face](https://huggingface.co), and the organizations of contributors. [Further breakdown of organizations forthcoming.]

	</details>

	### Technical Specifications
	This section provides information for people who work on model development.
	<details>
	<summary>Click to expand</summary><br/>

	Please see [the BLOOM training README](https://github.com/bigscience-workshop/bigscience/tree/master/train/tr11-176B-ml#readme) for full details.

	Model Architecture: Modified from Megatron-LM GPT2 ([paper link](https://arxiv.org/abs/1909.08053)):

	1. Layer normalization applied to word embedding layer

	2. [ALiBI positional encodings](https://arxiv.org/pdf/2108.12409.pdf)

	Objective Function: [Cross Entropy with mean reduction](https://pytorch.org/docs/stable/generated/torch.nn.CrossEntropyLoss.html#torch.nn.CrossEntropyLoss)

	Number of Parameters: 350M parameters; 24 layers, 16 attention heads

	#### Infrastructure

	Compute Infrastructure: [Jean Zay](http://www.idris.fr/eng/jean-zay/jean-zay-presentation-eng.html) Public Supercomputer, provided by the French government

	Hardware: 384 A100 80GB GPUs (48 nodes)

	- Additional 32 A100 80GB GPUs (4 nodes) in reserve

	- 8 GPUs per node Using NVLink 4 inter-gpu connects, 4 OmniPath links

	- CPU: AMD

	- CPU memory: 512GB per node

	- GPU memory: 640GB per node

	- Inter-node connect: Omni-Path Architecture (OPA)

	- NCCL-communications network: a fully dedicated subnet

	- Disc IO network: shared network with other types of nodes

	Software:

	- [Megatron-DeepSpeed](https://github.com/bigscience-workshop/Megatron-DeepSpeed), BigScience fork

	- [DeepSpeed](https://github.com/microsoft/DeepSpeed)

	- [PyTorch](https://github.com/pytorch/pytorch)-1.11 w/ CUDA-11.5

	- [apex](https://github.com/NVIDIA/apex)


	#### Training


	_In progress._

	Checkpoint size:

	- fp16 weights: 1.04GB


	Training throughput: About 150 TFLOP per GPU per second

	Number of steps: 555750

	Dates:
	- Started: to determine
	- Ended: to determine


	Estimated cost of training: Unknown

	Server training location: Ile-de-France, France

	</details>


	### Environmental Impact

	<details>
	<summary>Click to expand</summary><br/>

	[More forthcoming when training has completed.]

	The training supercomputer, [Jean Zay]((http://www.idris.fr/eng/jean-zay/jean-zay-presentation-eng.html)), uses mostly nuclear energy.

	The heat generated by it is reused for heating campus housing.

	* Estimated carbon emissions: [Forthcoming]

	* Estimated electricity usage: [Forthcoming]
	</details>

	<p> </p>

	## 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 provides information for anyone considering using the model, or who is affected by the model.*


	<details>
	<summary>Click to expand</summary><br/>

	### 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 fine-tuned 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.

	### Misuse and Out-of-scope Use

	This section addresses what users ought not do with the model.

	See the [LLM LICENSE ](https://docs.google.com/document/d/10NMjEKjxR7mrZ5CvugGBVaF6nPEgNxFBIbkH7z5HB-0/edit), Attachment A, for detailed usage restrictions. The below list is non-exhaustive, but lists some easily foreseeable problematic use cases.

	#### Out-of-scope Uses

	Using the model in [high-stakes](#glossary-and-calculations) settings is out of scope for this model. The model is not designed for [critical decisions](#glossary-and-calculations) nor uses with any material consequences on an individual's livelihood or wellbeing. The model outputs content that appears factual but is not correct.

	##### Out-of-scope uses include:

	- Usage in biomedical domains, political and legal domains, or finance domains.

	- Usage for evaluating or scoring individuals, such as for employment, education, or credit.

	- Applying the model for critical automatic decisions, generating factual content, creating reliable summaries, or generating predictions that must be correct.

	#### Misuse

	Intentionally using the model for harm, violating rights, or other kinds of malicious activities is a misuse of this model. This includes:

	- Spam generation

	- Disinformation and influence operations

	- Disparagement and defamation

	- Harassment and abuse

	- Deception

	- Unconsented impersonation and imitation

	- Unconsented surveillance


	- Generating content without attribution to the model, as specified in the [RAIL License, Use Restrictions](https://docs.google.com/document/d/10NMjEKjxR7mrZ5CvugGBVaF6nPEgNxFBIbkH7z5HB-0/edit#heading=h.3blioxkgzsje).

	### Intended Users

	#### Direct Users

	- General Public

	- Researchers

	- Students

	- Educators

	- Engineers/developers

	- Non-commercial entities

	- Community advocates, including human and civil rights groups

	#### Indirect Users

	- Users of derivatives created by Direct Users, such as those using software with an [intended use](#intended-use).

	- Users of [Derivatives of the Model, as described in the License](https://docs.google.com/document/d/117RhytMYC9HS-1NmWHEn9XBK7vJ5kdv9OcG6AV69Vec/edit#bookmark=id.pvl8781qfes3).

	#### Others Affected (Parties prenantes)

	- People and groups referred to by the LLM

	- People and groups exposed to outputs of, or decisions based on, the LLM

	- People and groups whose original work is included in the LLM
	</details>
	<p> </p>

	## 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.



	<details>
	<summary>Click to expand</summary><br/>

	Details for each dataset are provided in individual [Data Cards](https://huggingface.co/spaces/bigscience/BigScienceCorpus).

	Training data includes:

	- 45 natural languages.

	- 12 programming languages.

	- In 1.5TB of pre-processed text, converted into 350B unique tokens.

	See the [Model README, Datasets for more](https://github.com/bigscience-workshop/bigscience/tree/master/train/tr11-176B-ml#datasets).

	#### Languages
	The pie chart shows the distribution of languages in training data.

	![pie chart showing the distribution of languages in training data](https://github.com/bigscience-workshop/model_card/blob/main/assets/data/pie_chart.svg?raw=true)




	The following table shows the further distribution of Niger-Congo and Indic languages in the training data.
	<details>
	<summary>Click to expand</summary><br/>

	\| Niger Congo \| Percentage \| \| Indic \| Percentage \|
	\|----------------\|------------ \|------ \|-----------\|------------\|
	\| Chi Tumbuka \| 0.00002 \| \| Assamese \| 0.01 \|
	\| Kikuyu \| 0.00004 \| \| Odia \| 0.04 \|
	\| Bambara \| 0.00004 \| \| Gujarati \| 0.04 \|
	\| Akan \| 0.00007 \| \| Marathi \| 0.05 \|
	\| Xitsonga \| 0.00007 \| \| Punjabi \| 0.05 \|
	\| Sesotho \| 0.00007 \| \| Kannada \| 0.06 \|
	\| Chi Chewa \| 0.0001 \| \| Nepali \| 0.07 \|
	\| Setswana \| 0.0002 \| \| Telugu \| 0.09 \|
	\| Northern Sotho \| 0.0002 \| \| Malayalam \| 0.10 \|
	\| Fon \| 0.0002 \| \| Urdu \| 0.10 \|
	\| Kirundi \| 0.0003 \| \| Tamil \| 0.20 \|
	\| Wolof \| 0.0004 \| \| Bengali \| 0.50 \|
	\| Kuganda \| 0.0004 \| \| Hindi \| 0.70 \|
	\| Chi Shona \| 0.001 \|
	\| Isi Zulu \| 0.001 \|
	\| Igbo \| 0.001 \|
	\| Xhosa \| 0.001 \|
	\| Kinyarwanda \| 0.003 \|
	\| Yoruba \| 0.006 \|
	\| Swahili \| 0.02 \|
	</details>

	The following table shows the distribution of programming languages.
	<details>
	<summary>Click to expand</summary><br/>

	\| Extension \| Language \| Number of files \|
	\|----------------\|------------\|-----------------\|
	\| java \| Java \| 5,407,724 \|
	\| php \| PHP \| 4,942,186 \|
	\| cpp \| C++ \| 2,503,930 \|
	\| py \| Python \| 2,435,072 \|
	\| js \| JavaScript \| 1,905,518 \|
	\| cs \| C# \| 1,577,347 \|
	\| rb \| Ruby \| 6,78,413 \|
	\| cc \| C++ \| 443,054 \|
	\| hpp \| C++ \| 391,048 \|
	\| lua \| Lua \| 352,317 \|
	\| go \| GO \| 227,763 \|
	\| ts \| TypeScript \| 195,254 \|
	\| C \| C \| 134,537 \|
	\| scala \| Scala \| 92,052 \|
	\| hh \| C++ \| 67,161 \|
	\| H \| C++ \| 55,899 \|
	\| tsx \| TypeScript \| 33,107 \|
	\| rs \| Rust \| 29,693 \|
	\| phpt \| PHP \| 9,702 \|
	\| c++ \| C++ \| 1,342 \|
	\| h++ \| C++ \| 791 \|
	\| php3 \| PHP \| 540 \|
	\| phps \| PHP \| 270 \|
	\| php5 \| PHP \| 166 \|
	\| php4 \| PHP \| 29 \|

	</details>
	</details>
	<p> </p>

	## Risks and Limitations
	This section identifies foreseeable harms and misunderstandings.



	<details>
	<summary>Click to expand</summary><br/>

	Model may:

	- Overrepresent some viewpoints and underrepresent others

	- Contain stereotypes

	- Contain personal 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.
	</details>
	<p> </p>

	## Evaluation
	<details>
	<summary>Click to expand</summary><br/>

	### Metrics
	This section describes the different ways performance is calculated, and why.

	[More Forthcoming]

	Includes:

	\| Metric \| Why chosen \|
	\|--------------------\|--------------------------------------------------------------------\|
	\| F1 \| Standard for benchmarking \|
	\| Accuracy \| Standard for benchmarking \|
	\| Perplexity \| Standard metric for quantifying model improvements during training \|
	\| Cross Entropy Loss \| Standard objective for language models \|

	And multiple different metrics for specific tasks.

	### Factors
	This section lists some different aspects of what BLOOM models. Its focus is on those aspects that are likely to give rise to high variance in model behavior.

	- Language, such as English or Yoruba
	- Domain, such as newswire or stories
	- Demographic characteristics, such as gender or nationality

	### Results
	Results are based on the [Factors](#factors) and [Metrics](#metrics).

	Train-time evaluation:

	As of 19.May.2022, 18:00:

	- Training Loss: 2.04

	- Validation Loss: 2.21

	- Perplexity: 9.15

	[More evaluation types forthcoming at the end of model training.]
	</details>

	<BR/>

	## Recommendations

	This section provides information on warnings and potential mitigations.



	<details>
	<summary>Click to expand</summary><br/>

	- 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 pre-trained with the LLM 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>
	<p> </p>

	## Glossary and Calculations

	This section defines common terms and how metrics are calculated.



	<details>
	<summary>Click to expand</summary><br/>

	- Loss: 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.


	- Perplexity: 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.

	- High-stakes settings: 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/).

	- Critical decisions: Such as those defined in [the United States' proposed Algorithmic Accountability Act](https://www.congress.gov/117/bills/s3572/BILLS-117s3572is.pdf).

	- Human Rights: Includes those rights defined in the [Universal Declaration of Human Rights](https://www.un.org/sites/un2.un.org/files/2021/03/udhr.pdf).

	- Personal Data and Information: 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).

	- Sensitive Characteristics: 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))

	- Deception: 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>
	<p> </p>

	## Model Card Authors
	Ordered roughly chronologically and by amount of time spent.

	Margaret Mitchell, Giada Pistilli, Yacine Jernite, Ezinwanne Ozoani, Marissa Gerchick, Nazneen Rajani, Sasha Luccioni, Irene Solaiman, Maraim Masoud, Somaieh Nikpoor, Carlos Muñoz Ferrandis, Stas Bekman, Danish Contractor, David Lansky, Angelina McMillan-Major, Tristan Thrush, Suzana Ilić, Gérard Dupont, Shayne Longpre, Manan Dey, Stella Biderman, Douwe Kiela, Emi Baylor, Teven Le Scao, Aaron Gokaslan, Julien Launay