GeoSPRINT: Geometric Redundancy-Aware Step Pruning for Inference in Diffusion Trajectories
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
Diffusion models achieve high sample quality but remain expensive at inference time because sampling requires many sequential neural function evaluations (NFEs). Existing acceleration methods either use fixed step-skipping schedules, adapt step sizes based on local numerical error, or require additional training. We introduce GeoSPRINT (Geometric Step Pruning for Inference in Trajectories), a training-free framework for constructing non-uniform sampling schedules from the geometry of denoising trajectories. GeoSPRINT detects geometrically redundant steps using a hyperplanarity test in latent space, implemented efficiently via QR factorization, and converts the resulting redundancy profile into a sampling schedule that allocates more steps to high-curvature regions of the trajectory. In addition, we introduce the trajectory projection score αtraj, a residual-variance metric that quantifies trajectory straightness and serves as a model-free diagnostic for rectified flow quality. Across CIFAR-10 (32×32), LSUN Church (256×256), and Stable Diffusion v1.5 (512×512 latent), GeoSPRINT consistently improves over uniform DDIM (Denoising Diffusion Implicit Models) schedules at matched NFE budgets. On CIFAR-10, GeoSPRINT improves FID (Fr'echet Inception Distance) by 0.7-1.1 over DDIM across 49-89 NFEs and surpasses DPM-Solver++ at NFE≥30 despite using a first-order DDIM solver. On LSUN Church, it reduces FID from 1.48 to 1.26 at 52 steps, and on Stable Diffusion v1.5 it achieves up to 1.93 FID improvement over DDIM. These results show that trajectory geometry provides a useful global signal for allocating inference steps and that schedule quality can substantially improve diffusion sampling efficiency without retraining.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 18 of its 20 references that are in the map. First posted on arXiv 2 Sep 2026.
Builds on
18 of 20 references.
Closest ancestors · 12
Same direction, newest first
- 2026Formalizing the Sampling Design Space of Diffusion-Based Generative Models via Adaptive Solvers and Wasserstein-Bounded Timesteps13 Feb
- 2024Adaptive Non-uniform Timestep Sampling for Accelerating Diffusion Model Training15 Nov
- 2024Simplifying, Stabilizing and Scaling Continuous-Time Consistency Models14 Oct
- 2023Consistency Models2 Mar
- 2023UniPC: A Unified Predictor-Corrector Framework for Fast Sampling of Diffusion Models9 Feb
- 2022DPM-Solver++: Fast Solver for Guided Sampling of Diffusion Probabilistic Models2 Nov
- 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps2 Jun
- 2021High-Resolution Image Synthesis with Latent Diffusion Models20 Dec
- 2021On Fast Sampling of Diffusion Probabilistic Models31 May
- 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
- 2020Denoising Diffusion Implicit Models6 Oct
- 2020Denoising Diffusion Probabilistic Models19 Jun
Other references · 6
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
- 2022Video Diffusion Modelscited by 2,280
- 2022Equivariant Diffusion for Molecule Generation in 3Dcited by 666
- 2024Consistency Flow Matching: Defining Straight Flows with Velocity Consistencycited by 101
- 2024Rectified Diffusion: Straightness Is Not Your Need in Rectified Flowcited by 59
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026GeoShrink: Accelerating Diffusion Transformers with Two Lines of Code
- 2025A-FloPS: Accelerating Diffusion Models via Adaptive Flow Path Sampler
- 2025Guided Path Sampling: Steering Diffusion Models Back on Track with Principled Path Guidance
- 2025SCoT: Unifying Consistency Models and Rectified Flows via Straight-Consistent Trajectories
- 2026Denoising as Path Planning: Training-Free Acceleration of Diffusion Models with DPCache
- 2024Elucidating Flow Matching ODE Dynamics with Respect to Data Geometries and Denoisers
- 2026Geometry-Aware Diffusion Guidance via Curvature-Adaptive Tubular Correction
- 2025Shortcutting Pre-trained Flow Matching Diffusion Models is Almost Free Lunch