climategpt-70b / README.md
dthulke's picture
update model card
e3c1318
metadata
language:
  - en
datasets:
  - OpenAssistant/oasst1
  - databricks/databricks-dolly-15k
base_model: meta-llama/Llama-2-70b-hf
tags:
  - climate
co2_eq_emissions:
  emissions: 40600
  training_type: pre-training
  geographical_location: Washington, USA
  hardware_used: 8x NVIDIA H100 HBM

ClimateGPT 70B

ClimateGPT is an ensemble of AI models designed to augment human decisions on the fast-moving field of climate change. ClimateGPT 70B is a 70 billion transformer decoder model that was adapted from Llama 2 to the domain of climate science using continuous pre-training on a collection of 4.2B tokens from curated climate documents. The model is further instruction fine-tuned on a dataset of instruction-completion pairs manually collected by AppTek in cooperation with climate scientists. ClimateGPT 7B outperforms Llama 2 70B Chat on our climate-specific benchmarks. The model is designed to be used together with retrieval augmentation to extend the knowledge, and increase the factuality of the model and with cascaded machine translation to increase the language coverage.

A paper describing our approach will be released soon.

Model Details

  • Trained by: AppTek
  • Powered by: Erasmus AI
  • Verified by: EQTYLab
  • Model type: decoder-only Transformer
  • Language(s) (NLP): English
  • License: TO BE ADDED
  • Continued pre-trained from: Llama 2 70B
  • Context length: 4K tokens
  • Input: Text-only data
  • Output: Model generates text only
  • Paper: The paper will be released soon.
  • Website: eci.io

Uses

  • This model is intended to be directly used as a question answering model that is specialized in the climate domain.
  • The model is aimed at providing useful feedback for decision makers, scientists and journalists involved in climate discussions.
  • The model can also be used as a starting point for interested developers for further fine-tuning.
  • The model is NOT intended to be a general-purpose chatbot (although it has chat capabilities).
  • For the full system including cascaded MT, RAG, etc., we recommend the user to go to our demo website: TO BE ADDED.
  • Despite the efforts from the development team to eliminate them, as every other chat-capable LLMs, this model may generate biased, offensive or inaccurate responses.

Downstream Use

ClimateGPT 70B is an instruction-tuned model that can be directly used for climate-specific question-answering applications. It was trained to perform well with retrieval augmentation and supports up to 5 references in context.

The model was trained using ChatML so the following format should be followed when prompting, including the <|im_start|>, <|im_end|> tags, system, user, context and assistant identifiers and [[0]], [[1]]] etc. tokens to indicate references.

"""
<|im_start|>system
{system_message}<|im_end|>
<|im_start|>user
{prompt}<|im_end|>
<|im_start|>context
[[0]] "{reference1_title}", {reference1_year}
{reference1_text}
[[1]] "{reference2_title}", {reference2_year}
{reference2_text}
[...]<|im_end|>
<|im_start|>assistant
"""

Training

  • For the Llama2 training data, we refer the user to https://huggingface.co/meta-llama/Llama-2-70b-hf.
  • For continued pre-training, 4.2B climate domain tokens (tokenized by the Llama tokenizer) are used.
  • For instruction fine-tuning, about 272K instruction-completion pairs (both in the climate domain but also general domain) are used.

Evaluation

Detailed evaluation results are presented on our model card website: eci.io/model-card

Environmental Impact

  • Hardware Type: 8x NVIDIA H100 HBM
  • Power Consumption per GPU: 775W
  • Hours used: 2,182 hrs
  • Cloud Provider: MLFoundry
  • Compute Region: Washington, USA
  • Energy Mix: 100% Hydro Power (24g CO2eq/kWh according to IPCC 2014)
  • Carbon Emitted: 40.6kg CO2eq

Citation

BibTeX: Paper will be released soon.