Improving Zero-shot Generalization in Offline Reinforcement Learning using Generalized Similarity Functions
Authors: Bogdan Mazoure, Ilya Kostrikov, Ofir Nachum, Jonathan Tompson
AI-generated Key Points
- Reinforcement learning (RL) agents struggle to generalize their learned behaviors to new scenarios
- Incorporating additional signals beyond the reward function, such as self-supervised learning (SSL), can improve generalization capabilities in RL agents
- Challenges arise when applying online approaches in the offline RL setting
- The study proposes a new framework called Generalized Similarity Functions (GSF) that uses contrastive learning to train an offline RL agent
- GSF aggregates observations based on the similarity of their expected future behavior using generalized value functions
- GSF recovers existing SSL objectives and enhances zero-shot generalization performance on offline Procgen benchmark
- Poor estimation of similarity between observations hinders online algorithms for generalization in the offline setting
- The study provides insights into improving zero-shot generalization in offline reinforcement learning with GSF
- GSF has potential applications in real-world scenarios like autonomous driving at different times of day or night.
Abstract: Reinforcement learning (RL) agents are widely used for solving complex sequential decision making tasks, but still exhibit difficulty in generalizing to scenarios not seen during training. While prior online approaches demonstrated that using additional signals beyond the reward function can lead to better generalization capabilities in RL agents, i.e. using self-supervised learning (SSL), they struggle in the offline RL setting, i.e. learning from a static dataset. We show that performance of online algorithms for generalization in RL can be hindered in the offline setting due to poor estimation of similarity between observations. We propose a new theoretically-motivated framework called Generalized Similarity Functions (GSF), which uses contrastive learning to train an offline RL agent to aggregate observations based on the similarity of their expected future behavior, where we quantify this similarity using \emph{generalized value functions}. We show that GSF is general enough to recover existing SSL objectives while also improving zero-shot generalization performance on a complex offline RL benchmark, offline Procgen.
Tools for this paper
Ask questions about this paper to our AI assistant
You can also chat with multiple papers at once here.
The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.
Similar papers summarized with our AI tools
Navigate through even more similar papers through a
tree representationLook for similar papers (in beta version)
We compare the title and abstract with papers indexed in our database. For arXiv papers, we search within the same category; for DOI and other sources, we search across sources and subjects. You can rerun the search as our database grows.
Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.