Efficient Hierarchical Implicit Flow Q-learning for Offline Goal-conditioned Reinforcement Learning
Organizations: School of Software, Shandong University, Jinan, Shandong, China · Luke EI, Jinan, China
Abstract
Offline goal-conditioned reinforcement learning (GCRL) is a practical reinforcement learning paradigm that aims to learn goal-conditioned policies from reward-free offline data. Despite recent advances in hierarchical architectures such as HIQL, long-horizon control in offline GCRL remains challenging due to the limited expressiveness of Gaussian policies and the inability of high-level policies to generate effective subgoals. To address these limitations, we propose the goal-conditioned mean flow policy, which introduces an average velocity field into hierarchical policy modeling for offline GCRL. Specifically, the mean flow policy captures complex target distributions for both high-level and low-level policies through a learned average velocity field, enabling efficient action generation via one-step sampling. Furthermore, considering the insufficiency of goal representation, we introduce a LeJEPA loss that repels goal representation embeddings during training, thereby encouraging more discriminative representations and improving generalization. Experimental results show that our method achieves strong performance across both state-based and pixel-based tasks in the OGBench benchmark.
Figures & tables
| Environment | Dataset | GCBC | GCIVL | GCIQL | QRL | CRL | HIQL | HIFQL (ours) |
|---|---|---|---|---|---|---|---|---|
| pointmaze | pointmaze-medium-navigate-v0 | 99.5 0.9 | ||||||
| pointmaze-large-navigate-v0 | ||||||||
| pointmaze-giant-navigate-v0 | ||||||||
| pointmaze-teleport-navigate-v0 | ||||||||
| pointmaze-medium-stitch-v0 | ||||||||
| pointmaze-large-stitch-v0 |
| Environment | Dataset | GCBC | GCIVL | GCIQL | QRL | CRL | HIQL | HIFQL (ours) |
|---|---|---|---|---|---|---|---|---|
| visual-antmaze | visual-antmaze-medium-navigate-v0 | |||||||
| visual-antmaze-large-navigate-v0 | ||||||||
| visual-antmaze-giant-navigate-v0 | ||||||||
| visual-antmaze-teleport-navigate-v0 |
Offline RL
Offline reinforcement learning learns decision-making policies from fixed datasets without interacting with the environment during training.
See all 298 papers