只处理需要删的区域,让视频去物更快不卡顿
YOSE: You Only Select Essential Tokens for Efficient DiT-based Video Object Removal

- 根据遮罩区域动态选关键帧,变长处理提升效率
- 推理速度随遮罩面积线性增长,最高快2.5倍
- 适合需要实时视频编辑的开发者和创作者
基于扩散变换器(DiT)的视频去物技术虽有优异视觉效果,但推理延迟高。例如,MiniMax Remover 虽达到顶尖画质,却仅约10FPS,主要因对全时空标记进行密集计算,即便仅小范围遮罩需处理。本文提出 YOSE 框架,引入两个核心组件:批量可变长度索引(BVI)与扩散过程模拟器(DiffSim)。BVI 是可微分动态索引操作,依据遮罩信息自适应选择必要标记,实现样本间可变长度处理。DiffSim 通过模拟未遮罩区域在 DiT 自注意力中的影响,保持遮罩区域语义一致性。该设计使推理时间近似与遮罩区域大小线性相关,而传统方法计算量恒定。大量实验表明,YOSE 在70%情况下实现最高2.5倍加速,同时保持与基线相当的视觉质量。代码已开源:https://github.com/Wucy0519/YOSE-CVPR26。
原文摘要 · Abstract (English)
Recent advances in Diffusion Transformer (DiT)-based video generation technologies have shown impressive results for video object removal. However, these methods still suffer from substantial inference latency. For instance, although MiniMax Remover achieves state-of-the-art visual quality, it operates at only around 10FPS, primarily due to dense computations over the entire spatiotemporal token space, even when only a small masked region actually requires processing. In this paper, we present YOSE, You Only Select Essential Tokens, an efficient fine-tuning framework. YOSE introduces two key components: Batch Variable-length Indexing (BVI) and Diffusion Process Simulator (DiffSim) Module. BVI is a differentiable dynamic indexing operator that adaptively selects essential tokens based on mask information, enabling variable-length token processing across samples. DiffSim provides a diffusion process approximation mechanism for unmasked tokens, which simulates the influence of unmasked regions within DiT self-attention to maintain semantic consistency for masked tokens. With these designs, YOSE achieves mask-aware acceleration, where the inference time scales approximately linearly with the masked regions, in contrast to full-token diffusion methods whose computation remains constant regardless of the mask size. Extensive experiments demonstrate that YOSE achieves up to 2.5X speedup in 70% of cases while maintaining visual quality comparable to the baseline. Code is available at: https://github.com/Wucy0519/YOSE-CVPR26.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。