dalle-mega / README.md
boris's picture
Update The Estimated Carbon Emission for DALL-E Mega (#9)
be8f4b3
metadata
inference: false
co2_eq_emissions:
  emissions: 450300
  source: MLCo2 Machine Learning Impact calculator
  geographical_location: East USA
  hardware_used: TTPU v3-256
tags:
  - text-to-image
license: apache-2.0
language: en
model-index:
  - name: dalle-mega
    results: []
task:
  name: Text to Image
  type: text-to-image

DALL·E Mega Model Card

This model card focuses on the DALL·E Mega model associated with the DALL·E mini space on Hugging Face, available here. The app is called “dalle-mini”, but incorporates “DALL·E Mini” and “DALL·E Mega” models. The DALL·E Mega model is the largest version of DALLE Mini. For more information specific to DALL·E Mini, see the DALL·E Mini model card.

Model Details

  • Developed by: Boris Dayma, Suraj Patil, Pedro Cuenca, Khalid Saifullah, Tanishq Abraham, Phúc Lê, Luke, Luke Melas, Ritobrata Ghosh

  • Model type: Transformer-based text-to-image generation model

  • Language(s): English

  • License: Apache 2.0

  • Model Description: This is a model that can be used to generate images based on text prompts. As the model developers wrote in the project report about DALL·E mini, “OpenAI had the first impressive model for generating images with DALL·E. DALL·E mini is an attempt at reproducing those results with an open-source model.”

  • Resources for more information:

  • Cite as:

@misc{Dayma_DALL·E_Mini_2021,
      author = {Dayma, Boris and Patil, Suraj and Cuenca, Pedro and Saifullah, Khalid and Abraham, Tanishq and Lê Khắc, Phúc and Melas, Luke and Ghosh, Ritobrata},
      doi = {10.5281/zenodo.5146400},
      month = {7},
      title = {DALL·E Mini},
      url = {https://github.com/borisdayma/dalle-mini},
      year = {2021}
}

Uses

Direct Use

The model is intended to be used to generate images based on text prompts for research and personal consumption. Intended uses include supporting creativity, creating humorous content, and providing generations for people curious about the model’s behavior. Intended uses exclude those described in the Misuse and Out-of-Scope Use section.

Downstream Use

The model could also be used for downstream use cases, including:

  • Research efforts, such as probing and better understanding the limitations and biases of generative models to further improve the state of science
  • Development of educational or creative tools
  • Generation of artwork and use in design and artistic processes.
  • Other uses that are newly discovered by users. This currently includes poetry illustration (give a poem as prompt), fan art (putting a character in various other visual universes), visual puns, fairy tale illustrations (give a fantasy situation as prompt), concept mashups (applying a texture to something completely different), style transfers (portraits in the style of), … We hope you will find your own application!

Downstream uses exclude the uses described in Misuse and Out-of-Scope Use.

Misuse, Malicious Use, and Out-of-Scope Use

The model should not be used to intentionally create or disseminate images that create hostile or alienating environments for people. This includes generating images that people would foreseeably find disturbing, distressing, or offensive; or content that propagates historical or current stereotypes.

Out-of-Scope Use

The model was not trained to be factual or true representations of people or events, and therefore using the model to generate such content is out-of-scope for the abilities of this model.

Misuse and Malicious Use

Using the model to generate content that is cruel to individuals is a misuse of this model. This includes:

  • Generating demeaning, dehumanizing, or otherwise harmful representations of people or their environments, cultures, religions, etc.
  • Intentionally promoting or propagating discriminatory content or harmful stereotypes.
  • Impersonating individuals without their consent.
  • Sexual content without consent of the people who might see it.
  • Mis- and disinformation
  • Representations of egregious violence and gore
  • Sharing of copyrighted or licensed material in violation of its terms of use.
  • Sharing content that is an alteration of copyrighted or licensed material in violation of its terms of use.

Limitations and Bias

Limitations

The model developers discuss the limitations of the model further in the DALL·E Mini technical report:

  • Faces and people in general are not generated properly.
  • Animals are usually unrealistic.
  • It is hard to predict where the model excels or falls short…Good prompt engineering will lead to the best results.
  • The model has only been trained with English descriptions and will not perform as well in other languages

Bias

CONTENT WARNING: Readers should be aware this section contains content that is disturbing, offensive, and can propagate historical and current stereotypes.

The model was trained on unfiltered data from the Internet, limited to pictures with English descriptions. Text and images from communities and cultures using other languages were not utilized. This affects all output of the model, with white and Western culture asserted as a default, and the model’s ability to generate content using non-English prompts is observably lower quality than prompts in English.

While the capabilities of image generation models are impressive, they may also reinforce or exacerbate societal biases. The extent and nature of the biases of DALL·E Mini and DALL·E Mega models have yet to be fully documented, but initial testing demonstrates that they may generate images that contain negative stereotypes against minoritized groups. Work to analyze the nature and extent of the models’ biases and limitations is ongoing.

Our current analyses demonstrate that:

  • Images generated by the model can include disturbing and harmful stereotypes across protected classes; identity characteristics; and sensitive, social, and occupational groups.
  • When the model generates images with people in them, it tends to output people who we perceive to be white, while people of color are underrepresented.
  • Images generated by the model can contain biased content that depicts power differentials between people of color and people who are white, with white people in positions of privilege.
  • The model is generally only usable for generating images based on text in English, limiting accessibility of the model for non-English speakers and potentially contributing to the biases in images generated by the model.

The technical report discusses these issues in more detail, and also highlights potential sources of bias in the model development process.

Limitations and Bias Recommendations

  • Users (both direct and downstream) should be made aware of the biases and limitations.
  • Content that is potentially problematic should be filtered out, e.g., via automated models that detect violence or pornography.
  • Further work on this model should include methods for balanced and just representations of people and cultures, for example, by curating the training dataset to be both diverse and inclusive.

Training

Training Data

For details on the DALL·E Mega training data, see the DALL·E Mega training journal.

Training Procedure

The simplified training procedure for DALL·E Mega is as follows:

  • Hardware: 1 pod TPU v3-256 = 32 nodes of TPU VM v3-8 (8 TPU per node) = 256 TPU v3
  • Optimizer: Distributed Shampoo
  • Model Partition Specificiations: 8 model parallel x 32 data parallel
  • Batch: 44 samples per model x 32 data parallel x 3 gradient accumulation steps = 4224 increasing samples per update
  • Learning rate: warmup to 0.0001 for 10,000 steps and then kept constant until plateau
  • Gradient checkpointing used on each Encoder/Decoder layer (ie, MHA + FFN)
  • Distributed Shampoo + Normformer Optimizations have proved to be effective and efficiently scaling this model.
  • It should also be noted that the learning rate and other parameters are sometimes adjusted on the fly, and batch size increased over time as well.

There is more information about the full procedure and technical material in the DALL·E Mega training journal.

Evaluation Results

For evaluation results related to DALL·E Mega, see this technical report and the DALL·E Mega training journal.

Environmental Impact

DALL·E Mega is still training. So far, as of June 28, 2022, the model developers report that DALL·E Mega has been training for about 40-45 days on a TPU v3-256. Using those numbers, we estimate the following CO2 emissions using the Machine Learning Impact calculator presented in Lacoste et al. (2019). The hardware, runtime, cloud provider, and compute region were utilized to estimate the carbon impact.

  • Hardware Type: TPU v3-256
  • Hours used: 1344 hours (56 days)
  • Cloud Provider: GCP
  • Compute Region: us-east1
  • Carbon Emitted (Power consumption x Time x Carbon produced based on location of power grid): 18013.47 kg CO2 eq.

Citation

@misc{Dayma_DALL·E_Mini_2021,
      author = {Dayma, Boris and Patil, Suraj and Cuenca, Pedro and Saifullah, Khalid and Abraham, Tanishq and Lê Khắc, Phúc and Melas, Luke and Ghosh, Ritobrata},
      doi = {10.5281/zenodo.5146400},
      month = {7},
      title = {DALL·E Mini},
      url = {https://github.com/borisdayma/dalle-mini},
      year = {2021}
}

This model card was written by: Boris Dayma, Margaret Mitchell, Ezi Ozoani, Marissa Gerchick, Irene Solaiman, Clémentine Fourrier, Sasha Luccioni, Emily Witko, Nazneen Rajani, and Julian Herrera.