Zero-shot RL stops losing the goal in a dense task code: one sparsity penalty lifts DMC average return by 53 points
Adding a sparsity penalty to TD-JEPA's task encoder concentrates the inferred reward near the goal, raising average return on DMC by 53 points and average success rate on OGBench by 10.4 percentage points in zero-shot reinforcement learning.