arXiv:2511.02690cs.LG2025-11NeurIPS被引 3

通过渐进式约束训练,让大模型推理更高效、更快

Curriculum Design for Trajectory-Constrained Agent: Compressing Chain-of-Thought Tokens in LLMs

  • 用逐步收紧约束的课程学习法训练智能体
  • 在数学推理任务中实现推理过程压缩,提速显著
  • 适合资源受限场景下部署的大模型推理优化

在部署时需严格遵守资源预算或安全要求等约束条件下训练智能体,带来巨大挑战,尤其当这些约束使任务变得复杂时。本文提出一种课程学习策略,训练过程中逐步收紧约束,使智能体能分阶段掌握部署需求。受无约束强化学习中自适应节奏学习的启发,该方法先在简化约束下训练,再逐步引入完整部署条件,实现平滑过渡到高难度环境。我们基于二叉树马尔可夫决策过程(MDP)的强化学习代理进行了理论分析,证明该课程策略相比从一开始就施加完整轨迹约束的基线方法,能显著加速训练。此外,我们在多种场景中实证验证了该方法的有效性和普适性,涵盖二叉树MDP、多任务导航域以及两个基准的数学推理任务。结果表明,课程设计可有效提升受复杂轨迹约束的智能体在部署时的效率与性能。尤其在大语言模型(LLM)应用中,该策略可压缩输出思维链(chain-of-thought)token,实现在消费级硬件上的显著推理加速,展现出在资源受限部署中的强大适用性。

原文摘要 · Abstract (English)

Training agents to operate under strict constraints during deployment, such as limited resource budgets or stringent safety requirements, presents significant challenges, especially when these constraints render the task complex. In this work, we propose a curriculum learning strategy that gradually tightens constraints during training, enabling the agent to incrementally master the deployment requirements. Inspired by self-paced learning techniques in unconstrained reinforcement learning (RL), our approach facilitates a smoother transition to challenging environments by initially training on simplified versions of the constraints and progressively introducing the full deployment conditions. We provide a theoretical analysis using an RL agent in a binary-tree Markov Decision Process (MDP) to demonstrate that our curriculum strategy can accelerate training relative to a baseline approach that imposes the trajectory constraints from the outset. Moreover, we empirically validate the effectiveness and generality of our method across both RL and large language model (LLM) agents in diverse settings, including a binary-tree MDP, a multi-task navigation domain, and a math reasoning task with two benchmarks. These results highlight the potential of curriculum design in enhancing the efficiency and performance of agents operating under complex trajectory constraints during deployment. Moreover, when applied to LLMs, our strategy enables compression of output chain-of-thought tokens, achieving a substantial inference speedup on consumer hardware, demonstrating its effectiveness for resource-constrained deployment.

大模型推理课程学习链式思维压缩强化学习

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