Bloom 350m

Developed by: BigScience (website)

• 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

License: RAIL License v1.0 (link)

Release Date Estimate: Monday, 11.July.2022

Send Questions to: bigscience-contact@googlegroups.com

Cite as: BigScience, BigScience Language Open-science Open-access Multilingual (BLOOM) Language Model. International, May 2021-May 2022

Funded by:

• The French government.

• Hugging Face (website).

• Organizations of contributors. (Further breakdown of organizations forthcoming.)

Please see the BLOOM training README for full details on replicating training.

Model Architecture: Modified from Megatron-LM GPT2 (see paper, BLOOM Megatron code):

• Decoder-only architecture

• Layer normalization applied to word embeddings layer (StableEmbedding; see code, paper)

• ALiBI positional encodings (see paper), with GeLU activation functions

• 350 million parameters:

  • 24 layers, 16 attention heads

  • Hidden layers are 1024-dimensional

  • Sequence length of 2048 tokens used (see BLOOM tokenizer, tokenizer description)

Objective Function: Cross Entropy with mean reduction (see API documentation).

Compute infrastructure: Jean Zay Public Supercomputer, provided by the French government (see announcement).

• 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 (Github link)

  • DeepSpeed (Github link)

  • PyTorch (pytorch-1.11 w/ CUDA-11.5; see Github link)

  • apex (Github link)

Training

In progress.

Current training logs: Tensorboard link

• Checkpoint size:

  • Bf16 weights: 329GB

  • Full checkpoint with optimizer states: 2.3TB

• Training throughput: About 150 TFLOP per GPU per second

• Number of epochs: 1 (current target)

• Dates:

  • Started 11th March, 2022 11:42am PST

  • Estimated end: 5th July, 2022

• Estimated cost of training: Equivalent of $2-5M in cloud computing (including preliminary experiments)

• Server training location: Île-de-France, France

Tokenization

The BLOOM tokenizer (link) is a learned subword tokenizer trained using:

• A byte-level Byte Pair Encoding (BPE) algorithm

• A simple pre-tokenization rule, no normalization

• A vocabulary size of 250,680

It was trained on a subset of a preliminary version of the corpus using alpha-weighting per language.

The training supercomputer, Jean Zay (website), uses mostly nuclear energy. The heat generated by it is reused for heating campus housing.

Estimated carbon emissions: (Forthcoming upon completion of training.)

Estimated electricity usage: (Forthcoming upon completion of training.)

🎯 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 BLOOM License, 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 settings is out of scope for this model. The model is not designed for critical decisions 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 human 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

👥 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

• Users of Derivatives of the Model, as described in the License

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 for each dataset are provided in individual Data Cards.

Training data includes:

• 45 natural languages

• 12 programming languages

• In 1.5TB of pre-processed text, converted into 350B unique tokens (see the tokenizer section for more.)

Languages

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

The following table shows the further distribution of Niger-Congo and Indic languages in the training data.

The following table shows the distribution of programming languages.