让扩散策略实时运行,通过动态预测加速推理
TS-DP: Reinforcement Speculative Decoding For Temporal Adaptive Diffusion Policy Acceleration
- 用小型Transformer模型预判扩散过程,替代部分耗时去噪
- 强化学习动态调节推理强度,任务越难越精细计算
- 实测速度提升4.17倍,25赫兹实时控制无性能损失
扩散策略(DP)在具身控制中表现优异,但因需多次迭代去噪导致推理延迟高、计算成本大。具身任务具有时间上的动态复杂性,静态压缩方法如量化难以应对,而推测解码虽能无损适应却未被充分探索。本文提出首个具备时序自适应能力的推测解码框架TS-DP,解决两个核心挑战:如何在任务难度变化时以更低开销保持去噪质量,以及如何动态调整计算量。首先,通过蒸馏训练一个基于Transformer的轻量级草稿模型,替代主模型的部分去噪调用;其次,设计基于强化学习的调度器,根据任务难度动态调整推测参数,在保证精度的同时提升效率。在多个具身环境中的实验表明,TS-DP可实现最高4.17倍的推理加速,超过94%的草稿被接受,达到25赫兹的推理频率,实现无性能下降的实时扩散控制。
原文摘要 · Abstract (English)
Diffusion Policy (DP) excels in embodied control but suffers from high inference latency and computational cost due to multiple iterative denoising steps. The temporal complexity of embodied tasks demands a dynamic and adaptable computation mode. Static and lossy acceleration methods, such as quantization, fail to handle such dynamic embodied tasks, while speculative decoding offers a lossless and adaptive yet underexplored alternative for DP. However, it is non-trivial to address the following challenges: how to match the base model's denoising quality at lower cost under time-varying task difficulty in embodied settings, and how to dynamically and interactively adjust computation based on task difficulty in such environments. In this paper, we propose Temporal-aware Reinforcement-based Speculative Diffusion Policy (TS-DP), the first framework that enables speculative decoding for DP with temporal adaptivity. First, to handle dynamic environments where task difficulty varies over time, we distill a Transformer-based drafter to imitate the base model and replace its costly denoising calls. Second, an RL-based scheduler further adapts to time-varying task difficulty by adjusting speculative parameters to maintain accuracy while improving efficiency. Extensive experiments across diverse embodied environments demonstrate that TS-DP achieves up to 4.17 times faster inference with over 94% accepted drafts, reaching an inference frequency of 25 Hz and enabling real-time diffusion-based control without performance degradation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。