Environment Design for Inverse Reinforcement Learning

Abstract

Learning a reward function from demonstrations suffers from low sample-efficiency. Even with abundant data, current inverse reinforcement learning methods that focus on learning from a single environment can fail to handle slight changes in the environment dynamics. We tackle these challenges through adaptive environment design. In our framework, the learner repeatedly interacts with the expert, with the former selecting environments to identify the reward function as quickly as possible from the expert’s demonstrations in said environments. This results in improvements in both sample-efficiency and robustness, as we show experimentally, for both exact and approximate inference.

Cite

Text

Kleine Buening et al. "Environment Design for Inverse Reinforcement Learning." International Conference on Machine Learning, 2024.

Markdown

[Kleine Buening et al. "Environment Design for Inverse Reinforcement Learning." International Conference on Machine Learning, 2024.](https://mlanthology.org/icml/2024/kleinebuening2024icml-environment/)

BibTeX

@inproceedings{kleinebuening2024icml-environment,
  title     = {{Environment Design for Inverse Reinforcement Learning}},
  author    = {Kleine Buening, Thomas and Villin, Victor and Dimitrakakis, Christos},
  booktitle = {International Conference on Machine Learning},
  year      = {2024},
  pages     = {24808-24828},
  volume    = {235},
  url       = {https://mlanthology.org/icml/2024/kleinebuening2024icml-environment/}
}