Learning Multiple Timescales for Goal-Conditioned Reinforcement Learning
cs.AI, stat.ML
Submitted: 2026-10-01
Updated: 2026-10-01
Code: https://github.com/pedroroblesduten/GITA
License: http://creativecommons.org/licenses/by/4.0/
The gist: Existing approaches to offline goal-conditioned reinforcement learning (GCRL) struggle with long-horizon tasks.
Terminology
Abstract
Existing approaches to offline goal-conditioned reinforcement learning (GCRL) struggle with long-horizon tasks. Discounting shrinks value differences between distant states until they fall below the function approximation error, leaving the agent with no signal for ranking states. Temporal abstraction, which treats k environment steps as a single transition, restores this signal at long range, but no single fixed k suits all state-goal distances: large k preserves value differences across long temporal distances while collapsing distinctions between nearby states, and small k does the reverse. We make this trade-off explicit and introduce Generalized Implicit Temporal Abstraction (GITA), which conditions a single value function on k. GITA trains one policy by aggregating advantage-weighted supervision across multiple k values, so scales assigning larger positive advantages to a state-goal pair contribute more strongly to its update. GITA does not need to choose between local resolution and long-range signal; it retains both without committing to a single k. On OGBench, GITA outperforms a broad range of offline GCRL baselines, raising average success rate across all tasks by 25 percentage points (73% relative improvement) over HIQL. It also improves over the strongest fixed-k method, OTA, by 7 percentage points (14% relative).
Sources
- MSPR: Multi-scale Predictive Representations for Goal-conditioned Reinforcement Learning
- Layer Normalization
- Gaussian Error Linear Units (GELUs)
- Recursive Value Learning for Long-Horizon Offline Goal-Conditioned RL
- Adaptive Coarse-to-Fine Subgoal Refinement for Long-Horizon Offline Goal-Conditioned Reinforcement Learning
- Offline Reinforcement Learning: Tutorial, Review, and Perspectives on Open Problems
- Advantage-Weighted Regression: Simple and Scalable Off-Policy Reinforcement Learning
- Abstraction for Offline Goal-Conditioned Reinforcement Learning
Related papers
- MAVEN-T: Reinforced Heterogeneous Distillation for Real-Time Multi-Agent Trajectory Prediction
- Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models
- Agents' Last Exam
- The Clinician's Veto: Navigating Trust, Liability, and Uncertainty in Autonomous AI Prescribing
- MindHelper: Closed-Loop Embodied Mental-State Reasoning for Precision Intervention
- Incumbent Advantage: Brand Bias and Cognitive Manipulation Dynamics in LLM Recommendation Systems