arXiv:2609.05111cs.AI2026-09

用贝叶斯视角统一解释大模型训练的多种方法

Unifying ICL, SFT, KL-Regularized RL Through a Bayesian Lens

  • 将不同训练范式统一为贝叶斯后验的前向KL投影
  • 揭示SFT、ICL、RLHF等方法在目标函数上的等价性
  • 适合研究大模型训练机制或推理优化的读者

大型语言模型采用多种训练与评估范式:监督微调(SFT)、少样本上下文学习(ICL)、KL正则化强化学习(RLHF/RLVR)、在线策略蒸馏(OPD)以及测试时基于搜索和思维链的推理。这些方法常被视为本质不同,近期实证结果(如少样本提示对强化学习微调模型的影响混合)令人困惑。本文从贝叶斯视角出发,将这些过程置于同一框架下。核心是两步模板:(i) 利用先验模型和效用信号(似然、奖励或优势)构建输出或动作的广义贝叶斯后验 q*;(ii) 通过前向KL投影近似 q*,在参数空间(如SFT/RL)或上下文空间(如ICL)中实现。第一部分将少样本ICL和SFT形式化为对贝叶斯后验预测的加权投影;第二至第四部分表明,KL正则化RLHF/RLVR、奖励加权SFT、奖励加权ICL(RW-ICL)和优势加权SFT(AWSFT)均为对由奖励或优势诱导的后验进行前向KL投影。我们区分了这些等价性成立(目标函数与一阶更新)与不成立(学习信号来源与粒度)的条件。第五部分探讨现代推理流程的启示:将RLHF/RLVR视为“后验设计 + 投影”;为何重要性加权KL投影需冷启动或监督预热;DeepSeek-R1和o1风格推理模型结合了测试时贝叶斯搜索与训练时KL折衷。

原文摘要 · Abstract (English)

Large language models are now trained and evaluated under a diverse set of paradigms: supervised fine-tuning (SFT), few-shot in-context learning (ICL), KL-regularized RLHF/RLVR, on-policy distillation (OPD), and test-time reasoning with search and chain-of-thought. These methods are often discussed as fundamentally different, and recent empirical results--such as the mixed impact of few-shot prompting on RL-tuned reasoning models--can appear puzzling. This note develops a Bayesian perspective that puts these procedures on the same footing. At the core is a two-step template: (i) construct a (generalized) Bayes or Gibbs posterior q* over outputs or actions given a context, using a prior/reference model and a utility signal (log-likelihood, reward, or advantage); and (ii) approximate q* by a forward-KL projection onto a parametric family, either in-weights (SFT/RL) or in-context (ICL). Part I formalizes few-shot ICL and SFT as amortized and-weights projections onto the Bayes posterior predictive. Parts II-IV show that KL-regularized RLHF/RLVR, reward-weighted SFT, reward-weighted ICL (RW-ICL), and advantage-weighted SFT (AWSFT) are all instances of forward-KL projection onto posteriors induced by rewards or advantages. We disentangle where these equivalences hold (objectives and first-order updates) and where they do not (source and granularity of the learning signal). Part V sketches implications for modern reasoning pipelines: RLHF/RLVR recipes as "posterior design + projection", why cold-start or supervised warm-up is practically unavoidable for importance-weighted KL projections, and DeepSeek-R1 and o1-style reasoning models as combining test-time Bayesian search with training-time KL amortization.

大模型训练贝叶斯方法强化学习推理优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。