This post introduces the paper "Decoupling Exploration and Policy Optimization: Uncertainty Guided Tree Search" (arXiv: 2603.22273) by Zakaria Mhammedi and James Cohan.
Problem
Discovery requires active exploration—collecting new, informative data—but efficient autonomous exploration remains a major unsolved problem. The dominant paradigm uses Reinforcement Learning (RL) to train intrinsically motivated agents that maximize a composite objective of extrinsic and intrinsic rewards.
Key Idea
The authors argue this approach incurs unnecessary overhead: while policy optimization is essential for precise task execution, using such machinery solely to expand state coverage may be inefficient. They propose a new paradigm that:
- Explicitly separates exploration from exploitation
- Bypasses RL entirely during the exploration phase
- Uses a tree-search strategy inspired by the Go-With-The-Winner algorithm
- Leverages epistemic uncertainty measures to systematically drive exploration
- By eliminating policy optimization overhead, the method is an order of magnitude more efficient than standard intrinsic-motivation baselines on hard Atari benchmarks.
- Discovered trajectories can be distilled into deployable policies using existing supervised reverse learning algorithms.
- Achieves state-of-the-art results with significant margins on Montezuma's Revenge, Pitfall!, and Venture—without domain-specific knowledge.
- arXiv: https://arxiv.org/abs/2603.22273