atlas

Research map / Large language models / Instruction and fine-tuning

Self-Play Fine-Tuning Converts Weak Language Models to Strong Language Models

ICML 2024 · Large language models › Instruction and fine-tuning · approach: Efficient LLM fine-tuning systems · 11 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes SPIN, a self-play fine-tuning method that turns an SFT model into a stronger one by training it to distinguish its own previous-iteration generations from human SFT data, without extra annotations.

Problem

Improving an LLM beyond its SFT checkpoint without acquiring additional human-annotated or AI-feedback preference data; further SFT on the same data plateaus or degrades performance.

Approach

A self-play game in which the new LLM iterate is trained with a logistic-loss objective to prefer human SFT responses over responses sampled from the previous iterate (the opponent), with the opponent updated to the new model each round; the objective has a closed-form optimum at the data distribution, and resembles DPO but uses only SFT data.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Apply SPIN self-play fine-tuning through iteration 3zephyr-7b-sft-full · HuggingFace Open LLM LeaderboardAverage: 58.14 -> 63.16Helped
Apply SPIN self-play fine-tuning through iteration 3zephyr-7b-sft-full · GSM8Kacc: 26.76 -> 38.97Helped
Apply SPIN self-play fine-tuning through iteration 3zephyr-7b-sft-full · TruthfulQAmc2: 43.73 -> 54.90Helped
Apply SPIN self-play fine-tuning through iteration 3zephyr-7b-sft-full · Winograndeacc: 74.19 -> 73.72Hurt
Apply SPIN self-play fine-tuning through iteration 3zephyr-7b-sft-full · BB-sportsaccuracy: 96.0 -> 94.4Hurt
Apply SPIN self-play fine-tuning through iteration 2zephyr-7b-sft-full · MT-Benchscore: 5.94 -> 6.78Helped
Continue SFT for one more epoch on Ultrachat200k (ablation)zephyr-7b-sft-full · HuggingFace Open LLM LeaderboardAverage: 58.14 -> 57.23Hurt
Use iterative SPIN training instead of extending training within iteration 0 (ablation)zephyr-7b-sft-full · HuggingFace Open LLM LeaderboardAverageHelped
Increase SPIN training size from 14k to 50k (ablation)SPIN iteration 0 · HuggingFace Open LLM LeaderboardAverageHelped
Fine-tune SPIN iteration 3 with DPO for two epochs on 62k preference examplesSPIN iteration 3 · HuggingFace Open LLM LeaderboardAverage: 63.16 -> 64.05Helped
Fine-tune SPIN iteration 3 with DPO for two epochs on 62k preference examplesSPIN iteration 3 · TruthfulQAmc2: 54.90 -> 60.07Helped
Use SPIN iteration 1 rather than DPO trainingzephyr-7b-sft-full · HuggingFace Open LLM LeaderboardAverage: 61.31 -> 62.12Helped

Surprising findings

Limitations

Experimental setup

Datasets: UltraChat200k UltraFeedback Binarized

Benchmarks: HuggingFace Open LLM Leaderboard (ARC, TruthfulQA, Winogrande, GSM8k, HellaSwag, MMLU) MT-Bench BIG-Bench Hard (causal judgment, formal fallacies, sports understanding) OpenBookQA

Metrics: accuracy acc_norm mc2 average score MT-Bench score

Baselines: SFT (zephyr-7b-sft-full) DPO (zephyr-7b-beta) vicuna-13b-v1.5

Builds on: DPO zephyr-7b-sft-full Alignment Handbook AlphaGo Zero Wasserstein GAN

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy