arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2605.22015cs.CVcs.AR
收藏

ORBIS: 通过分布感知匹配的输出引导标记减少以加速视频扩散

ORBIS: Output-Guided Token Reduction with Distribution-Aware Matching for Video Diffusion Acceleration

KAIST(韩国科学技术院)

Hangyeol Lee, Joo-Young Kim

AI总结:

本文提出ORBIS,一种针对视频扩散Transformer的SW-HW协同设计加速器,通过利用前一时间步的输出激活获得更准确的token相似性,从而提高匹配质量并实现更高的标记减少比例,同时引入分布感知标记匹配算法和专用硬件设计,实现比现有方法更高的标记减少率、更快的速度和更低的能耗。

摘要:

扩散Transformer(DiT)已发展为生成高质量图像和视频的强大模型架构。在视频DiT中,3D空间时间注意力使token长度与帧数成正比,显著增加计算成本。标记减少方法通过利用空间冗余来缓解这一成本,但现有方法依赖于不准确的相似性估计和轻量级匹配算法,导致匹配质量差且仅带来微小的加速效果。为克服这些限制,我们提出了ORBIS,一种为视频DiT设计的SW-HW协同加速器。ORBIS利用前一时间步的输出激活以获得更准确的token间相似性,显著提高匹配质量并实现更高的token减少比例。我们进一步引入了分布感知标记匹配(DATM)算法,该算法捕捉全局token分布并显式最小化token对损失以获得额外收益。为了完全隐藏DATM延迟,我们设计了专用、深度流水线化的硬件并通过量化来最小化其硬件成本,仅占用总面积的2.4%,且精度损失可忽略不计。大量实验表明,ORBIS的token减少比例比最先进的方法AsymRnR高约2倍,相比NVIDIA A100 GPU实现了高达4.5倍的速度提升和79.3%的能耗降低。

英文摘要:

Diffusion Transformer (DiT) has emerged as a powerful model architecture for generating high-quality images and videos. In the case of video DiT, 3D Spatio-Temporal Attention increases token length in proportion to the number of frames, sharply increasing computational cost. Token reduction methods mitigate this cost by exploiting spatial redundancy, but existing approaches rely on inaccurate similarity estimates and lightweight matching algorithms, resulting in poor matching quality and only marginal acceleration. To overcome these limitations, we propose ORBIS, an SW-HW co-designed accelerator for video DiT. ORBIS leverages the output activation from the previous timestep to obtain more accurate inter-token similarity, substantially improving matching quality and enabling a higher token reduction ratio. We further introduce a Distribution-Aware Token Matching (DATM) algorithm that captures global token distribution and explicitly minimizes token-pair loss for additional gains. To fully hide DATM latency, we design specialized, deeply pipelined hardware and minimize its hardware cost through quantization, occupying only 2.4% of total area with negligible accuracy loss. Extensive experiments show that ORBIS achieves about 2x higher token reduction ratio than the state-of-the-art approach, AsymRnR, while delivering up to 4.5x speedup and 79.3% energy reduction compared to an NVIDIA A100 GPU.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑