Unifying ICL, SFT, KL-Regularized RL Through a Bayesian Lens
Organizations: Fudan University
Abstract
Supervised fine-tuning (SFT), few-shot in-context learning (ICL), KL-regularized RLHF/RLVR, and on-policy distillation are usually treated as distinct post-training paradigms. We develop a unified Bayesian perspective in which each is an instance of a two-step template: construct a (generalized) Bayes or Gibbs posterior from a reference model and a utility signal (log-likelihood, reward, or advantage), then approximate it by a forward-KL projection onto a parametric family, either in-weights (SFT/RL) or in-context (ICL). This yields a single chain of equivalences: few-shot ICL is an amortized projection onto the Bayes posterior predictive, and reward-weighted SFT, reward-weighted ICL, and advantage-weighted SFT are forward-KL projections of reward-induced Gibbs posteriors. The framework explains why supervised warm-up is practically unavoidable for importance-weighted projections, and interprets R1/o1-style reasoning models as combining test-time Bayesian search with training-time amortization. Matched-budget experiments on Qwen3 models corroborate the picture: operators that share their learning-signal granularity produce nearly identical updates when support is good and diverge when it degrades, and reward-weighted projection performs on par with standard baselines.
Figures & tables
| Regime | Seed | ||
|---|---|---|---|
| stable | 42 | 1.000 | |
| stable | 43 | 1.000 | |
| stable | 44 | 1.000 | |
| stress | 42 | 0.107 | |
| stress | 43 | 0.219 | |
| stress | 44 | 0.358 |
| Generation contract | raw reward sum | zero-signal group frac. | parse success |
|---|---|---|---|
| original (thinking on, 1024 tokens) | |||
| fixed (thinking off, 2048 tokens) |
| Model / operator | accuracy (/256) |
|---|---|
| Qwen3-4B base | ( ) |
| warm-start (teacher bridge) | ( ) |
| GRPO (KL-regularized RL) | ( ) |
| SFT (answer-only) | ( ) |
| reward-weighted context distillation | ( ) |
RL for Language Models
Reinforcement learning for language models uses feedback-driven policy updates to improve reasoning, alignment, and interactive behavior.
See all 435 papers