BYOL-Explore: Exploration by Bootstrapped Prediction
Zhaohan Guo, Shantanu Thakoor, Miruna Pislar, Bernardo Ávila Pires, Florent Altché, Corentin Tallec, Alaa Saade, Daniele Calandriello, Jean-Bastien Grill, Yunhao Tang, Michal Valko, Rémi Munos
Abstract
We present BYOL-Explore, a conceptually simple yet general approach for curiosity-driven exploration in visually-complex environments. BYOL-Explore learns a world representation, the world dynamics, and an exploration policy alltogether by optimizing a single prediction loss in the latent space with no additional auxiliary objective. We show that BYOL-Explore is effective in DM-HARD-8, a challenging partially-observable continuous-action hard-exploration benchmark with visually-rich 3-D environments. On this benchmark, we solve the majority of the tasks purely through augmenting the extrinsic reward with BYOL-Explore's intrinsic reward, whereas prior work could only get off the ground with human demonstrations. As further evidence of the generality of BYOL-Explore, we show that it achieves superhuman performance on the ten hardest exploration games in Atari while having a much simpler design than other competitive agents. * Equal contribution. Preprint. Under review.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers44
- Video Prediction Models as Rewards for Reinforcement LearningAlejandro Escontrela, Ademi Adeniji, Wilson Yan, Ajay Jain et al.NeurIPS 2023 · 117 citations
- TACO: Temporal Latent Action-Driven Contrastive Loss for Visual Reinforcement LearningRuijie Zheng, Xiyao Wang, Yanchao Sun, Shuang Ma et al.NeurIPS 2023 · 89 citations
- Mask-based Latent Reconstruction for Reinforcement LearningTao Yu, Zhizheng Zhang, Cuiling Lan, Yan Lu et al.NeurIPS 2022 · 80 citations
- DynaMo: In-Domain Dynamics Pretraining for Visuo-Motor ControlZichen Jeff Cui, Hengkai Pan, Aadhithya Iyer, Siddhant Haldar et al.NeurIPS 2024 · 61 citations
- Bridging State and History Representations: Understanding Self-Predictive RLTianwei Ni, Benjamin Eysenbach, Erfan Seyedsalehi, Michel Ma et al.ICLR 2024 · 50 citations
Builds on25
- Bootstrap Your Own Latent - A New Approach to Self-Supervised LearningJean-Bastien Grill, Florian Strub, Florent Altché, Corentin Tallec et al.NeurIPS 2020 · 9,171 citations
- Dream to Control: Learning Behaviors by Latent ImaginationDanijar Hafner, Timothy P. Lillicrap, Jimmy Ba, Mohammad NorouziICLR 2020 · 1,852 citations
- CURL: Contrastive Unsupervised Representations for Reinforcement LearningMichael Laskin, Aravind Srinivas, Pieter AbbeelICML 2020 · 1,261 citations
- Agent57: Outperforming the Atari Human BenchmarkAdrià Puigdomènech Badia, Bilal Piot, Steven Kapturowski, Pablo Sprechmann et al.ICML 2020 · 584 citations
- Planning to Explore via Self-Supervised World ModelsRamanan Sekar, Oleh Rybkin, Kostas Daniilidis, Pieter Abbeel et al.ICML 2020 · 489 citations
Related papers
- Curiosity in Hindsight: Intrinsic Exploration in Stochastic EnvironmentsDaniel Jarrett, Corentin Tallec, Florent Altché, Thomas Mesnard et al.ICML 2023 · 3 citations
- Latent World Models For Intrinsically Motivated ExplorationAleksandr Ermolov, Nicu SebeNeurIPS 2020 · 26 citations
- Active World Model Learning with Progress CuriosityKuno Kim, Megumi Sano, Julian De Freitas, Nick Haber et al.ICML 2020 · 46 citations
- Cell-Free Latent Go-ExploreQuentin Gallouédec, Emmanuel DellandréaICML 2023 · 4 citations
- Information is Power: Intrinsic Control via Information CaptureNicholas Rhinehart, Jenny Wang, Glen Berseth, John D. Co-Reyes et al.NeurIPS 2021 · 14 citations
