arXiv:2601.12894cs.ROcs.CV2026-01被引 4

通过动态剪枝加速扩散策略,实现机器人实时视觉运动控制。

Sparse ActionGen: Accelerating Diffusion Policy with Real-time Pruning

  • 根据环境动态调整剪枝策略,实现滚动自适应的计算剪枝与缓存复用。
  • 在多个机器人基准上实现最高4倍生成速度提升,性能零损失。
  • 适合需要低延迟动作生成的实时机器人控制场景。

扩散策略因能有效建模多模态动作分布而主导动作生成,但其多步去噪过程难以满足实时视觉运动控制需求。现有基于缓存的加速方法通常依赖静态剪枝策略,无法适应机器人-环境交互的动态特性,导致性能不佳。本文提出稀疏动作生成框架SAG,采用滚动自适应的剪枝-复用机制:先全局识别可剪枝计算,再利用缓存激活进行替代。为捕捉滚动动态,SAG设计了观测条件化的扩散剪枝器,采用参数与推理高效结构实现实时预测。此外,提出一种跨时间步与模块的“一劳永逸”复用策略,以锯齿模式最小化全局冗余。在多个机器人基准上的实验表明,SAG可实现最高4倍生成速度提升,且性能无损。

原文摘要 · Abstract (English)

Diffusion Policy has dominated action generation due to its strong capabilities for modeling multi-modal action distributions, but its multi-step denoising processes make it impractical for real-time visuomotor control. Existing caching-based acceleration methods typically rely on $\textit{static}$ schedules that fail to adapt to the $\textit{dynamics}$ of robot-environment interactions, thereby leading to suboptimal performance. In this paper, we propose $\underline{\textbf{S}}$parse $\underline{\textbf{A}}$ction$\underline{\textbf{G}}$en ($\textbf{SAG}$) for extremely sparse action generation. To accommodate the iterative interactions, SAG customizes a rollout-adaptive prune-then-reuse mechanism that first identifies prunable computations globally and then reuses cached activations to substitute them during action diffusion. To capture the rollout dynamics, SAG parameterizes an observation-conditioned diffusion pruner for environment-aware adaptation and instantiates it with a highly parameter- and inference-efficient design for real-time prediction. Furthermore, SAG introduces a one-for-all reusing strategy that reuses activations across both timesteps and blocks in a zig-zag manner, minimizing the global redundancy. Extensive experiments on multiple robotic benchmarks demonstrate that SAG achieves up to 4$\times$ generation speedup without sacrificing performance. Project Page: https://sparse-actiongen.github.io.

扩散模型机器人控制实时推理剪枝加速

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