arXiv:2511.20906cs.ROcs.AI2025-11被引 1

让机器人控制器根据任务难易动态调整计算量,省时又高效。

DASIP: Dynamic Test-Time Compute Scaling for Robot Control with Stochastic Interpolant Policies

  • 通过难度分类器实时决定每步的计算预算和求解器
  • 在多种任务上实现2.6至4.4倍的计算量减少
  • 适合对推理效率要求高的生成式机器人控制场景

基于扩散模型和流模型的策略在长时程机器人操作与模仿学习任务中表现优异。然而,这些控制器在每个控制步骤都使用固定的推理预算,无论任务复杂度如何,导致简单子任务计算浪费,复杂任务可能表现不足。为此,我们提出难度感知随机插值策略(DA-SIP),使机器人控制器能根据任务难度实时自适应调整积分时长。该方法通过观察分析任务难度,动态选择步数预算、最优求解器变体以及常微分方程/随机微分方程的积分方式。DA-SIP基于随机插值框架,为扩散与流模型提供统一的训练与推理配置支持。在多个多样化操作任务上的综合测试表明,DA-SIP在保持与固定最大计算量基线相当的任务成功率的同时,总计算时间减少2.6至4.4倍。通过在此框架内实现自适应计算,DA-SIP将生成式机器人控制器转变为高效、任务感知的系统,智能分配推理资源以获得最大收益。

原文摘要 · Abstract (English)

Diffusion- and flow-based policies deliver state-of-the-art performance on long-horizon robotic manipulation and imitation learning tasks. However, these controllers employ a fixed inference budget at every control step, regardless of task complexity, leading to computational inefficiency for simple subtasks while potentially underperforming on challenging ones. To address these issues, we introduce Difficulty-Aware Stochastic Interpolant Policy (DA-SIP), a framework that enables robotic controllers to adaptively adjust their integration horizon in real time based on task difficulty. Our approach employs a difficulty classifier that analyzes observations to dynamically select the step budget, the optimal solver variant, and ODE/SDE integration at each control cycle. DA-SIP builds upon the stochastic interpolant formulation to provide a unified framework that unlocks diverse training and inference configurations for diffusion- and flow-based policies. Through comprehensive benchmarks across diverse manipulation tasks, DA-SIP achieves 2.6-4.4x reduction in total computation time while maintaining task success rates comparable to fixed maximum-computation baselines. By implementing adaptive computation within this framework, DA-SIP transforms generative robot controllers into efficient, task-aware systems that intelligently allocate inference resources where they provide the greatest benefit.

机器人控制生成模型自适应计算扩散模型

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