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
Figure 1: Path mirage in offline GCRL. (a) Offline trajectories can pass through branching regions where future outcomes diverge toward success, failure, or risky areas. (b) Prior methods can be misled by deceptively successful paths, resulting in path mirages with occasional successes but persistent failures. (c) Our branch-aware graph planning identifies branching points near branching regions as landmarks and selects reliable goal-conditioned routes that avoid risky branches.
Figure 2: Visualization of uncertainty scores and the resulting landmarks on two OGBench tasks. High-uncertainty regions correspond to significant trajectory branching, leading to denser landmark selection.
Figure 3: (a) Illustration of downstream branches R(vj) . (b)-(c) Edge comparison on a teleport task in OGBench, where vj (near the teleport) induces stochastic branching. (b) A prior edge e(vi,vj) favors the transition due to local proximity, whereas (c) our goal-conditioned edge eg evaluates all branches {vk1,vk2,vk3}∈R(vj) and avoids selecting vj since one branch ( vk2 ) leads to a dead zone, i.e., d(sk2,g)=∞ for sk2∈Ck2 .
Figure 4
Figure 5: Visualization of OGBench tasks. (a)–(c) show three maze layouts: large , giant , and teleport . In the teleport layout, two blackholes teleport the agent to one of three whiteholes uniformly at random. (d) and (e) show the pointmaze and antmaze tasks, respectively.
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)
Table 1: Performance comparison on OGBench. We report the average success rate (%) with standard deviation ( ± ) over 5 seeds. The best results in each row are highlighted in bold .
Figure 6: Shortest waypoint paths in the teleport layout. (a–b) use the prior edge cost ep(v,w)=−Vθ(v,w) and illustrate failure cases, while (c–d) use the proposed goal-conditioned edge cost eg . Goals are selected from the evaluation set. (c–d) further illustrate how eg accounts for downstream branches and avoids unreliable teleport transitions, yielding more stable paths.
Figure 7: Environments and success rates with respect to the number of teleports B . Panels (a) and (b) show the environments with two and four teleports, respectively. Panel (c) shows the success rates of HIQL, BGP (w/o segmented learning), and BGP on the navigate datasets, and panel (d) shows those on the stitch datasets. Shaded regions denote 95% confidence intervals over 5 seeds.
Figure 9
Appendix figures & tables16 assets
Supplementary material from the paper’s appendix.
Appendix
Figure 9: Autoregressive prediction of the Transformer during evaluation, where predicted transition variables y^t+l are fed back as inputs to estimate multi-step uncertainty from the current state st .
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
Appendix
Table 3: Representations used for landmark clustering in each environment. Dimension denotes the dimensionality of the representation space Z .
Figure 10: Illustration of segmented learning. Each offline trajectory is partitioned into sub-trajectories τj(k) at cluster transition boundaries, shown in different colors, and M consecutive sub-trajectories are concatenated into a local training window τ~j(k) .
Figure 11: Visualizations of the OGBench environments used in our experiments. Panels (a)-(e) show the agents and the manipulation environment, panels (f)-(i) show the maze layouts, and panels (j)-(l) show example trajectories of the three dataset types.
Hyperparameter
Value
Transformer (Uncertainty Estimation)
Number of layers
4
Number of attention heads
2
Embedding dimension
256
Context length ( n )
5
Transformer learning rate
3×10−4
Appendix
Table 4: Common hyperparameters.
Domain
Dataset
Percentile ( α )
Num Clusters ( N )
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
Appendix
Table 5: Environment-specific hyperparameters.
Figure 12: antmaze-teleport environments with varying numbers of teleports. Panels (a)-(e) show the environments with B=0 , 1 , 2 , 3 , and 4 active blackholes, respectively, and panel (c) is the original OGBench environment.
Env
Dataset
HIQL
GCIQL
QRL
CRL
BGP(Ours)
BGP (α,N)
humanoidmaze
medium-navigate
89 ( ± 2)
27 ( ± 2)
21 ( ± 8)
60 ( ± 4)
95 ( ± 2)
(10,50)
medium-stitch
88 ( ± 2)
12 ( ± 3)
18 ( ± 2)
36 ( ± 2)
87 ( ± 6)
(10,50)
large-navigate
49 ( ± 4)
2 ( ± 1)
5 ( ± 1)
24 ( ± 4)
70 ( ± 7)
(20,50)
large-stitch
28 ( ± 3)
0 ( ± 0)
3 ( ± 1)
4 ( ± 1)
58 ( ± 8)
(50,50)
giant-navigate
12 ( ± 4)
0 ( ± 0)
1 ( ± 0)
3 ( ± 2)
75 ( ± 6)
(10,150)
giant-stitch
3 ( ± 2)
0 ( ± 0)
0 ( ± 0)
0 ( ± 0)
69 ( ± 8)
(10,150)
Appendix
Table 6: Success rates (%) on humanoidmaze and scene-play-v0 . The highest mean in each row is highlighted in bold . The final column reports the BGP hyperparameter settings (α,N) , with – indicating an unreported setting.
Figure 13: Representative trajectory analysis on the teleport environment.
Figure 14: Representative trajectory analysis on the giant environment.
Figure 15: Representative trajectory analysis on the large environment.
Figure 16: Hyperparameter sensitivity on the teleport environment.
Figure 17: Hyperparameter sensitivity on the giant environment.
Backbone
M=2
M=3
M=5
M=10 (excl. stitch)
HIQL
90.0±2.0
90.2±1.1
89.0±3.2
87.1±4.0
GCIQL
73.5±15.6
79.6±11.4
72.9±16.4
48.5±21.7
Appendix
Table 7: Sensitivity to the segment length ( M ). Results are averaged over five random seeds. The M=10 case excludes stitch datasets, as their inherent trajectory lengths are too short to support this segment length.
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]
Appendix
Table 8: Budget-aware evaluation on antmaze-teleport-navigate . We report the expected success rate (%) and 95% confidence interval (CI) as a function of the number of evaluation episodes, following Jackson et al. [55] .
Method
BGP
HIQL
GCIQL
CGCIVL
TMD
GAS
Time (s)
7563
4686
3632
4927
17577
8841
Appendix
Table 9: Simulation time across different methods.
Department of Electrical and Computer Engineering, Seoul National University, Seoul, South Korea · Department of Automotive Engineering, Ajou University, Gyeonggi-do, South Korea