Refining Compositional Diffusion for Reliable Long-Horizon Planning
- URL: http://arxiv.org/abs/2605.03075v1
- Date: Mon, 04 May 2026 18:44:19 GMT
- Title: Refining Compositional Diffusion for Reliable Long-Horizon Planning
- Abstract summary: We propose a training-free guidance method that steers compositional sampling toward high-density, globally coherent plans.<n> Experiments on challenging long-horizon tasks from OGBench, including locomotion, object manipulation, and pixel-based observations, demonstrate that RCD consistently outperforms existing methods.
- Score: 31.40044829548484
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: Compositional diffusion planning generates long-horizon trajectories by stitching together overlapping short-horizon segments through score composition. However, when local plan distributions are multimodal, existing compositional methods suffer from mode-averaging, where averaging incompatible local modes leads to plans that are neither locally feasible nor globally coherent. We propose Refining Compositional Diffusion (RCD), a training-free guidance method that steers compositional sampling toward high-density, globally coherent plans. RCD leverages the self-reconstruction error of a pretrained diffusion model as a proxy for the log-density of composed plans, combined with an overlap consistency term that enforces consistency at segment boundaries. We show that the combined guidance concentrates sampling on high-density plans that mitigate mode-averaging. Experiments on challenging long-horizon tasks from OGBench, including locomotion, object manipulation, and pixel-based observations, demonstrate that RCD consistently outperforms existing methods.
Related papers
- Geometry-Aware Dataset Condensation for Diffusion Model Training [103.45641113998839]
We propose to reformulate real subset selection as a geometry-aware distribution alignment problem.<n>By incorporating one-sided partial optimal transport, our method selectively aligns a compact subset with the full data distribution.<n>Experiments across diffusion variants, subset sizes, image resolutions, and training rounds show that our method achieves superior fidelity and distributional coverage.
arXiv Detail & Related papers (2026-06-04T08:53:58Z) - Coarse-to-Fine Compositional Diffusion for Long-Horizon Planning [37.38989133673186]
Existing methods improve global coherence by repeatedly propagating local consistency signals.<n>We propose Coarse-to-Fine Compositional Diffusion (CoFi), an inference-time sampler that separates global structure formation from local detail refinement.<n>CoFi not only improves both global coherence and local sample quality over prior compositional baselines, but also requires 2-8x fewer denoiser evaluations.
arXiv Detail & Related papers (2026-05-30T18:26:53Z) - Continuous-Time Distribution Matching for Few-Step Diffusion Distillation [57.28746398500951]
We introduce Continuous-Time Distribution Matching (CDM), migrating the DMD framework from discrete anchoring to continuous optimization.<n>CDM achieves this through two continuous-time designs. First, we replace the fixed discrete schedule with a dynamic continuous schedule of random length.<n>Second, we propose a continuous-time alignment objective that performs active off-trajectory matching on latents extrapolated via the student's velocity field.
arXiv Detail & Related papers (2026-05-07T14:56:39Z) - Compositional Visual Planning via Inference-Time Diffusion Scaling [41.86377754505579]
Diffusion models excel at short-horizon robot planning, yet scaling them to long-horizon tasks remains challenging due to computational constraints and limited training data.<n>We propose that the key to stable compositional generation lies in enforcing boundary agreement on the estimated clean data.<n>Our method formulates long-horizon planning as inference over a chain-structured factor graph of overlapping video chunks, where pretrained short-horizon video diffusion models provide local priors.
arXiv Detail & Related papers (2026-03-03T06:24:06Z) - GeodesicNVS: Probability Density Geodesic Flow Matching for Novel View Synthesis [54.39598154430305]
We propose a Data-to-Data Flow Matching framework that learns deterministic transformations directly between paired views.<n>PDG-FM constrains flow trajectories using geodesic interpolants derived from probability density metrics of pretrained diffusion models.<n>These results highlight the advantages of incorporating data-dependent geometric regularization into deterministic flow matching for consistent novel view generation.
arXiv Detail & Related papers (2026-03-01T09:30:11Z) - Compositional Diffusion with Guided Search for Long-Horizon Planning [25.850015543620128]
We propose Compositional Diffusion with Guided Search (CDGS) for compositional generative models.<n>Our method explores diverse combinations of local modes through population-based sampling, prunes infeasible candidates using likelihood-based filtering, and enforces global consistency through iterative resampling.<n>The approach generalizes across domains, enabling coherent text-guided panoramic images and long videos through effective local-to-global message passing.
arXiv Detail & Related papers (2025-12-31T22:03:19Z) - Projected Coupled Diffusion for Test-Time Constrained Joint Generation [49.69610867216755]
We propose Projected Coupled Diffusion (PCD), a novel test-time framework for constrained joint generation.<n>PCD introduces a coupled guidance term into the generative dynamics to encourage coordination between diffusion models and incorporates a projection step at each diffusion step to enforce hard constraints.<n>Our results show improved coupling effects and guaranteed constraint satisfaction without incurring excessive computational costs.
arXiv Detail & Related papers (2025-08-14T11:05:31Z) - CHD: Coupled Hierarchical Diffusion for Long-Horizon Tasks [10.13048343565914]
Diffusion-based planners have shown strong performance in short-horizon tasks but often fail in complex, long-horizon settings.<n>We propose Coupled Hierarchical Diffusion, a framework that models HL sub-goals and LL trajectories jointly within a unified diffusion process.<n> Experiments across maze navigation, tabletop manipulation, and household environments show that CHD consistently outperforms both flat and hierarchical diffusion baselines.
arXiv Detail & Related papers (2025-05-12T06:21:48Z) - Extendable Long-Horizon Planning via Hierarchical Multiscale Diffusion [62.91968752955649]
This paper tackles a novel problem, extendable long-horizon planning-enabling agents to plan trajectories longer than those in training data without compounding errors.<n>We propose an augmentation method that iteratively generates longer trajectories by stitching shorter ones.<n>HM-Diffuser trains on these extended trajectories using a hierarchical structure, efficiently handling tasks across multiple temporal scales.
arXiv Detail & Related papers (2025-03-25T22:52:46Z) - Generative Trajectory Stitching through Diffusion Composition [29.997765496994457]
CompDiffuser is a novel generative approach that can solve new tasks by learning to compositionally stitch together shorter trajectory chunks from previously seen tasks.<n>We conduct experiments on benchmark tasks of various difficulties, covering different environment sizes, agent state dimension, trajectory types, training data quality, and show that CompDiffuser significantly outperforms existing methods.
arXiv Detail & Related papers (2025-03-07T05:22:52Z) - Planning as In-Painting: A Diffusion-Based Embodied Task Planning
Framework for Environments under Uncertainty [56.30846158280031]
Task planning for embodied AI has been one of the most challenging problems.
We propose a task-agnostic method named 'planning as in-painting'
The proposed framework achieves promising performances in various embodied AI tasks.
arXiv Detail & Related papers (2023-12-02T10:07:17Z) - Compositional Diffusion-Based Continuous Constraint Solvers [98.1702285470628]
This paper introduces an approach for learning to solve continuous constraint satisfaction problems (CCSP) in robotic reasoning and planning.
By contrast, our model, the compositional diffusion continuous constraint solver (Diffusion-CCSP), derives global solutions to CCSPs.
arXiv Detail & Related papers (2023-09-02T15:20:36Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.