Inference-Time Pointwise Velocity for Diffusion Acceleration
Large rectified-flow image generators such as FLUX and Z-Image deliver state-of-the-art quality, but their sampler advances every latent coordinate under a single global time. We argue that this uniform schedule is wasteful: at any intermediate noise level some image regions are already structurally settled while others still carry most of their stochasticity, and the settled regions would benefit the unsettled ones if allowed to advance further. We introduce ITPV, an inference-time, training-free method that assigns each latent coordinate its own denoising speed. ITPV generalises adaptive Layer Normalization to accept a per-token time map, reusing the pretrained time embedder and modulation weights without architectural changes. At every step it constructs a single speculative future latent through a Tweedie-style extrapolation, queries the model once at this accelerated time, and grades each coordinate by the cosine similarity between its current and speculative velocities. Coordinates whose direction would not change after the jump are advanced aggressively; coordinates whose direction would change stay on the canonical schedule. The result is a strict generalisation of standard sampling: it recovers the canonical trajectory exactly in the limit of zero per-coordinate speedup, requires no caching or profiling, and integrates with classifier-free guidance through a single extra forward pass per step. On FLUX.2-Klein-4B-Base, Z-Image, and Stable Diffusion 3, ITPV at 16 sampling steps matches the perceptual quality of 50-step classifier-free guidance, as confirmed by automatic distribution, quality, and diversity metrics on 30K MS-COCO captions and by the GenEval2 compositional benchmark. At a matched compute budget, ITPV improves over every training-free accelerator we compare against, including state-of-the-art samplers, caching, and region-adaptive methods. We further validate our results with a blinded pairwise user study with 14 participants on two of the backbones.
Authors
- Itay Hazan (ORCID: https://orcid.org/0000-0001-7464-056X)
- Lior Wolf (ORCID: https://orcid.org/0000-0001-5578-8892)
Institutions
- Tel Aviv University (IL)
Publication Details
- Journal
- ACM Transactions on Graphics
- Published
- 2026-11-09
- DOI
- https://doi.org/10.1145/3842513
- Primary Topic
- Generative Adversarial Networks and Image Synthesis
- Type
- article
- Field-Weighted Citation Impact
- 0.00