#99 · Primary category: Foundation Models
SPIN
The official implementation of Self-Play Fine-Tuning (SPIN)
Project last updated:05/08/24
GitHub Stars
1.3K
Forks
106
Contributors
10
License
Apache-2.0
Why we included this project
SPIN takes a model that has already been instruction-tuned and makes it stronger by having it play against its own earlier versions. The model generates its own responses and learns to tell them apart from the original supervised fine-tuning examples, so each round of self-play nudges the policy forward without needing freshly labeled preference data. That makes it an interesting alternative to DPO for anyone who wants to lift a fine-tuned model's benchmark and MT-Bench scores without curating more human preference pairs. The repo includes the generation scripts, the fine-tuning loop, and the Hugging Face datasets and zephyr-7b checkpoints used at each iteration, so results are reproducible, and the two-step generate-then-finetune workflow can be adapted to a different base model.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
transformers
🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.
CLIP
CLIP (Contrastive Language-Image Pretraining), Predict the most relevant text snippet given an image
MiniCPM-V
A Pocket-Sized MLLM for Ultra-Efficient Image and Video Understanding on Your Phone
generative-models
Generative Models by Stability AI
unilm
Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities