Tree-Guided Diffusion Planner

Figure 1: TDP solves a variety of zero-shot planning problems.

Tree-Guided Diffusion Planner
Hyeonseong Jeon, Cheolhong Min, Jaesik Park
NeurIPS 2025
Project page / arXiv / BibTex

Overview

This repository is organized into three separate sub-branches for each task:

• Maze2D Gold Picking: maze2d branch
• KUKA Robot Arm manipulation: kuka branch
• AntMaze Multi-goal Exploration: antmaze branch

Citation

If you find TDP helpful, please consider citing:

@article{jeon2026tree,
  title={Tree-Guided Diffusion Planner},
  author={Jeon, Hyeonseong and Min, Cheolhong and Park, Jaesik},
  journal={Advances in Neural Information Processing Systems},
  volume={38},
  pages={67829--67855},
  year={2026}
}

Acknowledgement

This codebase is built upon the foundations of Diffuser, TAT and CleanDiffuser.