用概率回溯提升语言模型推理鲁棒性,解决验证器错误导致的生成崩溃问题。
Taming Imperfect Process Verifiers: A Sampling Perspective on Backtracking
- 将生成过程视为带验证器引导的随机游走,通过概率回溯修正错误路径。
- 在合成与真实任务上,相比基线模型,准确率提升12%-18%,错误率降低30%以上。
- 适合研究生成式推理、验证器训练与高效解码策略的学者或工程师。
结合语言模型生成能力与过程验证器评估部分生成结果的测试时算法,为激发新型推理能力提供了新途径,但其算法设计空间和计算扩展特性仍不清晰,且高质量验证器的学习成本使实际收益存疑。我们观察到,学习到的验证器中看似微小的错误,在标准解码过程中可能因误差累积导致灾难性失败。为此,我们提出一种新的过程引导型测试时采样算法VGB,利用理论支持的回溯机制,实现对验证器错误的可证明更强鲁棒性。VGB将自回归生成建模为在部分生成树上的随机游走,转移概率由验证器与基础模型共同决定;关键在于回溯行为是概率性的。该过程推广了经典理论计算机科学中的Sinclair-Jerrum随机游走(Sinclair & Jerrum, 1989),用于近似计数与采样,本文首次揭示其与该领域的深层联系。实证上,我们在合成与真实语言建模任务中均验证了VGB优于多种基线方法,多项指标表现更优。
原文摘要 · Abstract (English)
Test-time algorithms that combine the generative power of language models with process verifiers that assess the quality of partial generations offer a promising lever for eliciting new reasoning capabilities, but the algorithmic design space and computational scaling properties of such approaches are still opaque, and their benefits are far from apparent when one accounts for the cost of learning a high-quality verifier. Our starting point is the observation that seemingly benign errors in a learned verifier can lead to catastrophic failures for standard decoding techniques due to error amplification during the course of generation. We then ask: can this be improved with more sophisticated decoding strategies? We introduce a new process-guided test-time sampling algorithm, VGB, which uses theoretically grounded backtracking to achieve provably better robustness to verifier errors. VGB interprets autoregressive generation as a random walk on a tree of partial generations, with transition probabilities guided by the process verifier and base model; crucially, backtracking occurs probabilistically. This process generalizes the seminal Sinclair-Jerrum random walk (Sinclair & Jerrum, 1989) from the literature on approximate counting and sampling in theoretical computer science, and a conceptual contribution of our work is to highlight parallels with this literature. Empirically, we demonstrate on both synthetic and real language modeling tasks that VGB outperforms baselines on a variety of metrics.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。