English

Data-Efficient Hierarchical Goal-Conditioned Reinforcement Learning via Normalizing Flows

Robotics 2026-02-12 v1 Artificial Intelligence Machine Learning

Abstract

Hierarchical goal-conditioned reinforcement learning (H-GCRL) provides a powerful framework for tackling complex, long-horizon tasks by decomposing them into structured subgoals. However, its practical adoption is hindered by poor data efficiency and limited policy expressivity, especially in offline or data-scarce regimes. In this work, Normalizing flow-based hierarchical implicit Q-learning (NF-HIQL), a novel framework that replaces unimodal gaussian policies with expressive normalizing flow policies at both the high- and low-levels of the hierarchy is introduced. This design enables tractable log-likelihood computation, efficient sampling, and the ability to model rich multimodal behaviors. New theoretical guarantees are derived, including explicit KL-divergence bounds for Real-valued non-volume preserving (RealNVP) policies and PAC-style sample efficiency results, showing that NF-HIQL preserves stability while improving generalization. Empirically, NF-HIQL is evaluted across diverse long-horizon tasks in locomotion, ball-dribbling, and multi-step manipulation from OGBench. NF-HIQL consistently outperforms prior goal-conditioned and hierarchical baselines, demonstrating superior robustness under limited data and highlighting the potential of flow-based architectures for scalable, data-efficient hierarchical reinforcement learning.

Keywords

Cite

@article{arxiv.2602.11142,
  title  = {Data-Efficient Hierarchical Goal-Conditioned Reinforcement Learning via Normalizing Flows},
  author = {Shaswat Garg and Matin Moezzi and Brandon Da Silva},
  journal= {arXiv preprint arXiv:2602.11142},
  year   = {2026}
}

Comments

9 pages, 3 figures, IEEE International Conference on Robotics and Automation 2026

R2 v1 2026-07-01T10:32:21.253Z