Proceedings

Constrained Intrinsic Motivation for Reinforcement Learning

Download BibTeX Cite

Proceedings of the Thirty-Third International Joint Conference on Artificial Intelligence. Main Track. Pages 5608-5616.

https://doi.org/10.24963/ijcai.2024/620

Abstract

This paper investigates two fundamental problems that arise when utilizing Intrinsic Motivation (IM) for reinforcement learning in Reward-Free Pre-Training (RFPT) tasks and Exploration with Intrinsic Motivation (EIM) tasks: 1) how to design an effective intrinsic objective in RFPT tasks, and 2) how to reduce the bias introduced by the intrinsic objective in EIM tasks.

Existing IM methods suffer from static skills, limited state coverage, sample inefficiency in RFPT tasks, and suboptimality in EIM tasks.

To tackle these problems, we propose Constrained Intrinsic Motivation (CIM) for RFPT and EIM tasks, respectively: 1) CIM for RFPT maximizes the lower bound of the conditional state entropy subject to an alignment constraint on the state encoder network for efficient dynamic and diverse skill discovery and state coverage maximization; 2) CIM for EIM leverages constrained policy optimization to adaptively adjust the coefficient of the intrinsic objective to mitigate the distraction from the intrinsic objective.

In various MuJoCo robotics environments, we empirically show that CIM for RFPT greatly surpasses fifteen IM methods for unsupervised skill discovery in terms of skill diversity, state coverage, and fine-tuning performance. Additionally, we showcase the effectiveness of CIM for EIM in redeeming intrinsic rewards when task rewards are exposed from the beginning. Our code is available at https://github.com/x-zheng16/CIM.

  1. [Bai et al., 2021] Chenjia Bai, Lingxiao Wang, Lei Han, Jianye Hao, Animesh Garg, Peng Liu, and Zhaoran Wang. Principled exploration via optimistic bootstrapping and backward induction. In Proc. of the International Confer- ence on Machine Learning (ICML), pages 577-587, 2021.
  2. [Barto, 2013] Andrew G. Barto. Intrinsic motivation and re- inforcement learning. In Intrinsically Motivated Learning in Natural and Artificial Systems, pages 17-47. 2013.
  3. [Bellemare et al., 2016] Marc G. Bellemare, Sriram Srini- vasan, Georg Ostrovski, Tom Schaul, David Saxton, and Rémi Munos. Unifying count-based exploration and in- trinsic motivation. In Proc. of the Conference on Neural Information Processing Systems (NeurIPS), pages 1471- 1479, 2016.
  4. [Burda et al., 2019] Yuri Burda, Harrison Edwards, Amos J. Storkey, and Oleg Klimov. Exploration by random net- work distillation. In Proc. of the International Conference on Learning Representations (ICLR), 2019.
  5. [Chen et al., 2022] Eric Chen, Zhang-Wei Hong, Joni Pajari- nen, and Pulkit Agrawal. Redeeming intrinsic rewards via constrained optimization. In Proc. of the Conference on Neural Information Processing Systems (NeurIPS), pages 4996-5008, 2022.
  6. [Eysenbach et al., 2019] Benjamin Eysenbach, Abhishek Gupta, Julian Ibarz, and Sergey Levine. Diversity is all you need: Learning skills without a reward function. In Proc. of the International Conference on Learning Repre- sentations (ICLR), 2019.
  7. [Flet-Berliac et al., 2021] Yannis Flet-Berliac, Johan Ferret, Olivier Pietquin, Philippe Preux, and Matthieu Geist. Ad- versarially guided actor-critic. In Proc. of the International Conference on Learning Representations (ICLR), 2021.
  8. [Fu et al., 2017] Justin Fu, John D. Co-Reyes, and Sergey Levine. EX2: Exploration with exemplar models for deep reinforcement learning. In Proc. of the Conference on Neural Information Processing Systems (NeurIPS), pages 2577-2587, 2017.
  9. [Fu et al., 2020] Justin Fu, Aviral Kumar, Ofir Nachum, George Tucker, and Sergey Levine. D4RL: Datasets for deep data-driven reinforcement learning. arXiv preprint arXiv:2004.07219, 2020.
  10. [Gregor et al., 2017] Karol Gregor, Danilo Jimenez Rezende, and Daan Wierstra. Variational intrinsic control. In Proc. of the International Conference on Learning Representations (ICLR), Workshop Track, 2017.
  11. [Hansen et al., 2020] Steven Hansen, Will Dabney, André Barreto, David Warde-Farley, Tom Van de Wiele, and Volodymyr Mnih. Fast task inference with variational in- trinsic successor features. In Proc. of the International Conference on Learning Representations (ICLR), 2020.
  12. [Hazan et al., 2019] Elad Hazan, Sham M. Kakade, Karan Singh, and Abby Van Soest. Provably efficient maximum entropy exploration. In Proc. of the International Con- ference on Machine Learning (ICML), pages 2681-2691, 2019.
  13. [Laskin et al., 2021] Michael Laskin, Denis Yarats, Hao Liu, Kimin Lee, Albert Zhan, Kevin Lu, Catherine Cang, Ler- rel Pinto, and Pieter Abbeel. URLB: Unsupervised rein- forcement learning benchmark. In Proc. of the Conference on Neural Information Processing Systems (NeurIPS), Datasets and Benchmarks Track, 2021.
  14. [Laskin et al., 2022] Michael Laskin, Hao Liu, Xue Bin Peng, Denis Yarats, Aravind Rajeswaran, and Pieter Abbeel. Unsupervised reinforcement learning with con- trastive intrinsic control. In Proc. of the Conference on Neural Information Processing Systems (NeurIPS), pages 34478-34491, 2022.
  15. [Lee et al., 2021] Kimin Lee, Michael Laskin, Aravind Srinivas, and Pieter Abbeel. SUNRISE: A simple unified framework for ensemble learning in deep reinforcement learning. In Proc. of the International Conference on Ma- chine Learning (ICML), pages 6131-6141, 2021.
  16. [Liu and Abbeel, 2021a] Hao Liu and Pieter Abbeel. APS: Active pretraining with successor features. In Proc. of the International Conference on Machine Learning (ICML), pages 6736-6747, 2021.
  17. [Liu and Abbeel, 2021b] Hao Liu and Pieter Abbeel. Behav- ior from the void: Unsupervised active pre-training. In Proc. of the Conference on Neural Information Processing Systems (NeurIPS), pages 18459-18473, 2021.
  18. [Mutti et al., 2021] Mirco Mutti, Lorenzo Pratissoli, and Marcello Restelli. Task-agnostic exploration via policy gradient of a non-parametric state entropy estimate. In Proc. of the AAAI Conference on Artificial Intelligence (AAAI), pages 9028-9036, 2021.
  19. [Oord et al., 2018] Aaron van den Oord, Yazhe Li, and Oriol Vinyals. Representation learning with contrastive predic- tive coding. arXiv preprint arXiv:1807.03748, 2018.
  20. [Park et al., 2022] Seohong Park, Jongwook Choi, Jaekyeom Kim, Honglak Lee, and Gunhee Kim. Lipschitz- constrained unsupervised skill discovery. In Proc. of the International Conference on Learning Representations (ICLR), 2022.
  21. [Park et al., 2023] Seohong Park, Kimin Lee, Youngwoon Lee, and Pieter Abbeel. Controllability-aware unsuper- vised skill discovery. In Proc. of the International Confer- ence on Machine Learning (ICML), pages 27225-27245, 2023.
  22. [Pathak et al., 2017] Deepak Pathak, Pulkit Agrawal, Alexei A. Efros, and Trevor Darrell. Curiosity-driven exploration by self-supervised prediction. In Proc. of the International Conference on Machine Learning (ICML), pages 2778-2787, 2017.
  23. [Pathak et al., 2019] Deepak Pathak, Dhiraj Gandhi, and Ab- hinav Gupta. Self-supervised exploration via disagree- ment. In Proc. of the International Conference on Machine Learning (ICML), pages 5062-5071, 2019.
  24. [Seo et al., 2021] Younggyo Seo, Lili Chen, Jinwoo Shin, Honglak Lee, Pieter Abbeel, and Kimin Lee. State en- tropy maximization with random encoders for efficient ex- ploration. In Proc. of the International Conference on Ma- chine Learning (ICML), pages 9443-9454, 2021.
  25. [Sharma et al., 2020] Archit Sharma, Shixiang Gu, Sergey Levine, Vikash Kumar, and Karol Hausman. Dynamics- aware unsupervised discovery of skills. In Proc. of the International Conference on Learning Representations (ICLR), 2020.
  26. [Yang et al., 2023] Rushuai Yang, Chenjia Bai, Hongyi Guo, Siyuan Li, Bin Zhao, Zhen Wang, Peng Liu, and Xuelong Li. Behavior contrastive learning for unsupervised skill discovery. In Proc. of the International Conference on Ma- chine Learning (ICML), pages 39183-39204, 2023.
  27. [Zhang et al., 2021] Tianjun Zhang, Paria Rashidinejad, Jiantao Jiao, Yuandong Tian, Joseph E. Gonzalez, and Stu- art Russell. MADE: Exploration via maximizing devia- tion from explored regions. In Proc. of the Conference on Neural Information Processing Systems (NeurIPS), pages 9663-9680, 2021.
  28. [Zhao et al., 2022] Andrew Zhao, Matthieu Gaetan Lin, Yangguang Li, Yong-Jin Liu, and Gao Huang. A mixture of surprises for unsupervised reinforcement learning. In Proc. of the Conference on Neural Information Processing Systems (NeurIPS), pages 26078-26090, 2022.

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

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