Proceedings

BILE: An Effective Behavior-based Latent Exploration Scheme for Deep Reinforcement Learning

Download BibTeX Cite

Proceedings of the Thirty-Fourth International Joint Conference on Artificial Intelligence. Main Track. Pages 6497-6505.

https://doi.org/10.24963/ijcai.2025/723

Abstract

Efficient exploration of state spaces is critical for the success of deep reinforcement learning (RL). While many methods leverage exploration bonuses to encourage exploration instead of relying solely on extrinsic rewards, these bonus-based approaches often face challenges with learning efficiency and scalability, especially in environments with high-dimensional state spaces.

To address these issues, we propose BehavIoral metric-based Latent Exploration (BILE). The core idea is to learn a compact representation within the behavioral metric space that preserves value differences between states. By introducing additional rewards to encourage exploration in this latent space, BILE drives the agent to visit states with higher value diversity and exhibit more behaviorally distinct actions, leading to more effective exploration of the state space. Additionally, we present a novel behavioral metric for efficient and robust training of the state encoder, backed by theoretical guarantees. Extensive experiments on high-dimensional environments, including realistic indoor scenarios in Habitat, robotic tasks in Robosuite, and challenging discrete Minigrid benchmarks, demonstrate the superiority and scalability of our method over other approaches.

  1. [Badia et al., 2020a] Adrià Puigdomènech Badia, Bilal Piot, Steven Kapturowski, Pablo Sprechmann, Alex Vitvitskyi, Zhaohan Daniel Guo, and Charles Blundell. Agent57: Outperforming the atari human benchmark. In Interna- tional conference on machine learning, pages 507-517.
  2. PMLR, 2020.
  3. [Badia et al., 2020b] Adrià Puigdomènech Badia, Pablo Sprechmann, Alex Vitvitskyi, Daniel Guo, Bilal Piot, Steven Kapturowski, Olivier Tieleman, Martín Arjovsky, Alexander Pritzel, Andew Bolt, et al. Never give up: Learning directed exploration strategies. arXiv preprint arXiv:2002.06038, 2020.
  4. [Bai et al., 2021] Chenjia Bai, Peng Liu, Kaiyu Liu, Lingx- iao Wang, Yingnan Zhao, Lei Han, and Zhaoran Wang. Variational dynamic for self-supervised exploration in deep reinforcement learning. IEEE Transactions on neural networks and learning systems, 34(8):4776-4790, 2021.
  5. [Bellemare et al., 2016a] Marc Bellemare, Sriram Srini- vasan, Georg Ostrovski, Tom Schaul, David Saxton, and Remi Munos. Unifying count-based exploration and in- trinsic motivation. Advances in neural information pro- cessing systems, 29, 2016.
  6. [Bellemare et al., 2016b] Marc Bellemare, Sriram Srini- vasan, Georg Ostrovski, Tom Schaul, David Saxton, and Remi Munos. Unifying count-based exploration and in- trinsic motivation. Advances in neural information pro- cessing systems, 29, 2016.
  7. [Burda et al., 2018a] Yuri Burda, Harri Edwards, Deepak Pathak, Amos Storkey, Trevor Darrell, and Alexei A Efros. Large-scale study of curiosity-driven learning. arXiv preprint arXiv:1808.04355, 2018.
  8. [Burda et al., 2018b] Yuri Burda, Harrison Edwards, Amos Storkey, and Oleg Klimov. Exploration by random net- work distillation. arXiv preprint arXiv:1810.12894, 2018.
  9. [Castro, 2020] Pablo Samuel Castro. Scalable methods for computing state similarity in deterministic markov deci- sion processes. In Proceedings of the AAAI Conference on Artificial Intelligence, volume 34, pages 10069-10076, 2020.
  10. [Chevalier-Boisvert et al., 2024] Maxime Chevalier- Boisvert, Bolun Dai, Mark Towers, Rodrigo Perez- Vicente, Lucas Willems, Salem Lahlou, Suman Pal, Pablo Samuel Castro, and Jordan Terry. Minigrid & miniworld: Modular & customizable reinforcement learning environments for goal-oriented tasks. Advances in Neural Information Processing Systems, 36, 2024.
  11. [Eysenbach et al., 2018] Benjamin Eysenbach, Abhishek Gupta, Julian Ibarz, and Sergey Levine. Diversity is all you need: Learning skills without a reward function. arXiv preprint arXiv:1802.06070, 2018.
  12. [Ferns et al., 2011] Norm Ferns, Prakash Panangaden, and Doina Precup. Bisimulation metrics for continuous markov decision processes. SIAM Journal on Computing, 40(6):1662-1714, 2011.
  13. [Hafner et al., 2019] Danijar Hafner, Timothy Lillicrap, Jimmy Ba, and Mohammad Norouzi. Dream to control: Learning behaviors by latent imagination. arXiv preprint arXiv:1912.01603, 2019.
  14. [Henaff et al., 2022] Mikael Henaff, Roberta Raileanu, Minqi Jiang, and Tim Rocktäschel. Exploration via ellip- tical episodic bonuses. Advances in Neural Information Processing Systems, 35:37631-37646, 2022.
  15. [Houthooft et al., 2016] Rein Houthooft, Xi Chen, Yan Duan, John Schulman, Filip De Turck, and Pieter Abbeel. Vime: Variational information maximizing exploration. Advances in neural information processing systems, 29, 2016.
  16. [Mahankali et al., 2024] Srinath Mahankali, Zhang-Wei Hong, Ayush Sekhari, Alexander Rakhlin, and Pulkit Agrawal. Random latent exploration for deep rein- forcement learning. arXiv preprint arXiv:2407.13755, 2024.
  17. [Mnih et al., 2015] Volodymyr Mnih, Koray Kavukcuoglu, David Silver, Andrei A Rusu, Joel Veness, Marc G Belle- mare, Alex Graves, Martin Riedmiller, Andreas K Fidje- land, Georg Ostrovski, et al. Human-level control through deep reinforcement learning. nature, 518(7540):529-533, 2015.
  18. [Ostrovski et al., 2017a] Georg Ostrovski, Marc G Belle- mare, Aäron Oord, and Rémi Munos. Count-based explo- ration with neural density models. In International con- ference on machine learning, pages 2721-2730. PMLR, 2017.
  19. [Ostrovski et al., 2017b] Georg Ostrovski, Marc G Belle- mare, Aäron Oord, and Rémi Munos. Count-based explo- ration with neural density models. In International con- ference on machine learning, pages 2721-2730. PMLR, 2017.
  20. [Park et al., 2022] Seohong Park, Jongwook Choi, Jaekyeom Kim, Honglak Lee, and Gunhee Kim. Lipschitz- constrained unsupervised skill discovery. In International Conference on Learning Representations, 2022.
  21. [Park et al., 2023] Seohong Park, Oleh Rybkin, and Sergey Levine. Metra: Scalable unsupervised rl with metric- aware abstraction. arXiv preprint arXiv:2310.08887, 2023.
  22. [Pathak et al., 2017] Deepak Pathak, Pulkit Agrawal, Alexei A Efros, and Trevor Darrell. Curiosity-driven exploration by self-supervised prediction. In International conference on machine learning, pages 2778-2787.
  23. PMLR, 2017.
  24. [Pathak et al., 2019] Deepak Pathak, Dhiraj Gandhi, and Ab- hinav Gupta. Self-supervised exploration via disagree- ment. In International conference on machine learning, pages 5062-5071. PMLR, 2019.
  25. [Plappert et al., 2017] Matthias Plappert, Rein Houthooft, Prafulla Dhariwal, Szymon Sidor, Richard Y Chen, Xi Chen, Tamim Asfour, Pieter Abbeel, and Marcin Andrychowicz. Parameter space noise for exploration. arXiv preprint arXiv:1706.01905, 2017.
  26. [Raileanu et al., 2020] Roberta Raileanu, Tim Rocktäschel, et al. Ride: Rewarding impact-driven exploration for procedurally-generated environments. arXiv preprint arXiv:2002.12292, 2020.
  27. [Ratzlaff et al., 2020] Neale Ratzlaff, Qinxun Bai, Li Fuxin, and Wei Xu. Implicit generative modeling for efficient ex- ploration. In International Conference on Machine Learn- ing, pages 7985-7995. PMLR, 2020.
  28. [Schulman et al., 2017] John Schulman, Filip Wolski, Pra- fulla Dhariwal, Alec Radford, and Oleg Klimov. Proximal policy optimization algorithms. ArXiv, abs/1707.06347, 2017.
  29. [Seo et al., 2021] Younggyo Seo, Lili Chen, Jinwoo Shin, Honglak Lee, Pieter Abbeel, and Kimin Lee. State entropy maximization with random encoders for efficient explo- ration. In International Conference on Machine Learning, pages 9443-9454. PMLR, 2021.
  30. [Shyam et al., 2019] Pranav Shyam, Wojciech Jaśkowski, and Faustino Gomez. Model-based active exploration. In International conference on machine learning, pages 5779-5788. PMLR, 2019.
  31. [Stanton and Clune, 2016] Christopher Stanton and Jeff Clune. Curiosity search: producing generalists by en- couraging individuals to continually explore and acquire skills throughout their lifetime. PloS one, 11(9):e0162235, 2016.
  32. [Stanton and Clune, 2018] Christopher Stanton and Jeff Clune. Deep curiosity search: Intra-life exploration im- proves performance on challenging deep reinforcement learning problems. arXiv preprint arXiv:1806.00553, 2018.
  33. [Sutton and Barto, 2018] Richard S Sutton and Andrew G Barto. Reinforcement learning: An introduction. MIT press, 2018.
  34. [Tang et al., 2017] Haoran Tang, Rein Houthooft, Davis Foote, Adam Stooke, OpenAI Xi Chen, Yan Duan, John Schulman, Filip DeTurck, and Pieter Abbeel. # explo- ration: A study of count-based exploration for deep rein- forcement learning. Advances in neural information pro- cessing systems, 30, 2017.
  35. [Tao et al., 2020] Ruo Yu Tao, Vincent Franc ¸ois-Lavet, and Joelle Pineau. Novelty search in representational space for sample efficient exploration. Advances in Neural Informa- tion Processing Systems, 33:8114-8126, 2020.
  36. [Wang et al., 2023] Yiming Wang, Ming Yang, Renzhi Dong, Binbin Sun, Furui Liu, et al. Efficient potential- based exploration in reinforcement learning using inverse dynamic bisimulation metric. Advances in Neural Infor- mation Processing Systems, 36:38786-38797, 2023.
  37. [Wang et al., 2024] Yiming Wang, Kaiyan Zhao, Furui Liu, et al. Rethinking exploration in reinforcement learning with effective metric-based exploration bonus. In The Thirty-eighth Annual Conference on Neural Information Processing Systems, 2024.
  38. [Yang et al., 2023a] Ming Yang, Renzhi Dong, Yiming Wang, Furui Liu, Yali Du, Mingliang Zhou, and Leong Hou U. Tiecomm: Learning a hierarchical communica- tion topology based on tie theory. In International Con- ference on Database Systems for Advanced Applications, pages 604-613. Springer, 2023.
  39. [Yang et al., 2023b] Ming Yang, Yiming Wang, Yang Yu, Mingliang Zhou, et al. Mixlight: Mixed-agent coopera- tive reinforcement learning for traffic light control. IEEE Transactions on Industrial Informatics, 20(2):2653-2661, 2023.
  40. [Yang et al., 2024] Ming Yang, Kaiyan Zhao, Yiming Wang, Renzhi Dong, Yali Du, Furui Liu, Mingliang Zhou, and Leong Hou U. Team-wise effective communication in multi-agent reinforcement learning. Autonomous Agents and Multi-Agent Systems, 38(2):36, 2024.
  41. [Yarats et al., 2021] Denis Yarats, Rob Fergus, Alessandro Lazaric, and Lerrel Pinto. Reinforcement learning with prototypical representations. In International Conference on Machine Learning, pages 11920-11931. PMLR, 2021.
  42. [Zang et al., 2023] Hongyu Zang, Xin Li, Leiji Zhang, Yang Liu, Baigui Sun, Riashat Islam, Remi Tachet des Combes, and Romain Laroche. Understanding and addressing the pitfalls of bisimulation-based representations in offline re- inforcement learning. Advances in Neural Information Processing Systems, 36:28311-28340, 2023.
  43. [Zhang et al., 2020] Amy Zhang, Rowan Thomas McAllis- ter, Roberto Calandra, Yarin Gal, and Sergey Levine. Learning invariant representations for reinforcement learn- ing without reconstruction. ArXiv, abs/2006.10742, 2020.
  44. [Zhang et al., 2021] Tianjun Zhang, Huazhe Xu, Xiaolong Wang, Yi Wu, Kurt Keutzer, Joseph E Gonzalez, and Yuandong Tian. Noveld: A simple yet effective explo- ration criterion. Advances in Neural Information Process- ing Systems, 34:25217-25230, 2021.
  45. [Zhao et al., 2024] Kaiyan Zhao, Yiming Wang, Yuyang Chen, Yan Li, Xiaoguang Niu, et al. Efficient diversity- based experience replay for deep reinforcement learning. arXiv preprint arXiv:2410.20487, 2024.
  46. [Zhu et al., 2020] Yuke Zhu, Josiah Wong, Ajay Mandlekar, Roberto Martín-Martín, Abhishek Joshi, Soroush Nasiri- any, and Yifeng Zhu. robosuite: A modular simula- tion framework and benchmark for robot learning. arXiv preprint arXiv:2009.12293, 2020.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards