library_name: transformers
language:
- en
base_model:
- EleutherAI/gpt-neo-1.3B
license: apache-2.0
datasets:
- legacy-datasets/wikipedia
metrics:
- perplexity
- accuracy
new_version: Kimargin/GPT-NEO-1.3B-wiki
Model Card for GPT-NEO-1.3B-wiki
Model Details
Model Description
This model is based on EleutherAI/gpt-neo-1.3B and has been fine-tuned on the Wikipedia dataset. It is designed for text generation tasks such as summarization, question answering, and text completion in English. The model is fine-tuned to improve the fluency and factual accuracy of the generated content.
- Developed by: Kimargin
- Model type: Fine-tuned model
- Language(s): English
- License: Apache 2.0
- Finetuned from model: EleutherAI/gpt-neo-1.3B
Model Sources
- Repository: Kimargin/GPT-NEO-1.3B-wiki
Uses
Direct Use
This model can be used for tasks like text generation, summarization, and question-answering. It is useful for generating coherent and factual text based on English-language prompts.
Downstream Use
The model can be fine-tuned further for domain-specific applications such as legal or medical text generation, creating specialized question-answering systems, or generating structured content from prompts.
Out-of-Scope Use
The model should not be used in critical applications (e.g., legal, medical, or financial advice) as it may generate biased, inaccurate, or misleading information. It is also not suited for real-time decision-making.
Bias, Risks, and Limitations
Since the model was trained on Wikipedia data, it may inherit biases present in the dataset. Users should be cautious when using the model for generating sensitive or potentially biased content. The model may produce inaccurate or misleading text if given ambiguous or misleading prompts.
Recommendations
Users should verify the outputs of the model, especially in critical use cases, and should not rely solely on the model for factual accuracy without human verification.
How to Get Started with the Model
To use the model, you can load it as follows:
tokenizer = AutoTokenizer.from_pretrained("Kimargin/GPT-NEO-1.3B-wiki") model = AutoModelForCausalLM.from_pretrained("Kimargin/GPT-NEO-1.3B-wiki")
input_text = "What happened during World War II?" inputs = tokenizer(input_text, return_tensors="pt") outputs = model.generate(inputs["input_ids"], max_length=100) print(tokenizer.decode(outputs[0], skip_special_tokens=True))
Training Details
Training Data
The model was fine-tuned on a subset of the Wikipedia dataset, which contains a broad range of general knowledge topics. This dataset was chosen to improve the model's capability to generate accurate, general-domain knowledge.
Training Procedure
The model was fine-tuned using mixed precision (float16) on multiple GPUs for three epochs. The training was done to minimize perplexity and improve fluency in generated text.
Training Hyperparameters
- Learning rate: 5e-5
- Batch size: 16
- Epochs: 3
- Precision: float16 (mixed precision)
Evaluation
Testing Data
The model was evaluated using a validation subset of the Wikipedia dataset to measure its performance on general text generation tasks.
Metrics
- Perplexity: The model achieved a perplexity of 25.3 on the validation set.
- Accuracy: The accuracy of the model in generating factual answers was evaluated qualitatively.
Results
The model demonstrates good performance in generating coherent and contextually relevant text, but it may still struggle with niche or specialized topics that are underrepresented in its training data.
Environmental Impact
Training large models like GPT-Neo has a significant carbon footprint due to the computational resources required. The estimated environmental impact of fine-tuning this model is as follows:
- Hardware Type: NVIDIA A100 GPUs
- Hours used: 20 hours
- Cloud Provider: Google Cloud
- Compute Region: US-Central
- Carbon Emitted: ~50 kg CO2 (estimated using the ML Impact calculator)
Technical Specifications
Model Architecture and Objective
The model is a causal language model with 1.3 billion parameters, based on the GPT-Neo architecture. It generates text by predicting the next word in a sequence, making it suitable for text completion and generation tasks.
Compute Infrastructure
The model was trained on NVIDIA A100 GPUs using Google Cloud infrastructure.
Citation
If you use this model, please cite the original GPT-Neo model as follows:
@article{gpt-neo,
author = {EleutherAI},
title = {GPT-Neo: Large Scale Autoregressive Language Model},
year = {2021},
url = {https://github.com/EleutherAI/gpt-neo}
}