arXiv:2601.08763cs.LGcs.CL2026-01ACL被引 9

让大模型生成更独特的解法,提升复杂问题求解的多样性与效果。

Rewarding the Rare: Uniqueness-Aware RL for Creative Problem Solving in LLMs

  • 用大模型判断解法的高层次策略,按相似性聚类。
  • 小众正确解法获得更高奖励,显著提升 pass@k 和 AUC@K。
  • 适合需要多样解法的推理任务,如数学、物理和医学题

强化学习已成为大语言模型后训练的核心范式,尤其在复杂推理任务中。然而,常出现探索坍缩:策略过早集中于少数主导推理模式,虽提升 pass@1,却限制了整体解法多样性及 pass@k 的提升。我们指出,根源在于对局部词元行为的正则化,而非对解法集合的多样性建模。为此,提出独特性感知强化学习(Uniqueness-Aware RL),在采样层面显式奖励罕见但正确的高层策略。该方法利用大模型裁判对同一问题的多个解法进行高层策略聚类(忽略表面差异),并反比于聚类规模重新加权策略优势。结果是,新颖且正确的解法获得更高奖励。在数学、物理和医学推理基准上,本方法在大规模采样下持续提升 pass@k 与 pass@k 曲线下面积(AUC@K),同时不牺牲 pass@1,且保持强探索能力,揭示出更多样化的解法路径。

原文摘要 · Abstract (English)

Reinforcement learning (RL) has become a central paradigm for post-training large language models (LLMs), particularly for complex reasoning tasks, yet it often suffers from exploration collapse: policies prematurely concentrate on a small set of dominant reasoning patterns, improving pass@1 while limiting rollout-level diversity and gains in pass@k. We argue that this failure stems from regularizing local token behavior rather than diversity over sets of solutions. To address this, we propose Uniqueness-Aware Reinforcement Learning, a rollout-level objective that explicitly rewards correct solutions that exhibit rare high-level strategies. Our method uses an LLM-based judge to cluster rollouts for the same problem according to their high-level solution strategies, ignoring superficial variations, and reweights policy advantages inversely with cluster size. As a result, correct but novel strategies receive higher rewards than redundant ones. Across mathematics, physics, and medical reasoning benchmarks, our approach consistently improves pass@$k$ across large sampling budgets and increases the area under the pass@$k$ curve (AUC@$K$) without sacrificing pass@1, while sustaining exploration and uncovering more diverse solution strategies at scale.

强化学习大模型推理多样性优化策略聚类

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