Restart Sampling for Improving Generative Processes
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
Generative processes that involve solving differential equations, such as diffusion models, frequently necessitate balancing speed and quality. ODE-based samplers are fast but plateau in performance while SDE-based samplers deliver higher sample quality at the cost of increased sampling time. We attribute this difference to sampling errors: ODE-samplers involve smaller discretization errors while stochasticity in SDE contracts accumulated errors. Based on these findings, we propose a novel sampling algorithm called Restart in order to better balance discretization errors and contraction. The sampling method alternates between adding substantial noise in additional forward steps and strictly following a backward ODE. Empirically, Restart sampler surpasses previous SDE and ODE samplers in both speed and accuracy. Restart not only outperforms the previous best SDE results, but also accelerates the sampling speed by 10-fold / 2-fold on CIFAR-10 / ImageNet 64 × 64. In addition, it attains significantly better sample quality than ODE samplers within comparable sampling times. Moreover, Restart better balances text-image alignment/visual quality versus diversity than previous samplers in the large-scale text-to-image Stable Diffusion model pre-trained on LAION 512 × 512. Code is available at https://github.com/Newbeeer/diffusionrestartsampling
Key facts. Cited by 84 papers in the map (data through 14 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 17 of its 31 references that are in the map. First posted on arXiv 26 Jun 2023.
Builds on
17 of 31 references.
Closest ancestors · 9
Same direction, newest first
- 2022On Distillation of Guided Diffusion Models6 Oct
- 2022Classifier-Free Diffusion Guidance26 Jul
- 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps2 Jun
- 2022Elucidating the Design Space of Diffusion-Based Generative Models1 Jun
- 2021High-Resolution Image Synthesis with Latent Diffusion Models20 Dec
- 2021Gotta Go Fast When Generating Data with Score-Based Models28 May
- 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
- 2020Denoising Diffusion Implicit Models6 Oct
- 2020Denoising Diffusion Probabilistic Models19 Jun
Other references · 8
- 2014Microsoft COCO: Common Objects in Contextcited by 17,608
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,355
- 2022LAION-5B: An open large-scale dataset for training next generation image-text modelscited by 4,152
- 2021GLIDE: Towards Photorealistic Image Generation and Editing with Text-Guided Diffusion Modelscited by 3,614
- 2021CLIPScore: A Reference-free Evaluation Metric for Image Captioningcited by 2,424
- 2017User-friendly guarantees for the Langevin Monte Carlo with inaccurate gradientcited by 116
- 2022Poisson Flow Generative Modelscited by 95
- 2023PFGM++: Unlocking the Potential of Physics-Inspired Generative Modelscited by 73
Built on by
Citing papers per year: 2023: 11; 2024: 22; 2025: 30; 2026: 21.
84 papers.
Sort:
By area:
- 2023Consistency Trajectory Models: Learning Probability Flow ODE Trajectory of Diffusioncited by 459
- 2024Generative Flows on Discrete State-Spaces: Enabling Multimodal Flows with Applications to Protein Co-Designcited by 305
- 2023Space-Time Diffusion Features for Zero-Shot Text-Driven Motion Transfercited by 124
- 2024Dirichlet Flow Matching with Applications to DNA Sequence Designcited by 115
- 2025Inductive Moment Matchingcited by 113
- 2023Elucidating the Exposure Bias in Diffusion Modelscited by 101
- 2024Align Your Steps: Optimizing Sampling Schedules in Diffusion Modelscited by 96
- 2023Particle Guidance: non-I.I.D. Diverse Sampling with Diffusion Modelscited by 71
- 2024A Survey on Diffusion Models for Time Series and Spatio-Temporal Datacited by 67
- 2024TabDiff: a Mixed-type Diffusion Model for Tabular Data Generationcited by 63
- 2024StableMoFusion: Towards Robust and Efficient Diffusion-based Motion Generation Frameworkcited by 50
- 2025Inference-time Scaling of Diffusion Models through Classical Searchcited by 49
38 more
- 2025Fast Solvers for Discrete Diffusion Models: Theory and Applications of High-Order Algorithmscited by 47
- 2024Trajectory Consistency Distillation: Improved Latent Consistency Distillation by Semi-Linear Consistency Function with Trajectory Mappingcited by 46
- 2025One-step Diffusion Models with f-Divergence Distribution Matchingcited by 43
- 2024Accelerating Diffusion Models with Parallel Sampling: Inference at Sub-Linear Time Complexitycited by 42
- 2024textitJump Your Steps: Optimizing Sampling Schedule of Discrete Diffusion Modelscited by 40
- 2025Inference-Time Scaling for Flow Models via Stochastic Generation and Rollover Budget Forcingcited by 38
- 2025Efficient Diffusion Models: A Surveycited by 35
- 2024Spatiotemporal Skip Guidance for Enhanced Video Diffusion Samplingcited by 34
- 2024Understanding and Improving Training-free Loss-based Diffusion Guidancecited by 32
- 2023Object-aware Inversion and Reassembly for Image Editingcited by 30
- 2025Test-Time Scaling of Diffusion Models via Noise Trajectory Searchcited by 30
- 2025La-Proteina: Atomistic Protein Generation via Partially Latent Flow Matchingcited by 30
- 2023CoDi: Conditional Diffusion Distillation for Higher-Fidelity and Faster Image Generationcited by 26
- 2024Truncated Consistency Modelscited by 22
- 2023FRDiff : Feature Reuse for Universal Training-free Acceleration of Diffusion Modelscited by 20
- 2024DisCo-Diff: Enhancing Continuous Diffusion Models with Discrete Latentscited by 20
- 2025Taming Video Models for 3D and 4D Generation via Zero-Shot Camera Controlcited by 18
- 2025Not-So-Optimal Transport Flows for 3D Point Cloud Generationcited by 17
- 2025GLASS Flows: Transition Sampling for Alignment of Flow and Diffusion Modelscited by 17
- 2023Correcting Diffusion Generation through Resamplingcited by 15
- 2026Self-Refining Video Samplingcited by 15
- 2024SCott: Accelerating Diffusion Models with Stochastic Consistency Distillationcited by 14
- 2024Not All Prompts Are Made Equal: Prompt-based Pruning of Text-to-Image Diffusion Modelscited by 13
- 2025Dynamic Classifier-Free Diffusion Guidance via Online Feedbackcited by 13
- 2024Mixture of Efficient Diffusion Experts Through Automatic Interval and Sub-Network Selectioncited by 12
- 2025FlexiDiT: Your Diffusion Transformer Can Easily Generate High-Quality Samples with Less Computecited by 12
- 2024Diffusion Rejection Samplingcited by 11
- 2024Towards Understanding and Quantifying Uncertainty for Text-to-Image Generationcited by 11
- 2025Transition Matching: Scalable and Flexible Generative Modelingcited by 11
- 2024Zero-to-Hero: Enhancing Zero-Shot Novel View Synthesis via Attention Map Filteringcited by 9
- 2023Spectroscopy-Guided Discovery of Three-Dimensional Structures of Disordered Materials with Diffusion Modelscited by 8
- 2025Diffusion Adaptive Text Embedding for Text-to-Image Diffusion Modelscited by 8
- 2024Multistable Shape from Shading Emerges from Patch Diffusioncited by 6
- 2025Diffusion Model Quantization: A Reviewcited by 6
- 2025Towards a Golden Classifier-Free Guidance Path via Foresight Fixed Point Iterationscited by 6
- 2025Adaptive Stochastic Coefficients for Accelerating Diffusion Samplingcited by 6
- 2023Your Student is Better Than Expected: Adaptive Teacher-Student Collaboration for Text-Conditional Diffusion Modelscited by 5
- 2025Boost-and-Skip: A Simple Guidance-Free Diffusion for Minority Generationcited by 5
50 most cited of 84 All 84 (JSON)
- 2026Image Tokenizers Should Be Trained Between Images12 Oct 2026
- 2026TrajCFG: Token-Wise Classifier-Free Guidance via Denoising Trajectory Feedback9 Oct 2026
- 2026LiFT: Loop Flow Transformers4 Oct 2026
- 2026PartiCam: Camera Controlled Video Generation with Reward Guidance30 Sep 2026
- 2026Livin' on a Prior: Likelihood Score Approximation for Inverse Problems28 Sep 2026
- 2026ReAL: Accelerating Flow Matching through Segment Advancement with Shared Lookahead27 Sep 2026
- 2026On the Resilience of Text-to-Video Diffusion Models to Hardware Faults30 Aug 2026
- 2026SpecLoR: Spectral Lookahead Rectification for Motion-Coherent Text-to-Video Generation10 Jun 2026
- 2026Mitigating the Contractivity Trap in Diffusion ODEs via Stein Stabilization5 Jun 2026
- 2026CaloTrilogy: Toward a Breakthrough in One-Step, End-to-End, Physics-Guided Shower Generation for Modern Calorimeters2 Jun 2026
- 2026On the Error-Correcting Effects of Stochasticity in Discrete Diffusion26 May 2026
- 2026Triadic Dynamics Aware Diffusion Posterior Sampling for Inverse Problems: Optimizing Guidance and Stochasticity Schedules26 May 2026
38 more
- 2026Guided Trajectory Optimization with Sparse Scaling for Test-Time Diffusion21 May 2026
- 2026CAB: Accelerating Flow and Diffusion Sampling via Rectification and Corrected Adams-Bashforth16 May 2026
- 2026Is Monotonic Sampling Necessary in Diffusion Models?12 May 2026
- 2026Z2-Sampling: Zero-Cost Zigzag Trajectories for Semantic Alignment in Diffusion Models26 Apr 2026
- 2026One step further with Monte-Carlo sampler to guide diffusion better4 Mar 2026
- 2026Constrained Particle Seeking: Solving Diffusion Inverse Problems with Just Forward Passes2 Mar 2026
- 2026Learning To Sample From Diffusion Models Via Inverse Reinforcement Learning9 Feb 2026
- 2026Self-Refining Video Sampling26 Jan 2026
- 2026HyperAlign: Hypernetwork for Efficient Test-Time Alignment of Diffusion Models22 Jan 2026
- 2025Exploring the Design Space of Transition Matching13 Dec 2025
- 2025Sawtooth Sampling for Time Series Denoising Diffusion Implicit Models26 Nov 2025
- 2025Solving Diffusion Inverse Problems with Restart Posterior Sampling24 Nov 2025
- 2025Diffusion Adaptive Text Embedding for Text-to-Image Diffusion Models28 Oct 2025
- 2025Adaptive Stochastic Coefficients for Accelerating Diffusion Sampling27 Oct 2025
- 2025Towards a Golden Classifier-Free Guidance Path via Foresight Fixed Point Iterations24 Oct 2025
- 2025GLASS Flows: Transition Sampling for Alignment of Flow and Diffusion Models29 Sep 2025
- 2025Incomplete Data, Complete Dynamics: A Diffusion Approach24 Sep 2025
- 2025Dynamic Classifier-Free Diffusion Guidance via Online Feedback19 Sep 2025
- 2025Taming Video Models for 3D and 4D Generation via Zero-Shot Camera Control18 Sep 2025
- 2025La-Proteina: Atomistic Protein Generation via Partially Latent Flow Matching13 Jul 2025
- 2025RichControl: Structure- and Appearance-Rich Training-Free Spatial Control for Text-to-Image Generation3 Jul 2025
- 2025Transition Matching: Scalable and Flexible Generative Modeling30 Jun 2025
- 2025Ctrl-Z Sampling: Scaling Diffusion Sampling with Controlled Random Zigzag Explorations25 Jun 2025
- 2025SynBoost: A Synergistic Framework for Fast Sampling of Diffusion Models16 Jun 2025
- 2025The Effect of Stochasticity in Score-Based Diffusion Sampling: a KL Divergence Analysis13 Jun 2025
- 2025Inference-time Scaling of Diffusion Models through Classical Search29 May 2025
- 2025Test-Time Scaling of Diffusion Models via Noise Trajectory Search24 May 2025
- 2025Improving Compositional Generation with Diffusion Models Using Lift Scores19 May 2025
- 2025Diffusion Model Quantization: A Review8 May 2025
- 2025Sampling-Aware Quantization for Diffusion Models4 May 2025
- 2025Inference-Time Scaling for Flow Models via Stochastic Generation and Rollover Budget Forcing25 Mar 2025
- 2025Improving Discriminator Guidance in Diffusion Models20 Mar 2025
- 2025Inductive Moment Matching10 Mar 2025
- 2025FlexiDiT: Your Diffusion Transformer Can Easily Generate High-Quality Samples with Less Compute27 Feb 2025
- 2025One-step Diffusion Models with f-Divergence Distribution Matching21 Feb 2025
- 2025Not-So-Optimal Transport Flows for 3D Point Cloud Generation18 Feb 2025
- 2025Boost-and-Skip: A Simple Guidance-Free Diffusion for Minority Generation10 Feb 2025
- 2025Efficient Diffusion Models: A Survey3 Feb 2025
50 newest of 84 All 84 (JSON)
- 2023Consistency Trajectory Models: Learning Probability Flow ODE Trajectory of Diffusioncited by 459
- 2024Generative Flows on Discrete State-Spaces: Enabling Multimodal Flows with Applications to Protein Co-Designcited by 305
- 2024Dirichlet Flow Matching with Applications to DNA Sequence Designcited by 115
- 2025Inductive Moment Matchingcited by 113
- 2023Elucidating the Exposure Bias in Diffusion Modelscited by 101
- 2024Align Your Steps: Optimizing Sampling Schedules in Diffusion Modelscited by 96
- 2023Particle Guidance: non-I.I.D. Diverse Sampling with Diffusion Modelscited by 71
- 2024A Survey on Diffusion Models for Time Series and Spatio-Temporal Datacited by 67
- 2024TabDiff: a Mixed-type Diffusion Model for Tabular Data Generationcited by 63
- 2024StableMoFusion: Towards Robust and Efficient Diffusion-based Motion Generation Frameworkcited by 50
- 2025Fast Solvers for Discrete Diffusion Models: Theory and Applications of High-Order Algorithmscited by 47
- 2024Trajectory Consistency Distillation: Improved Latent Consistency Distillation by Semi-Linear Consistency Function with Trajectory Mappingcited by 46
12 most cited of 68 All 68 (JSON)
- 2026Image Tokenizers Should Be Trained Between Images12 Oct 2026
- 2026LiFT: Loop Flow Transformers4 Oct 2026
- 2026PartiCam: Camera Controlled Video Generation with Reward Guidance30 Sep 2026
- 2026ReAL: Accelerating Flow Matching through Segment Advancement with Shared Lookahead27 Sep 2026
- 2026On the Resilience of Text-to-Video Diffusion Models to Hardware Faults30 Aug 2026
- 2026SpecLoR: Spectral Lookahead Rectification for Motion-Coherent Text-to-Video Generation10 Jun 2026
- 2026Mitigating the Contractivity Trap in Diffusion ODEs via Stein Stabilization5 Jun 2026
- 2026CaloTrilogy: Toward a Breakthrough in One-Step, End-to-End, Physics-Guided Shower Generation for Modern Calorimeters2 Jun 2026
- 2026On the Error-Correcting Effects of Stochasticity in Discrete Diffusion26 May 2026
- 2026Triadic Dynamics Aware Diffusion Posterior Sampling for Inverse Problems: Optimizing Guidance and Stochasticity Schedules26 May 2026
- 2026Guided Trajectory Optimization with Sparse Scaling for Test-Time Diffusion21 May 2026
- 2026CAB: Accelerating Flow and Diffusion Sampling via Rectification and Corrected Adams-Bashforth16 May 2026
12 newest of 68 All 68 (JSON)
- 2023Space-Time Diffusion Features for Zero-Shot Text-Driven Motion Transfercited by 124
- 2023Object-aware Inversion and Reassembly for Image Editingcited by 30
- 2025La-Proteina: Atomistic Protein Generation via Partially Latent Flow Matchingcited by 30
- 2025Not-So-Optimal Transport Flows for 3D Point Cloud Generationcited by 17
- 2026Self-Refining Video Samplingcited by 15
- 2024Towards Understanding and Quantifying Uncertainty for Text-to-Image Generationcited by 11
- 2024Zero-to-Hero: Enhancing Zero-Shot Novel View Synthesis via Attention Map Filteringcited by 9
- 2023Spectroscopy-Guided Discovery of Three-Dimensional Structures of Disordered Materials with Diffusion Modelscited by 8
- 2024Multistable Shape from Shading Emerges from Patch Diffusioncited by 6
- 2025Boost-and-Skip: A Simple Guidance-Free Diffusion for Minority Generationcited by 5
- 2025Incomplete Data, Complete Dynamics: A Diffusion Approachcited by 3
- 2026One step further with Monte-Carlo sampler to guide diffusion bettercited by 0
12 most cited of 14 All 14 (JSON)
- 2026TrajCFG: Token-Wise Classifier-Free Guidance via Denoising Trajectory Feedback9 Oct 2026
- 2026Livin' on a Prior: Likelihood Score Approximation for Inverse Problems28 Sep 2026
- 2026One step further with Monte-Carlo sampler to guide diffusion better4 Mar 2026
- 2026Self-Refining Video Sampling26 Jan 2026
- 2025Incomplete Data, Complete Dynamics: A Diffusion Approach24 Sep 2025
- 2025La-Proteina: Atomistic Protein Generation via Partially Latent Flow Matching13 Jul 2025
- 2025Not-So-Optimal Transport Flows for 3D Point Cloud Generation18 Feb 2025
- 2025Boost-and-Skip: A Simple Guidance-Free Diffusion for Minority Generation10 Feb 2025
- 2024Towards Understanding and Quantifying Uncertainty for Text-to-Image Generation4 Dec 2024
- 2024Zero-to-Hero: Enhancing Zero-Shot Novel View Synthesis via Attention Map Filtering29 May 2024
- 2024Multistable Shape from Shading Emerges from Patch Diffusion23 May 2024
- 2023Spectroscopy-Guided Discovery of Three-Dimensional Structures of Disordered Materials with Diffusion Models9 Dec 2023
12 newest of 14 All 14 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2023Exploring the Optimal Choice for Generative Processes in Diffusion Models: Ordinary vs Stochastic Differential Equations
- 2023Elucidating the solution space of extended reverse-time SDE for diffusion models
- 2024On the Trajectory Regularity of ODE-based Diffusion Sampling
- 2024An Ordinary Differential Equation Sampler with Stochastic Start for Diffusion Bridge Models
- 2023Fast ODE-based Sampling for Diffusion Models in Around 5 Steps
- 2026Formalizing the Sampling Design Space of Diffusion-Based Generative Models via Adaptive Solvers and Wasserstein-Bounded Timesteps
- 2025Adaptivity and Convergence of Probability Flow ODEs in Diffusion Generative Models
- 2025Are First-Order Diffusion Samplers Really Slower? A Fast Forward-Value Approach