VOOZH about

URL: https://huggingface.co/CoolSpring/Qwen2-0.5B-Abyme

⇱ CoolSpring/Qwen2-0.5B-Abyme · Hugging Face


👁 Built with Axolotl


Qwen2-0.5B-Abyme

This model is a fine-tuned version of Qwen/Qwen2-0.5B on the Magpie-Align/Magpie-Qwen2-Pro-300K-Filtered dataset. It was created to explore the effects of training the smallest model in the Qwen2 series on data extracted from the largest model in the Qwen2 series (as of July 18th, 2024).

It achieves the following results on the evaluation set:

  • Loss: 0.8229

Model description

Qwen2-0.5B-Abyme is a 0.5 billion parameter language model fine-tuned on a dataset of conversation samples from the much larger 72 billion parameter Qwen2-72B model. The purpose of this experiment is to investigate whether a smaller model can effectively learn and reproduce the knowledge and capabilities of a significantly larger model through the fine-tuning process.

Intended uses & limitations

This model is intended for research purposes to study the knowledge transfer and distillation capabilities of language models. It may have practical applications in scenarios where the computational resources for running large language models are limited, and a smaller, fine-tuned model can provide comparable performance.

However, it is important to note that the model's capabilities and limitations are yet to be fully evaluated. Its performance may vary depending on the task and domain, and it may exhibit biases or limitations inherited from the original models.

Training and evaluation data

The model was fine-tuned on the Magpie-Align/Magpie-Qwen2-Pro-300K-Filtered dataset, which contains 300,000 conversation samples from the Qwen2-72B model. 5% of this dataset was held out as the evaluation set for calculating the reported loss metric.

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 2e-05
  • train_batch_size: 4
  • eval_batch_size: 4
  • seed: 42
  • gradient_accumulation_steps: 4
  • total_train_batch_size: 16
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: cosine
  • lr_scheduler_warmup_steps: 100
  • num_epochs: 1

Training results

Training Loss Epoch Step Validation Loss
0.9947 0.0004 1 0.9683
0.8385 0.2501 597 0.8338
0.7636 0.5002 1194 0.8249
0.8124 0.7502 1791 0.8229

Framework versions

  • Transformers 4.42.3
  • Pytorch 2.3.1+cu121
  • Datasets 2.19.1
  • Tokenizers 0.19.1

Open LLM Leaderboard Evaluation Results

Detailed results can be found here

Metric Value
Avg. 4.76
IFEval (0-Shot) 19.15
BBH (3-Shot) 2.28
MATH Lvl 5 (4-Shot) 1.51
GPQA (0-shot) 0.45
MuSR (0-shot) 1.48
MMLU-PRO (5-shot) 3.70
Downloads last month
5
Safetensors
Model size
0.5B params
Tensor type
BF16
·

Model tree for CoolSpring/Qwen2-0.5B-Abyme

Base model

Qwen/Qwen2-0.5B
Finetuned
(143)
this model
Merges
3 models
Quantizations
2 models

Dataset used to train CoolSpring/Qwen2-0.5B-Abyme