← Selected research

Inference-time alignment · Value-guided decoding

Inference-Time Language Model Alignment via Integrated Value Guidance

Findings of EMNLP 2024

IVG combines fine-grained token steering with longer-horizon candidate selection, using two complementary value functions learned from the same preference data.

Inference-time alignmentDPOPrefix valuesBeam search
01

Research question

How can smaller preference models align a frozen target LLM at inference time—without updating its parameters?

A

Local control

Adjust each next-token distribution with a dense preference signal.

B

Long horizon

Compare partial responses after a chunk of generation, not only one token.

C

Frozen target

Guide much larger target models without task-specific fine-tuning.

02

Core method

Integrated Value GuidanceTwo value functions, two decoding scales
Shared supervisionPreference dataset

One source of human preference information

Implicit value
TrainDPO policy + reference
↓
DeriveToken log-ratio
↓
Token-wise guidanceFine-grained local steering at every step
Explicit value
TrainSequence reward model
↓
DistillPrefix value scorer
↓
Chunk-level rankingLonger-horizon selection across partial responses
Integrate during decodingSample locally → expand chunks → rank prefixes → retain top beams

The target language model remains frozen throughout generation.

Training

Shared preference data

Both value functions are learned from the same comparisons but expose complementary signals.

Generation

Multi-model KV cache

Base, tuned/reference, and reward models maintain synchronized caches and beam states.

Trade-off

Accuracy vs. compute

Implicit guidance runs every token; explicit guidance scores only at chunk boundaries.

03

Key results

+7.00Mistral-7B

Length-controlled win-rate points on AlpacaEval 2.0.

+8.17Mixtral-8×7B

Length-controlled win-rate points on AlpacaEval 2.0.

7B → 70BCross-scale guidance

Small value models guide frozen targets up to 70B and MoE scale.

AlpacaEval 2.0 LC win ratebase decoding vs. IVG
Mistral-7B-Instruct-v0.2
Base
19.51
IVG
26.51
Mixtral-8×7B-Instruct-v0.1
Base
25.58
IVG
33.75
Three task families

Controlled sentiment, summarization, and instruction following.

805 prompts

AlpacaEval 2.0 evaluation with GPT-4-Turbo judging.

No target fine-tuning

Alignment is applied entirely at inference time.

04

Paper and code