Escaping Path Mirages in Offline Goal-Conditioned Reinforcement Learning
Organizations: Graduate School of Artificial Intelligence, UNIST, Ulsan, South Korea
Abstract
Offline goal-conditioned reinforcement learning remains challenging in stochastic long-horizon settings, where compounding value estimation errors hinder reliable goal reaching. While prior methods have sought to address this challenge through various approaches, a key challenge remains path mirage, where agents overcommit to spuriously successful trajectories in offline datasets induced by stochasticity, often leading to failure on difficult tasks. To address this issue, we propose Branch-aware Graph Planning (BGP), which captures stochastic branching structures by identifying high-variability points as graph nodes and designing goal-conditioned edges with segmented learning to avoid unreliable branches and reduce unnecessary stochasticity along paths. As a result, BGP enables reliable goal reaching even in highly stochastic long-horizon environments, and experiments on diverse OGBench tasks show that it substantially outperforms prior state-of-the-art offline goal-conditioned RL methods. Our implementation is available at https://umlic.github.io/papers/BGP/.
Figures & tables
| Env | Dataset | HIQL | GCIQL | QRL | CRL | TMD | CGCIVL | GAS | BGP(Ours) |
|---|---|---|---|---|---|---|---|---|---|
| antmaze | large-navigate | 91 ( 2) | 34 ( 4) | 75 ( 6) | 83 ( 4) | 82 ( 2) | 91 ( 2) | 93 ( 1) | 90 ( 1) |
| large-stitch | 67 ( 5) | 7 ( 2) | 18 ( 2) | 11 ( 2) | 37 ( 3) | 79 ( 3) | 96 ( 1) | 93 ( 1) | |
| large-explore | 4 ( 5) | 0 ( 0) | 0 ( 0) | 0 ( 0) | 0 ( 0) | 6 ( 5) | 94 ( 3) | 97 ( 3) | |
| giant-navigate | 65 ( 5) | 0 ( 0) | 14 ( 3) | 16 ( 3) | 1 ( 1) | 73 ( 5) | 78 ( 3) | 88 ( 2) | |
| giant-stitch | 2 ( 2) | 0 ( 0) | 0 ( 0) | 0 ( 0) | 3 ( 1) | 36 ( 7) | 88 ( 4) | 88 ( 2) | |
| teleport-navigate | 42 ( 3) | 35 ( 5) | 35 ( 5) | 53 ( 2) | 27 ( 12) | 29 ( 13) | 39 ( 3) | 90 ( 1) |
Appendix figures & tables16 assets
Supplementary material from the paper’s appendix.
Appendix
| Environment | Representation for clustering | Dimension |
|---|---|---|
| antmaze | PCA projection | 2 |
| pointmaze | PCA projection | 2 |
| visual-antmaze | Hilbert representation | 16 |
| humanoidmaze | Hilbert representation | 16 |
| manipulation tasks | Hilbert representation | 16 |
| Hyperparameter | Value |
|---|---|
| Transformer (Uncertainty Estimation) | |
| Number of layers | 4 |
| Number of attention heads | 2 |
| Embedding dimension | 256 |
| Context length ( ) | 5 |
| Transformer learning rate | |
| Domain | Dataset | Percentile ( ) | Num Clusters ( ) |
| pointmaze | teleport-navigate | 1 | 100 |
| teleport-stitch | 1 | 100 | |
| giant-navigate | 10 | 100 | |
| giant-stitch | 10 | 100 | |
| large-navigate | 1 | 50 | |
| large-stitch | 1 | 50 |
| Env | Dataset | HIQL | GCIQL | QRL | CRL | BGP(Ours) | BGP |
|---|---|---|---|---|---|---|---|
| humanoidmaze | medium-navigate | 89 ( 2) | 27 ( 2) | 21 ( 8) | 60 ( 4) | 95 ( 2) | |
| medium-stitch | 88 ( 2) | 12 ( 3) | 18 ( 2) | 36 ( 2) | 87 ( 6) | ||
| large-navigate | 49 ( 4) | 2 ( 1) | 5 ( 1) | 24 ( 4) | 70 ( 7) | ||
| large-stitch | 28 ( 3) | 0 ( 0) | 3 ( 1) | 4 ( 1) | 58 ( 8) | ||
| giant-navigate | 12 ( 4) | 0 ( 0) | 1 ( 0) | 3 ( 2) | 75 ( 6) | ||
| giant-stitch | 3 ( 2) | 0 ( 0) | 0 ( 0) | 0 ( 0) | 69 ( 8) |
| Backbone | (excl. stitch) | |||
|---|---|---|---|---|
| HIQL | ||||
| GCIQL |
| Evaluation episodes | 1 | 5 | 13 | 35 | 100 | 200 |
|---|---|---|---|---|---|---|
| Success rate (%) | 65 | 71 | 76 | 80 | 83 | 84 |
| 95% CI | [12, 88] | [30, 88] | [50, 88] | [62, 88] | [63, 88] | [67, 88] |
| Method | BGP | HIQL | GCIQL | CGCIVL | TMD | GAS |
|---|---|---|---|---|---|---|
| Time (s) | 7563 | 4686 | 3632 | 4927 | 17577 | 8841 |
Offline RL
Offline reinforcement learning learns decision-making policies from fixed datasets without interacting with the environment during training.
See all 304 papers