Diffusion Subgoal Planning for Long-Horizon Offline Goal-Conditioned Reinforcement Learning
cs.AI
Submitted: 2026-09-28
Updated: 2026-10-20
Comments: 31 pages, 9 figures
Code: https://github.com/Henry0132/DSP
Project page: https://henry0132.github.io/DSP
License: http://creativecommons.org/licenses/by/4.0/
The gist: Offline goal-conditioned reinforcement learning (GCRL) learns goal-directed policies from reward-free data, but in long-horizon tasks, goal-conditioned value functions often provide unstable guidance
Terminology
Abstract
Offline goal-conditioned reinforcement learning (GCRL) learns goal-directed policies from reward-free data, but in long-horizon tasks, goal-conditioned value functions often provide unstable guidance due to sparse rewards and discounting. Hierarchical methods partially mitigate this issue via subgoal decomposition; however, high-level decision-making still relies on noise-sensitive value estimates, leading to unstable behavior in complex environments. We address this limitation by proposing Diffusion Subgoal Planning (DSP), a diffusion-based framework for high-level subgoal generation. DSP casts high-level planning as guided generative inference over goal-conditioned subgoals and learns both conditional and unconditional flows, enabling classifier-free guidance to introduce a goal-directed bias at inference time. By removing explicit value-based guidance from high-level planning, DSP generates reachable and goal-directed subgoals through a generative model while retaining hierarchical execution. Experiments on offline GCRL benchmarks demonstrate that DSP outperforms prior methods on a range of navigation and manipulation tasks, with particularly strong performance in maze environments that require multi-step subgoal planning.
Sources
- Option-aware Temporally Abstracted Value for Offline Goal-Conditioned Reinforcement Learning
- Is Conditional Generative Modeling all you need for Decision-Making?
- Simple Hierarchical Planning with Diffusion
- Score Regularized Policy Optimization through Diffusion Behavior
- Chain-of-Goals Hierarchical Policy for Long-Horizon Offline Goal-Conditioned RL
- Diffusion Guidance Is a Controllable Policy Improvement Operator
- Learning to Reach Goals via Iterated Supervised Learning
- Physics-informed Value Learner for Offline Goal-Conditioned Reinforcement Learning
- IDQL: Implicit Q-Learning as an Actor-Critic Method with Diffusion Policies
- Classifier-Free Diffusion Guidance
- Planning with Diffusion for Flexible Behavior Synthesis
- Prior-Guided Diffusion Planning for Offline Reinforcement Learning
- Offline Goal-Conditioned Reinforcement Learning with Projective Quasimetric Planning
- Offline Reinforcement Learning with Implicit Q-Learning
- Flow Matching for Generative Modeling
- Goal-Conditioned Reinforcement Learning: Problems and Solutions
- SAGAS: Semantic-Aware Graph-Assisted Stitching for Offline Temporal Logic Planning
- Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flow
- Stochastic Interpolants via Conditional Dependent Coupling
- Drift Flow Matching
Related papers
- MAVEN-T: Reinforced Heterogeneous Distillation for Real-Time Multi-Agent Trajectory Prediction
- Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models
- Agents' Last Exam
- The Clinician's Veto: Navigating Trust, Liability, and Uncertainty in Autonomous AI Prescribing
- MindHelper: Closed-Loop Embodied Mental-State Reasoning for Precision Intervention
- Incumbent Advantage: Brand Bias and Cognitive Manipulation Dynamics in LLM Recommendation Systems