Hierarchical Entity-Centric Reinforcement Learning (HECRL)

Official PyTorch code release of the paper "Hierarchical Entity-centric Reinforcement Learning with Factored Subgoal Diffusion" by Dan Haramati, Carl Qi, Tal Daniel, Amy Zhang, Aviv Tamar and George Konidaris.

This repository is pytorch-based and incorporates torch.compile acceleration.

Hierarchical Entity-centric Reinforcement Learning with Factored Subgoal Diffusion

Dan Haramati • Carl Qi • Tal Daniel • Amy Zhang • Aviv Tamar • George Konidaris

ICLR 2026

World Modeling Workshop at Mila 2026 - Oral (top 7%)

Code coming very soon