让语言模型学会推理多种合理后续词,提升推理多样性与准确性
BOW: Training Language Models to Reason Over Plausible Next Words
- 用强化学习训练模型生成涵盖多种可能后续词的中立描述
- 在十项推理任务上表现优于基线模型,尤其在歧义词处理上更优
- 适合需要多样性和鲁棒性推理的应用,如对话系统、文本生成
当前的下一步词预测(NWP)仅基于单一观测到的延续进行训练,尽管许多上下文存在多个合理后续词。近期基于强化学习的方法虽显式处理此矛盾,但易将已有偏好固化为自信轨迹。本文提出BOW框架,训练模型生成自包含、中立且全面的合理后续词空间描述。策略从完整上下文生成推理路径,而固定评分器仅依据该路径计算核心奖励,不依赖原始上下文。BOW-Reg引入轻量级广度正则化以防止过早坍缩。在两种模型架构上,BOW性能与原模型相当,常超越训练过的基线,在十项通用推理基准上表现优异。在测试模糊指代和词义歧义的基准中,BOW-Reg在SharedRef正确率上最高,且在双架构上均实现最低的HoWN-Simple单义坍缩率。人工评估显示BOW-Reg生成的推理路径更宽泛,直接评价表明这些路径仍具预测能力。
原文摘要 · Abstract (English)
Next-word prediction (NWP) trains language models against a single observed continuation, even though many contexts admit multiple plausible next words. Recent RL-based next-word reasoning methods make this tension explicit: they reward a model for producing a rationale that supports one context-conditioned continuation, which can turn a pre-existing preference into a confident, self-justifying trajectory. We introduce BOW, an RL framework that instead trains models to produce self-contained, neutral, and comprehensive descriptions of the plausible next-word space. The policy generates a next-word reasoning trajectory from the full context, but a frozen scorer computes the core reward from that trajectory alone, without separately receiving the context. BOW-Reg adds a lightweight breadth regularizer to this core reward to discourage premature collapse. On two model backbones, BOW remains competitive with the original models and often outperforms trained baselines across ten general reasoning benchmarks. On benchmarks testing ambiguous references and word meanings, BOW-Reg achieves the highest SharedRef correctness and the lowest HoWN-Simple single-sense collapse on both backbones. Human evaluation further shows that BOW-Reg produces broader next-word reasoning trajectories, while direct next-word-prediction evaluation shows that these trajectories remain predictive.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。