arXiv:2601.17982cs.CLcs.AI2026-01Conference of the …被引 3

通过奖励语义多样性,让小模型在有限算力下更高效地探索解题思路。

SD-E$^2$: Semantic Exploration for Reasoning Under Token Budgets

  • 用嵌入空间的语义差异衡量推理路径多样性,替代表面形式新颖性。
  • 在GSM8K上比基线模型高27.4个百分点,平均每题发现9.8种不同策略。
  • 适合资源受限场景下的小模型推理能力提升,尤其擅长复杂问题求解。

小型语言模型(SLMs)在复杂推理任务中表现不佳,主要因为受限于严格的计算预算,探索成本过高。本文提出语义多样性探索-利用(SD-E²)框架,通过强化学习显式优化生成推理轨迹中的语义多样性。采用冻结的句子嵌入模型,为每个推理路径分配多样性奖励,该奖励捕捉(i)覆盖的语义上不同的解题策略数量,以及(ii)其在嵌入空间中的平均成对差异度,而非表面形式的新颖性。该多样性奖励与结果正确性和解题效率结合,形成经过z-score归一化的多目标目标函数,有效稳定训练过程。在GSM8K数据集上,SD-E²相比基础Qwen2.5-3B-Instruct模型、强基线GRPO-CFL和GRPO-CFEE分别提升27.4、5.2和1.5个百分点,平均每个问题发现9.8种语义上不同的策略。在MedMCQA上性能提升至49.64%(基线38.37%),并在更难的AIME基准(1983–2025)上达到13.28%(基线6.74%)。结果表明,奖励语义新颖性可为训练推理型小模型提供更高效的探索-利用信号。通过引入认知适应——调整推理结构而非逐令牌计算,SD-E²为资源受限模型提供了互补的效率提升路径。

原文摘要 · Abstract (English)

Small language models (SLMs) struggle with complex reasoning because exploration is expensive under tight compute budgets. We introduce Semantic Diversity-Exploration-Exploitation (SD-E$^2$), a reinforcement learning framework that makes exploration explicit by optimizing semantic diversity in generated reasoning trajectories. Using a frozen sentence-embedding model, SD-E$^2$ assigns a diversity reward that captures (i) the coverage of semantically distinct solution strategies and (ii) their average pairwise dissimilarity in embedding space, rather than surface-form novelty. This diversity reward is combined with outcome correctness and solution efficiency in a z-score-normalized multi-objective objective that stabilizes training. On GSM8K, SD-E$^2$ surpasses the base Qwen2.5-3B-Instruct and strong GRPO baselines (GRPO-CFL and GRPO-CFEE) by +27.4, +5.2, and +1.5 percentage points, respectively, while discovering on average 9.8 semantically distinct strategies per question. We further improve MedMCQA to 49.64% versus 38.37% for the base model and show gains on the harder AIME benchmark (1983-2025), reaching 13.28% versus 6.74% for the base. These results indicate that rewarding semantic novelty yields a more compute-efficient exploration-exploitation signal for training reasoning-capable SLMs. By introducing cognitive adaptation-adjusting the reasoning process structure rather than per-token computation-SD-E$^2$ offers a complementary path to efficiency gains in resource-constrained models.

小模型推理增强强化学习语义多样性

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