让大模型在长任务中高效探索,只在关键节点发力
Spark: Strategic Policy-Aware Exploration via Dynamic Branching for Long-Horizon Agentic Learning
- 根据决策重要性动态选择探索节点,避免无效计算
- 用更少样本达成更高成功率,未见场景下表现稳定
- 无需人工干预,模型自主判断该在哪扩展探索
强化学习已使大语言模型具备智能体能力,但长时序任务训练仍因高质量轨迹稀缺而困难,尤其在资源受限情况下。现有方法通常扩大采样规模并平均分配计算资源,导致大量计算浪费于无关步骤,且无法保证样本质量。为此,我们提出 extbf{Spark}( extbf{S}trategic extbf{P}olicy- extbf{A}ware explo extbf{R}ation via extbf{K}ey-state dynamic branching),一种通过在关键决策状态动态分支实现资源高效探索的新框架。核心思想是在关键决策点激活自适应分支探索,以探测潜在高价值轨迹,从而实现精准资源分配,优先保障采样质量而非盲目覆盖。该设计利用代理内在的决策信号,降低对人工先验的依赖,使代理能自主扩展探索并提升泛化能力。在多种任务(如具身规划)上的实验表明, extsc{Spark} 以显著更少的训练样本取得更高成功率,在未见过的场景中也表现出强泛化能力。代码与模型权重已开源:https://github.com/jinyangwu/SPARK。
原文摘要 · Abstract (English)
Reinforcement learning has empowered large language models to act as intelligent agents, yet training them for long-horizon tasks remains challenging due to the scarcity of high-quality trajectories, especially under limited resources. Existing methods typically scale up rollout sizes and indiscriminately allocate computational resources among intermediate steps. Such attempts inherently waste substantial computation budget on trivial steps while failing to guarantee sample quality. To address this, we propose \textbf{Spark} (\textbf{S}trategic \textbf{P}olicy-\textbf{A}ware explo\textbf{R}ation via \textbf{K}ey-state dynamic branching), a novel framework that selectively branches at critical decision states for resource-efficient exploration. Our key insight is to activate adaptive branching exploration at critical decision points to probe promising trajectories, thereby achieving precise resource allocation that prioritizes sampling quality over blind coverage. This design leverages the agent's intrinsic decision-making signals to reduce dependence on human priors, enabling the agent to autonomously expand exploration and achieve stronger generalization. Experiments across diverse tasks (e.g., embodied planning), demonstrate that \textsc{Spark} achieves superior success rates with significantly fewer training samples, exhibiting robust generalization even in unseen scenarios. Our code and checkpoints are available at https://github.com/jinyangwu/SPARK.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。