atlas

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

ReFT: Representation Finetuning for Language Models

NeurIPS 2024 · Large language models › Instruction and fine-tuning · approach: Gradient-free adaptation and merging · 10 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 Representation Finetuning (ReFT), which trains low-rank interventions on frozen LM hidden representations as a PEFT alternative, instantiated as LoReFT and DiReFT.

Problem

Parameter-efficient finetuning methods adapt LLMs by updating weights, whereas interpretability work suggests representations carry rich semantics that could be edited more efficiently and powerfully.

Approach

Define ReFT as learned interventions on hidden representations at chosen layers and positions of a frozen LM; LoReFT edits the representation within a learned low-rank linear subspace (building on distributed alignment search) and DiReFT is a cheaper LoRA-like ablation applied directly to representations.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use LoReFT instead of the strongest listed PEFT baselineLLaMA-family models · Commonsense reasoning (8-task Avg.)accuracy: 78.1 -> 80.2; 81.5 -> 83.3; 80.5 -> 81.8; 85.2 -> 86.6Helped
Remove LoReFT orthogonality constraint and difference operation (DiReFT) (ablation)LoReFT · Commonsense reasoning (8-task Avg.)accuracy: 80.2 -> 79.0; 83.3 -> 82.2; 81.8 -> 80.9; 86.6 -> 85.4Hurt
Use LoReFT instead of LoRALLaMA-family models · Arithmetic reasoning (4-task Avg.)accuracy: 46.9 -> 42.6; 51.1 -> 49.6Hurt
Remove LoReFT orthogonality constraint and difference operation (DiReFT) (ablation)LoReFT · Arithmetic reasoning (4-task Avg.)accuracy: 42.6 -> 40.6; 49.6 -> 48.0Hurt
Use LoReFT instead of LoRALlama-2 7B · Alpaca-Eval v1.0win-rate: 81.48 -> 85.60Helped
Remove LoReFT orthogonality constraint and difference operation (DiReFT) (ablation)LoReFT · Alpaca-Eval v1.0win-rate: 85.60 -> 84.85Hurt
Halve LoReFT rank (ablation)LoReFT on Llama-2 7B · Alpaca-Eval v1.0win-rate: 85.60 -> 84.12Hurt
Train LoReFT with 1K examples (ablation)LoReFT on Llama-2 7B · Alpaca-Eval v1.0win-rate: 85.60 -> 81.91Hurt
Use LoReFT instead of REDRoBERTa-base · GLUE (Avg.)accuracy: 84.3 -> 84.2Hurt
Use LoReFT instead of LoRARoBERTa-large · GLUE (Avg.)accuracy: 88.1 -> 88.2Helped
Remove LoReFT orthogonality constraint and difference operation (DiReFT) (ablation)LoReFT · GLUE (Avg.)accuracy: 84.2 -> 83.2; 88.2 -> 87.4Hurt

Surprising findings

Limitations

Experimental setup

Datasets: Commonsense170K Math10K UltraFeedback Alpaca-52K GLUE GSM8K

Benchmarks: BoolQ PIQA SIQA HellaSwag WinoGrande ARC-e ARC-c OBQA AQuA GSM8K MAWPS SVAMP

Metrics: accuracy win-rate parameter percentage

Baselines: LoRA DoRA Prefix-tuning Series Adapter Parallel Adapter BitFit RED VeRA full finetuning ChatGPT GPT-3.5 Turbo 1106

Builds on: distributed alignment search distributed interchange intervention LoRA DoRA pyvene

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