English

Compositional Monte Carlo Tree Diffusion for Extendable Planning

Machine Learning 2026-01-07 v2

Abstract

Monte Carlo Tree Diffusion (MCTD) integrates diffusion models with structured tree search to enable effective trajectory exploration through stepwise reasoning. However, MCTD remains fundamentally limited by training trajectory lengths. While periodic replanning allows plan concatenation for longer plan generation, the planning process remains locally confined, as MCTD searches within individual trajectories without access to global context. We propose Compositional Monte Carlo Tree Diffusion (C-MCTD), a framework that elevates planning from individual trajectory optimization to reasoning over complete plan compositions. C-MCTD introduces three complementary components: (1) Online Composer, which performs globally-aware planning by searching across entire plan compositions; (2) Distributed Composer, which reduces search complexity through parallel exploration from multiple starting points; and (3) Preplan Composer, which accelerates inference by leveraging cached plan graphs.

Keywords

Cite

@article{arxiv.2510.21361,
  title  = {Compositional Monte Carlo Tree Diffusion for Extendable Planning},
  author = {Jaesik Yoon and Hyeonseo Cho and Sungjin Ahn},
  journal= {arXiv preprint arXiv:2510.21361},
  year   = {2026}
}

Comments

24 pages, 4 figures, NeurIPS 25 Spotlight

R2 v1 2026-07-01T07:03:46.641Z