RFG扩散模型推理框架
2 steps1 pieces3 Sources
斯坦福大学团队提出**Reward-Free Guidance (RFG)**框架,旨在解决扩散大语言模型(dLLMs)在推理阶段难以通过额外训练或外部奖励模型进一步提升性能的问题。该框架无需额外训练、数据或显式奖励模型,仅利用现有检查点即可实现推理时的自我改进。
【背景】传统自回归LLM常依赖过程奖励模型(PRM)进行逐步引导,但dLLM的任意顺序生成特性导致中间状态为部分掩码文本,使得直接应用PRM极具挑战。RFG通过将奖励信号参数化为策略模型与参考模型之间的对数似然比,巧妙地绕过了这一障碍。
实验数据显示,RFG在数学推理和代码生成基准测试中表现优异。早期版本(v1)报告准确率提升最高达9.2%,而最新修订版(v2)显示其能将最先进post-trained dLLMs的性能提升最高16.1%,且增益效果可媲美甚至超越资源密集型的强化学习技术。这一进展表明,dLLMs可在不增加计算成本的前提下,通过纯推理阶段的算法优化实现显著的能力跃升。
How it got here
论文v1提交,提出RFG作为测试时扩展方法,在四个基准上实现最高9.2%的准确率提升。
arxiv.org ↗论文v2修订发布,标题调整为强调“自我改进”,实验结果显示性能提升幅度扩大至最高16.1%。
arxiv.org ↗
What is still unsettled
- RFG在不同规模dLLM上的泛化能力如何?
- 该方法是否适用于非数学/代码类的通用推理任务?
- 相比显式训练奖励模型,RFG在极端复杂逻辑链中的稳定性边界在哪里?