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

Institutions

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
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
NOV
article

Inference-Time Pointwise Velocity for Diffusion Acceleration

Itay Hazan, Lior Wolf
ACM Transactions on Graphics
Generative Adversarial Networks and Image Synthesis
article

Inference-Time Pointwise Velocity for Diffusion Acceleration

Itay Hazan, Lior Wolf
article en

Abstract

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.

ACM Transactions on GraphicsVol. 45(6)
Tel Aviv University (IL)
Generative Adversarial Networks and Image Synthesis
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.

Laika access coming soon