atlas

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

What Makes Good Data for Alignment? A Comprehensive Study of Automatic Data Selection in Instruction Tuning

ICLR 2024 · Large language models › Instruction and fine-tuning · approach: Instruction data selection

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.

Studies complexity, quality, and diversity metrics for instruction-tuning data selection, proposes Evol Complexity/Quality scorers and an embedding-based diversity filter, and trains the Deita models on 6K selected samples.

Problem

Instruction tuning data is often heuristically built or manually selected, and there is no principled understanding of what makes good alignment data or how to select it automatically and efficiently.

Approach

Measure data along complexity, quality, and diversity; score complexity and quality with ChatGPT-ranked evolved variants used to train LLaMA scorers; then select samples greedily by combined score (c*q) subject to an embedding-distance diversity filter.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Select SFT data with Deita instead of random selectionLLaMA-1-13B · MT-Benchscore: 5.84 -> 6.46Helped
Increase Deita-selected SFT data from 6K to 10K examples (ablation)LLaMA-1-13B · MT-Benchscore: 6.46 -> 6.60Helped
Increase Deita-selected SFT data from 6K to 10K examples (ablation)Mistral-7B · MT-Benchscore: 7.22 -> 7.32Helped
Add DPO training with 10K preference pairs (ablation)Deita-Mistral-7B 6K · MT-Benchscore: 7.22 -> 7.55Helped
Add DPO training with 10K preference pairs (ablation)Deita-Mistral-7B 6K · Open LLM Leaderboardaverage: 64.94 -> 69.86Helped
Select data using Evol Complexity instead of other complexity metricsLLaMA-1-13B instruction tuning · X_sota and X_baseMT-BenchHelped
Select data using Evol Quality instead of other quality metricsLLaMA-1-13B instruction tuning · X_sota and X_baseMT-BenchHelped
Use response length to select higher-quality dataLLaMA-1-13B instruction tuning · X_sotaMT-BenchNo clear effect
Select data with Repr Filter rather than random selectionLLaMA-1-13B instruction tuning · X_sota and X_baseMT-BenchHelped
Use Repr Filter instead of Instag DiversityLLaMA-1-13B instruction tuning · X_sota and X_baseMT-BenchHelped
Select samples by response perplexityLLaMA-1-13B instruction tuning · X_sota and X_baseMT-BenchHurt
Use model-based sentence representations rather than E5-Large-V2 semantic representations in Repr Filter (ablation)Repr Filter · X_sotaMT-BenchHelped

Surprising findings

Limitations

Experimental setup

Datasets: WizardLM (Alpaca) WizardLM (ShareGPT) UltraChat ShareGPT Alpaca Dolly OpenAssistant (OAssist) FLAN 2022 UltraFeedback LIMA test set

Benchmarks: MT-Bench AlpacaEval Open LLM Leaderboard (ARC, HellaSwag, MMLU, TruthfulQA)

Metrics: MT-Bench score AlpacaEval win rate (%) accuracy

Baselines: Random Selection Instag Complexity Instruction Node Direct Scoring IFD LIMA Alpagasus TAGLM Vicuna-13B-v1.3 WizardLM-13B Zephyr-beta Mistral-7B-Instruct-v0.2

Builds on: Evol-Instruct WizardLM InsTag Direct Preference Optimization Zephyr recipe

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