acceptodds
Published as a conference paper at NeurIPS 2026

IdealCache: Rethinking Cache Scheduling in Diffusion Transformers via Ideal Trajectories

shang xue, Li Zhu, Ping Chen

Abstract

Feature caching accelerates Diffusion Transformers by reusing intermediate computations across denoising timesteps, yet existing methods share a fundamental blind spot: they treat the standard reference denoising trajectory as the optimisation target, approximating its per-step states and thus inheriting its output as an implicit quality ceiling. We argue that this target is arbitrary — what truly matters is the ideal result, not the intermediate path taken to reach it. We propose IdealCache, a training-free framework that abandons step-wise imitation of the reference trajectory. Instead, it anchors caching on a per-input ensemble of ideal trajectories constructed from candidate target results, aggregates their geometry into a single graph, and discovers optimal non-uniform schedules via a constrained shortest path on this graph. By decoupling from the reference process, IdealCache breaks the performance bottleneck of trajectory-approximation; our schedules can attain near-lossless fidelity in generation and even surpass on task-specific metrics (e.g., maintaining higher semantic consistency in image editing). Without retraining, IdealCache achieves speedup on Qwen-Image, on Qwen-Image-Edit, and on HunyuanVideo, with competitive or superior quality across tasks.

Related papers

See more related submissions when the market opens.