English

CDE: Concept-Driven Exploration for Reinforcement Learning

Robotics 2026-03-10 v2

Abstract

Intelligent exploration remains a critical challenge in reinforcement learning (RL), especially in visual control tasks. Unlike low-dimensional state-based RL, visual RL must extract task-relevant structure from raw pixels, making exploration inefficient. We propose Concept-Driven Exploration (CDE), which leverages a pre-trained vision-language model (VLM) to generate object-centric visual concepts from textual task descriptions as weak, potentially noisy supervisory signals. Rather than directly conditioning on these noisy signals, CDE trains a policy to reconstruct the concepts via an auxiliary objective, learning general representations of the concepts and using reconstruction accuracy as an intrinsic reward to guide exploration toward task-relevant objects. Across five challenging simulated visual manipulation tasks, CDE achieves efficient, targeted exploration and remains robust to both synthetic errors and noisy VLM predictions. Finally, we demonstrate real-world transfer by deploying CDE on a Franka arm, attaining an 80\% success rate in a real-world manipulation task.

Keywords

Cite

@article{arxiv.2510.08851,
  title  = {CDE: Concept-Driven Exploration for Reinforcement Learning},
  author = {Le Mao and Andrew H. Liu and Renos Zabounidis and Yanan Niu and Zachary Kingston and Joseph Campbell},
  journal= {arXiv preprint arXiv:2510.08851},
  year   = {2026}
}

Comments

Preprint

R2 v1 2026-07-01T06:28:21.657Z