Under review at ICLR 2027, NeurIPS 2026 Workshop (PTA)
Asymmetric On-Policy Distillation Learns In-Context Exploration
Sequential decision-making under partial observability is difficult because agents must explore to infer hidden, task-relevant information before acting effectively. For robots with limited sensing, e.g., without cameras or under occlusion, this information often can only be gathered through contact. Learning such information-gathering behaviors tabula rasa via online reinforcement learning is sample-inefficient, whereas offline imitation will aggregate actions and fail to learn exploration. We propose ASTEROID, a simple student-teacher supervised learning framework that can learn test-time exploration. Although a privileged expert never needs to explore, we demonstrate that a history-conditioned student can learn to do so when trained with on-policy contexts: the student first collects on-policy context under partial observability, and the privileged expert provides action labels conditioned on that context. By distilling expert actions directly conditioned on the student’s on-policy history, the student learns to systematically resolve partial observability: search while uncertain and exploit once the hidden state is revealed. On simulated manipulation with only touch or only a wrist camera, ASTEROID uses ~100× fewer interactions than RL baselines and outperforms frontier models at exploration. Finally, it transfers policies trained in simulation to vision-denied real-world robot manipulation, exhibiting efficient exploration and increasing success by +40% over standard distillation.