Fast and Precise: Adjusting Planning Horizon with Adaptive Subgoal Search
Michal Zawalski, Michal Tyrolski, Konrad Czechowski, Tomasz Odrzygózdz, Damian Stachura, Piotr Piekos, Yuhuai Wu, Lukasz Kucinski, Piotr Milos
Abstract
Complex reasoning problems contain states that vary in the computational cost required to determine the right action plan. To take advantage of this property, we propose Adaptive Subgoal Search (AdaSubS), a search method that adaptively adjusts the planning horizon. To this end, AdaSubS generates diverse sets of subgoals at different distances. A verification mechanism is employed to filter out unreachable subgoals swiftly, making it possible to focus on feasible further subgoals. In this way, AdaSubS benefits from the efficiency of planning with longerterm subgoals and the fine control with shorter-term ones, and thus scales well to difficult planning problems. We show that AdaSubS significantly surpasses hierarchical planning algorithms on three complex reasoning tasks: Sokoban, the Rubik's Cube, and the inequality-proving benchmark INT.
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 08d4cec5-00a1-4709-9a88-f3723f9cccafCited by top-tier papers9
- Thought Cloning: Learning to Think while Acting by Imitating Human ThinkingShengran Hu, Jeff CluneNeurIPS 2023 · 49 citations
- Double-Ended Synthesis Planning with Goal-Constrained Bidirectional SearchKevin Yu, Jihye Roh, Ziang Li, Wenhao Gao et al.NeurIPS 2024 · 38 citations
- Hierarchical Imitation Learning with Vector Quantized ModelsKalle Kujanpää, Joni Pajarinen, Alexander IlinICML 2023 · 17 citations
- Contrastive Representations for Temporal ReasoningAlicja Ziarko, Michal Bortkiewicz, Michal Zawalski, Benjamin Eysenbach et al.NeurIPS 2025 · 8 citations
- Flattening Hierarchies with Policy BootstrappingJohn L. Zhou, Jonathan C. KaoNeurIPS 2025 · 8 citations
Builds on7
- Hierarchical Foresight: Self-Supervised Learning of Long-Horizon Tasks via Visual Subgoal GenerationSuraj Nair, Chelsea FinnICLR 2020 · 152 citations
- Long-Horizon Visual Planning with Goal-Conditioned Hierarchical PredictorsKarl Pertsch, Oleh Rybkin, Frederik Ebert, Shenghao Zhou et al.NeurIPS 2020 · 96 citations
- World Model as a Graph: Learning Latent Landmarks for PlanningLunjun Zhang, Ge Yang, Bradly C. StadieICML 2021 · 90 citations
- INT: An Inequality Benchmark for Evaluating Generalization in Theorem ProvingYuhuai Wu, Albert Q. Jiang, Jimmy Ba, Roger Baker GrosseICLR 2021 · 60 citations
- Hallucinative Topological Memory for Zero-Shot Visual PlanningKara Liu, Thanard Kurutach, Christine Tung, Pieter Abbeel et al.ICML 2020 · 49 citations
Related papers
- Subgoal Search For Complex Reasoning TasksKonrad Czechowski, Tomasz Odrzygózdz, Marek Zbysinski, Michal Zawalski et al.NeurIPS 2021 · 41 citations
- AdaMix: Adaptive Mixing for Short and Long Reasoning AdaptersHao Luo, Xiao Yan, Xinyan Li, Qiming Zeng et al.ACL 2026
- Beyond Autoregression: Discrete Diffusion for Complex Reasoning and PlanningJiacheng Ye, Jiahui Gao, Shansan Gong, Lin Zheng et al.ICLR 2025
- Re-FORC: Adaptive Reward Prediction for Efficient Chain-of-Thought ReasoningRenos Zabounidis, Aditya Golatkar, Michael Kleinman, Alessandro Achille et al.ICML 2026 · 4 citations
- What If We Allocate Test-Time Compute Adaptively?Ahsan Bilal, Muhammad Ahmed Mohsin, Muhammad Umer, Ali Subhan et al.ICML 2026 · 3 citations
