arXiv:2605.13316cs.CV2026-05被引 1

通过测试时动态剪枝,实现动作扩散模型5倍加速且无性能损失。

Test-time Sparsity for Extreme Fast Action Diffusion

论文配图:Test-time Sparsity for Extreme Fast Action Diffusion
图 1 · 摘自论文原文
  • 测试时动态预测可剪枝的残差计算,实现每步推理的自适应加速。
  • 在保持0.475秒(47.5Hz)推理频率下,降低92%计算量,零损失。
  • 支持多轮推演和开放环境,适合高实时性动作生成任务。

动作扩散模型虽能生成高保真动作,但因其迭代去噪机制导致计算开销巨大。现有加速技术依赖缓存特征复用,但在开放环境中因策略动态变化难以适应。本文提出测试时稀疏化方法,通过动态预测每步前向中可剪枝的残差计算来加速。针对两个瓶颈:一是设计轻量级共享编码器的剪枝器,并将编码与剪枝解耦,通过异步并行处理所有去噪步骤,将非解码延迟降至毫秒级;二是引入全向特征复用策略,从当前前向、前去噪步骤及早期推演迭代中选择性复用特征,实现95%稀疏度。通过少量动作轨迹采样监督逐步训练稀疏化过程。实验表明,本方法减少92%浮点运算量,推理速度提升5倍,达到47.5Hz,性能无损。代码已开源。

原文摘要 · Abstract (English)

Action diffusion excels at high-fidelity action generation but incurs heavy computational costs owing to its iterative denoising nature. Despite current technologies showing promise in accelerating diffusion transformers by reusing the cached features, they struggle to adapt to policy dynamics arising from diverse perceptions and multi-round rollout iterations in open environments. We propose test-time sparsity to tackle this challenge, which aims to accelerate action diffusion by dynamically predicting prunable residual computations for each model forward at test time. However, two bottlenecks remain in this paradigm: 1) repetitive conditional encoding and pruning offset most potential speed gains, and 2) the features cached from previous denoising timesteps cannot constrain large pruning errors under aggressive sparsity. To address the first bottleneck, we design a highly parallelized inference pipeline that minimizes the non-decoder delay to milliseconds. Specifically, we first design a lightweight pruner that shares the encoder with the diffusion transformer. Then, we decouple the encoding and pruning from the autoregressive denoising loop by processing all denoising timesteps in parallel, and overlap the pruner with the decoder forward inference through asynchronism. To overcome the second bottleneck, we introduce an omnidirectional reusing strategy, which achieves 95% sparsity by selectively reusing features cached from the current forward, previous denoising timesteps, and earlier rollout iterations. To learn the rollout-level reusing strategies, we sample a few action trajectories to supervise the sparsified diffusion step by step. Extensive experiments demonstrate that our method reduces FLOPs by 92% and accelerates action generation by 5x, achieving lossless performance with an inference frequency of 47.5 Hz. Our code is available at https://github.com/ky-ji/Test-time-Sparsity.

动作生成扩散模型稀疏推理实时生成

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