arXiv:2605.17295cs.LGcs.CL2026-05

提出离线重要性采样方法,让大模型生成更多元解法路径。

DISA: Offline Importance Sampling for Distribution-Matching LLM-RL

论文配图:DISA: Offline Importance Sampling for Distribution-Matching LLM-RL
图 1 · 摘自论文原文
  • 离线采样轨迹并用重要性采样估算分布匹配所需的分区函数
  • 在六项数学与三项代码任务中表现优于现有方法,最高提升13.8分
  • 适合需要多解法生成的推理系统,尤其关注策略多样性

现代推理代理需生成多个有效解法路径、计划或工具使用轨迹。标准奖励最大化强化学习易陷入单一高奖励模式,而分布匹配强化学习旨在将概率质量均匀分配至整个奖励塑造的解法集合。实现此目标需计算提示相关的轨迹空间分区函数。现有分布匹配方法在线联合学习该函数,导致分区函数校准误差直接影响策略更新且无法独立诊断。本文提出DISA(Decoupled Importance-Sampled Anchoring),将校准问题移出强化学习循环:离线生成提议轨迹,通过重要性采样估计分区函数,并在策略优化前冻结其值。该解耦设计保持分布匹配目标,同时在数据、梯度、损失和诊断层面严格分离分区函数估计与策略学习。实验显示,在两个开源大模型上,六项数学与三项代码基准测试中,DISA表现与或超越在线耦合基线FlowRL,数学平均性能优于GRPO与GSPO,且在相同离线轨迹上比LoRASFT蒸馏高出最多13.8个Mean@8点。人工评估进一步表明,DISA保留了显著更多的策略级多样性,提案强度与逆温度的敏感性分析结果符合理论预测的偏差-方差模式。

原文摘要 · Abstract (English)

Modern reasoning agents are increasingly evaluated on their ability to generate multiple valid solution paths, plans, or tool-use traces for a given input. Standard reward-maximizing RL tends to collapse onto the most easily reinforced high-reward mode, whereas distribution-matching RL aims to allocate probability mass across the entire reward-shaped solution set. Achieving this objective requires computing a prompt-dependent partition function over the trajectory space. Because existing distribution-matching methods learn this partition function online alongside the policy, calibration errors in the partition function directly distort policy updates and remain impossible to diagnose independently. We introduce DISA, short for Decoupled Importance-Sampled Anchoring, which moves this calibration problem outside the RL loop. DISA draws proposal trajectories offline, estimates the partition function via importance sampling, and freezes the resulting partition-function estimate before policy optimization begins. This decoupling preserves the distribution-matching objective while strictly separating partition-function estimation from policy learning in data, gradients, loss, and diagnostics. Empirically, on two open-weight backbones across six math and three code benchmarks, DISA matches or exceeds the online-coupled distribution-matching baseline FlowRL, outperforms rewardmaximization baselines GRPO and GSPO on math averages, and exceeds LoRASFT distillation by up to 13.8 Mean@8 points on the same offline trajectories. An LLM-as-judge evaluation further shows that DISA retains substantially more strategy-level diversity than reward-maximization baselines, and sensitivity studies on the proposal strength and inverse temperature follow the bias-variance pattern predicted by the analysis.

强化学习大模型推理分布匹配多样性生成

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