arXiv Computer Vision
Sep 25

Beyond Attention Masks: Instruction Anchoring for Efficient In-Context Diffusion Generation

The paper introduces AnchorCache, a parameter‑free token‑layout and attention‑mask design that decouples reference tokens from the target in in‑context diffusion transformers. By inserting static text anchors, the method conditions reference representations on the instruction during cache construction, enabling exact key‑value reuse across denoising steps. To restore quality lost by this structural change, the authors employ teacher‑forced velocity distillation followed by a brief on‑policy stage, achieving full‑attention quality while delivering up to 6.40× speedup in diffusion transformer inference across image, speech, and video benchmarks.

By Yangshuai Liu, Zheming Li, Jiaao Li, Kang He, Ziliang Lai, Zhitai Liu, Chengru Song
arXiv AI
Oct 27

ToDoPro: Faster and Better Reference-Based Generation: Early Token Reduction via Downsampled Self-Probing

ToDoPro introduces an early token‑reduction strategy for reference‑based image generation that uses a few denoising steps on 2× downsampled reference images to probe token importance with the existing Diffusion Transformer. Tokens are then hierarchically pruned: low‑importance tokens are discarded, mid‑importance tokens are replaced by their downsampled versions, and high‑importance tokens are retained. Experiments on three base models and two benchmarks show significant inference speedups, reduced peak GPU memory, and improved generation quality by filtering out noisy reference tokens.

By Caiyi Sun, Yujing Sun, Siu-Ming Yiu