Extendable Planning via Multiscale Diffusion
Chang Chen, Hany Hamed, Doojin Baek, Taegu Kang, Samyeul Noh, Yoshua Bengio, Sungjin Ahn
Abstract
Long-horizon planning is crucial in complex environments, but diffusion-based planners like Diffuser are limited by the trajectory lengths observed during training. This creates a dilemma: long trajectories are needed for effective planning, yet they degrade model performance. In this paper, we introduce this extendable long-horizon planning challenge and propose a two-phase solution. First, Progressive Trajectory Extension incrementally constructs longer trajectories through multi-round compositional stitching. Second, the Hierarchical Multiscale Diffuser enables efficient training and inference over long horizons by reasoning across temporal scales. To avoid the need for multiple separate models, we propose Adaptive Plan Pondering and the Recursive HM-Diffuser, which unify hierarchical planning within a single model. Experiments show our approach yields strong performance gains, advancing scalable and efficient decision-making over long-horizons.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext ada68226-4c4c-475e-8a0f-705bdd2e80ffCited by top-tier papers3
- Compositional Monte Carlo Tree Diffusion for Extendable PlanningJaesik Yoon, Hyeonseo Cho, Sungjin AhnNeurIPS 2025 · 2 citations
- Multi-scale Temporal Prediction via Incremental Generation and Multi-agent CollaborationZhitao Zeng, Guojian Yuan, Junyuan Mao, Yuxuan Wang et al.NeurIPS 2025
- Fast Monte Carlo Tree Diffusion: 100× Speedup via Parallel and Sparse PlanningJaesik Yoon, Hyeonseo Cho, Yoshua Bengio, Sungjin AhnNeurIPS 2025
Builds on22
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Diffusion Models Beat GANs on Image SynthesisPrafulla Dhariwal, Alexander Quinn NicholNeurIPS 2021 · 13,211 citations
- Denoising Diffusion Implicit ModelsJiaming Song, Chenlin Meng, Stefano ErmonICLR 2021 · 11,743 citations
- Planning with Diffusion for Flexible Behavior SynthesisMichael Janner, Yilun Du, Joshua B. Tenenbaum, Sergey LevineICML 2022 · 1,115 citations
- Model Based Reinforcement Learning for AtariLukasz Kaiser, Mohammad Babaeizadeh, Piotr Milos, Blazej Osinski et al.ICLR 2020 · 969 citations
Related papers
- Generative Trajectory Stitching through Diffusion CompositionYunhao Luo, Utkarsh A. Mishra, Yilun Du, Danfei XuNeurIPS 2025 · 48 citations
- Simple Hierarchical Planning with DiffusionChang Chen, Fei Deng, Kenji Kawaguchi, Caglar Gulcehre et al.ICLR 2024 · 79 citations
- SkillDiffuser: Interpretable Hierarchical Planning via Skill Abstractions in Diffusion-Based Task ExecutionZhixuan Liang, Yao Mu, Hengbo Ma, Masayoshi Tomizuka et al.CVPR 2024
- Compositional Foundation Models for Hierarchical PlanningAnurag Ajay, Seungwook Han, Yilun Du, Shuang Li et al.NeurIPS 2023 · 137 citations
- Energy-based Compositional Diffusion PlanningTao Sun, Utkarsh Mishra, Jiaxin Lu, Danfei Xu et al.ICML 2026
