atlas

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

OpenChat: Advancing Open-source Language Models with Mixed-Quality Data

ICLR 2024 · Large language models › Instruction and fine-tuning · approach: Fine-tuning for reasoning data · 27 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 OpenChat, which fine-tunes LLaMA-2-13B on mixed-quality ShareGPT data using Conditioned-RLFT (C-RLFT), treating GPT-4 vs GPT-3.5 sources as coarse rewards and class-conditioned prompts, solved via RL-free reward-weighted supervised learning.

Problem

SFT treats all training data equally despite mixed quality, while RLHF/RLFT methods need costly pairwise or ranking preference labels. The paper asks how to use mixed-quality SFT data without preference labels.

Approach

Treat data source (GPT-4 vs GPT-3.5) as a coarse reward class, condition the policy on class-specific prompt templates, and regularize toward a class-conditioned reference policy. The KL-regularized optimum reduces to reward-weighted regression, which is solved with single-stage supervised learning and no RL or preference labels.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Apply C-RLFT to mixed-quality ShareGPT instead of the Llama-2-13B base modelLlama-2-13B · AGIEvalaverage accuracy: 35.0 -> 36.4Helped
Add coarse-grained reward weighting to OpenChat (ablation)OpenChat-13B · AlpacaEvalwin rate (%): 89.1 -> 89.5Helped
Add coarse-grained reward weighting to OpenChat (ablation)OpenChat-13B · MT-benchwin rate (%): 52.8 -> 57.5Helped
Add coarse-grained reward weighting to OpenChat (ablation)OpenChat-13B · Vicuna-benchwin rate (%): 80.0 -> 85.0Helped
Add a class-conditioned policy to OpenChat (ablation)OpenChat-13B · AlpacaEvalwin rate (%): 79.1 -> 89.5Helped
Add a class-conditioned policy to OpenChat (ablation)OpenChat-13B · MT-benchwin rate (%): 38.1 -> 57.5Helped
Add a class-conditioned policy to OpenChat (ablation)OpenChat-13B · Vicuna-benchwin rate (%): 64.4 -> 85.0Helped
Use GPT-4 ShareGPT conversations only rather than all ShareGPT conversations for SFT (ablation)Only SFT · Three-benchmark averageaverage win rate (%): 52.7 -> 64.5Helped
Use GPT-3.5 ShareGPT conversations only rather than all ShareGPT conversations for SFT (ablation)Only SFT · Three-benchmark averageaverage win rate (%): 52.7 -> 42.8Hurt
Expand the SFT ShareGPT dataset from about 70k to about 125k conversations (ablation)Vicuna-v1.5-13B · Three-benchmark averageaverage win rate (%): 52.7 -> 54.4No clear effect
Use the GPT-3.5 inference prompt instead of the GPT-4 prompt (ablation)OpenChat-13B · Instruction-following evaluationperformanceHurt
Put the class-conditioned prompt only once at the beginning instead of repeating it every turn (ablation)OpenChat-13B · Instruction-following evaluationperformanceHurt

Surprising findings

Limitations

Experimental setup

Datasets: ShareGPT

Benchmarks: AlpacaEval MT-bench Vicuna-bench AGIEval

Metrics: win rate score accuracy

Baselines: vicuna-v1.5-13b wizardlm-v1.2-13b llama-2-chat-13b ultralm-13b gpt-3.5-turbo gpt-4 claude llama-2-chat-70b

Builds on: KL-regularized RL RLHF DPO Vicuna goal-conditioned supervised learning

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