Chain-of-Trajectories: Unlocking the Intrinsic Generative Optimality of Diffusion Models via Graph-Theoretic Planning
On the map: Diffusion models → Diffusion and flow policies for robotics
Abstract (arXiv)
Diffusion models operate in a reflexive System 1 mode, constrained by a fixed, content-agnostic sampling schedule. This rigidity arises from the curse of state dimensionality, where the combinatorial explosion of possible states in the high-dimensional noise manifold renders explicit trajectory planning intractable and leads to systematic computational misallocation. To address this, we introduce Chain-of-Trajectories (CoTj), a train-free framework enabling System 2 deliberative planning. Central to CoTj is Diffusion DNA, a low-dimensional signature that quantifies per-stage denoising difficulty and serves as a proxy for the high-dimensional state space, allowing us to reformulate sampling as graph planning on a directed acyclic graph. Through a Predict-Plan-Execute paradigm, CoTj dynamically allocates computational effort to the most challenging generative phases. Experiments across multiple generative models demonstrate that CoTj discovers context-aware trajectories, improving output quality and stability while reducing redundant computation. This work establishes a new foundation for resource-aware, planning-based diffusion modeling. The code is available at https://github.com/UnicomAI/CoTj.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in Diffusion and flow policies for robotics, one of 14 directions in Diffusion models. Builds on 34 of its 41 references that are in the map. First posted on arXiv 16 Mar 2026.
Builds on
34 of 41 references.
References · 34
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
- 2020Denoising Diffusion Implicit Modelscited by 9,655
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,094
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
- 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Modelscited by 3,155
- 2022Elucidating the Design Space of Diffusion-Based Generative Modelscited by 2,994
- 2025Wan: Open and Advanced Large-Scale Video Generative Modelscited by 2,536
- 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Stepscited by 1,961
- 2021SDEdit: Guided Image Synthesis and Editing with Stochastic Differential Equationscited by 1,903
22 more
- 2023Consistency Modelscited by 1,778
- 2023Pick-a-Pic: An Open Dataset of User Preferences for Text-to-Image Generationcited by 950
- 2022DPM-Solver++: Fast Solver for Guided Sampling of Diffusion Probabilistic Modelscited by 924
- 2022Fast Sampling of Diffusion Models with Exponential Integratorcited by 526
- 2023UniPC: A Unified Predictor-Corrector Framework for Fast Sampling of Diffusion Modelscited by 441
- 2023DeepCache: Accelerating Diffusion Models for Freecited by 346
- 2024SDXL-Lightning: Progressive Adversarial Diffusion Distillationcited by 258
- 2025Z-Image: An Efficient Image Generation Foundation Model with Single-Stream Diffusion Transformercited by 247
- 2024Timestep Embedding Tells: It's Time to Cache for Video Diffusion Modelcited by 245
- 2021Gotta Go Fast When Generating Data with Score-Based Modelscited by 218
- 2024FlowEdit: Inversion-Free Text-Based Editing Using Pre-Trained Flow Modelscited by 214
- 2024PeRFlow: Piecewise Rectified Flow as Universal Plug-and-Play Acceleratorcited by 95
- 2025Lumina-Image 2.0: A Unified and Efficient Image Generative Frameworkcited by 93
- 2024Align Your Steps: Optimizing Sampling Schedules in Diffusion Modelscited by 91
- 2023AutoDiffusion: Training-Free Optimization of Time Steps and Architectures for Automated Diffusion Model Accelerationcited by 70
- 2025DiCache: Let Diffusion Model Determine Its Own Cachecited by 34
- 2025Unified Continuous Generative Modelscited by 19
- 2025LeMiCa: Lexicographic Minimax Path Caching for Efficient Diffusion-Based Video Generationcited by 9
- 2024AdjointDEIS: Efficient Gradients for Diffusion Modelscited by 7
- 2026MeanCache: From Instantaneous to Average Velocity for Accelerating Flow Matching Inferencecited by 7
- 2025Optimizing for the Shortest Path in Denoising Diffusion Modelcited by 6
- 2025Fuzzy Reasoning Chain (FRC): An Innovative Reasoning Framework from Fuzziness to Claritycited by 2
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026Plan First, Diffuse Later: Extrinsic Graph Guidance for Long-Horizon Diffusion Planning
- 2025Path Planning for Masked Diffusion Model Sampling
- 2026Formalizing the Sampling Design Space of Diffusion-Based Generative Models via Adaptive Solvers and Wasserstein-Bounded Timesteps
- 2025Planner Aware Path Learning in Diffusion Language Models Training
- 2024Tracing the Roots: Leveraging Temporal Dynamics in Diffusion Trajectories for Origin Attribution
- 2026DiRecT: Safe Diffusion-Based Planning via Receding-Horizon Denoising
- 2025Monte Carlo Tree Diffusion for System 2 Planning
- 2025Optimal Stepsize for Diffusion Sampling