arXiv Computer Vision

SoftCache: Adaptive Step Size Caching for Diffusion Transformers

SoftCache introduces an adaptive step size caching mechanism for diffusion transformers, replacing hard binary caching with a soft approach that uses a headroom signal to determine how far cached steps advance. By allowing steps to jump beyond scheduled nodes and shrink as a refresh nears, SoftCache achieves significant speedups—between 3.5× and 6.9×—across image and video models such as FLUX, HunyuanVideo, and Wan2.1, outperforming all baselines on four fidelity metrics.

arXiv Machine Learning
Sep 14

Fixed State, Long Reach: What a Constant-Size Cache Buys Block Diffusion at Scale

The paper investigates how block‑diffusion language models can use a constant‑size cache to enable efficient parallel decoding. By employing sequence mixers that summarize completed blocks into a reusable state and a block‑causal training objective, the authors pretrain three 3B block‑diffusion denoisers (attention, Mamba, and hybrid) on 300 B tokens. The resulting state‑space cache remains O(1) in memory and latency regardless of context length, yielding significant speed‑up and memory savings compared to traditional attention‑based caches, especially at very long sequences.

By Vaibhav Singh, Pierre-Andr\'e No\"el, Torsten Scholak, Eugene Belilovsky, Oleksiy Ostapenko
arXiv AI
Oct 1

In-Flight KV Cache with Clean Anchors for Faster Autoregressive Video Diffusion

arXiv:2609.32540v2 Announce Type: replace-cross Abstract: Few-step autoregressive video diffusion generates a long video by splitting the video into temporal chunks and generating chunk-by-chunk, eac...

By Yikai Wang, Xiao Han, Mengmeng Xu, Juan Camilo Perez, Yiannis Douratsos, Sen He, Zijian Zhou, Fei Zhang, Zhaochong An, Juan-Manuel Perez-Rua, Chen Change Loy, Tao Xiang
arXiv Computer Vision
Oct 8

MORCA: Offline-to-Online Reinforcement Learning for Adaptive Cache Reuse in Video Diffusion Acceleration

arXiv:2610.10457v1 Announce Type: new Abstract: Diffusion Transformers (DiTs) achieve remarkable performance in video synthesis, but their iterative denoising process suffers from high inference late...

By Yuxiang Xiong, Ruiyan Wang, Wenqiang Wang, Teng Hu, Songhang Shen, Bohao Feng, Hongqian Deng, Ran Yi
arXiv Machine Learning
2d ago

DIFFLET: A Fast Inference Engine for Diffusion Transformer Models on AWS Trainium

DIFFLET is a serving engine for diffusion transformer models on AWS Trainium that allows five model pipelines to share a single compilation backend. It splits joint attention across cores by attention head and uses a host controller to decide whether to run the denoiser or reuse previous outputs at each denoising step. Across five image and video model families, DIFFLET’s fused attention accelerates denoiser steps by 1.2–4.3× compared to PyTorch/XLA, and resident serving is 1.5–3.4× faster than standalone runs that reload weights.

By Bin Ma, Min Yu, Vijay Kodamalla, Jialin Liu, Dong Li