atlas

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

An Emulator for Fine-tuning Large Language Models using Small Language Models

ICLR 2024 · Large language models › Instruction and fine-tuning · approach: Fine-tuning generalization and delta compression · 3 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.

Introduces emulated fine-tuning (EFT), which reweights a base model's conditionals by the log-ratio of a fine-tuned and base model to decouple pre-training and fine-tuning scale, and proposes up-scaling to approximate fine-tuning large models using small fine-tuned models.

Problem

Understanding which capabilities come from pre-training versus fine-tuning and at what scale, and how to get the benefits of fine-tuning a large model without the compute cost of fine-tuning it.

Approach

Treat the fine-tuned-minus-base log-probability difference as an implicit reward from KL-constrained RL, and apply it as a importance weight on a different-sized base model's conditionals at sampling time, so pre-training and fine-tuning scales can be chosen independently without any training.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Scale pre-training while holding fine-tuning scale fixedEFT across Llama-1, Llama-2, and Falcon · Anthropic-HH and ELI5factualityHelped
Scale fine-tuning while holding pre-training scale fixedEFT across Llama-1, Llama-2, and Falcon · Anthropic-HH and ELI5helpfulnessHelped
Combine a large base model with a small fine-tuned model (up-scaling)EFT up-scaling · Anthropic-HH and ELI5factuality: nearly 70%Helped
Combine a large base model with a small fine-tuned model (up-scaling)EFT up-scaling · Anthropic-HH and ELI5helpfulnessHelped
Use speculative decoding with block size 8 instead of no speculative decoding (ablation)EFT up-scaling with Llama-2-70B-base + Llama-2-7B-chat · Anthropic-HHtokens/sec: 6.0 -> 13.8Helped
Use speculative decoding with block size 8 instead of no speculative decoding (ablation)EFT up-scaling with Llama-2-70B-base + Llama-2-7B-chat · ELI5tokens/sec: 6.1 -> 15.1Helped
Apply top-p 0.95 filtering to up-scaling weights rather than no truncation (ablation)Llama-2 up-scaling · Llama-2 up-scalingfactual errors/prompt: 0.300 -> 0.289Helped
Apply top-p 0.95 filtering to up-scaling weights rather than no truncation (ablation)Llama-2 up-scaling · Llama-2 up-scalinghelpfulness: 66.8 -> 67.0Helped
Apply top-p 0.9 or 0.8 filtering to up-scaling weights rather than no truncation (ablation)Llama-2 up-scaling · Llama-2 up-scalingfactual errors/prompt: 0.300 -> 0.352; 0.300 -> 0.348Hurt
Apply top-p 0.9 or 0.8 filtering to up-scaling weights rather than no truncation (ablation)Llama-2 up-scaling · Llama-2 up-scalinghelpfulness: 66.8 -> 67.2; 66.8 -> 67.0Helped
Use speculative block size 16 instead of block size 8 (ablation)EFT up-scaling with Llama-2-70B-base + Llama-2-7B-chat · Anthropic-HHtokens/sec: 13.8 -> 12.1Hurt
Use speculative block size 16 instead of block size 8 (ablation)EFT up-scaling with Llama-2-70B-base + Llama-2-7B-chat · ELI5tokens/sec: 15.1 -> 14.2Hurt

Surprising findings

Limitations

Experimental setup

Datasets: Anthropic Helpful-Harmless ELI5

Benchmarks: Anthropic Helpful-Harmless prompts ELI5 prompts

Metrics: GPT-4 helpfulness probability GPT-4 factual error count GPT-4 harmlessness score tokens per second

Baselines: Llama-2-chat 7B Llama-2-chat 70B Vicuna Falcon-instruct DPO

Builds on: KL-constrained RLHF DPO speculative decoding contrastive decoding

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