arXiv:2412.01951cs.AIcs.CL2024-12被引 106

通过自验证提升模型性能,让模型自己‘打磨’出更优回答。

Self-Improvement in Language Models: The Sharpening Mechanism

  • 用模型自身作验证器,在后训练阶段聚焦高质量输出。
  • 基于SFT的方法在初始模型覆盖充分时达到最优效果。
  • 基于RLHF的方法可在线探索,无需依赖初始覆盖范围。

语言模型的自我改进可能通过模型评估并优化自身生成内容来实现,而无需外部反馈。尽管自我改进无法创造模型中原本不存在的信息,但为何仍能提升能力?本文提出‘锐化’视角:由于模型在判断回答质量上优于生成正确回答,因此可将模型本身作为后训练阶段的验证器,使模型集中于高质序列,从而摊销生成优质内容的昂贵推理成本。我们引入新的统计框架分析锐化机制,并建立理论极限。分析基于SFT和RLHF的两类自改进算法发现:(i) 当初始模型具备足够覆盖性时,SFT方法为极小极大最优;(ii) RLHF方法可通过在线探索超越SFT,无需依赖初始覆盖。实验验证了推理时与成本摊销下的锐化机制有效性。这些发现为自改进算法的设计与评估提供了基础理解。

原文摘要 · Abstract (English)

Recent work in language modeling has raised the possibility of self-improvement, where a language models evaluates and refines its own generations to achieve higher performance without external feedback. It is impossible for this self-improvement to create information that is not already in the model, so why should we expect that this will lead to improved capabilities? We offer a new perspective on the capabilities of self-improvement through a lens we refer to as sharpening. Motivated by the observation that language models are often better at verifying response quality than they are at generating correct responses, we formalize self-improvement as using the model itself as a verifier during post-training in order to ``sharpen'' the model to one placing large mass on high-quality sequences, thereby amortizing the expensive inference-time computation of generating good sequences. We begin by introducing a new statistical framework for sharpening in which the learner aims to sharpen a pre-trained base policy via sample access, and establish fundamental limits. Then we analyze two natural families of self-improvement algorithms based on SFT and RLHF. We find that (i) the SFT-based approach is minimax optimal whenever the initial model has sufficient coverage, but (ii) the RLHF-based approach can improve over SFT-based self-improvement by leveraging online exploration, bypassing the need for coverage. Finally, we empirically validate the sharpening mechanism via inference-time and amortization experiments. We view these findings as a starting point toward a foundational understanding that can guide the design and evaluation of self-improvement algorithms.

自改进模型优化锐化机制

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