Efficient Model-Based Reinforcement Learning Through Optimistic Thompson Sampling

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Bayrooti, Jasmine, Ek, Carl Henrik, Prorok, Amanda
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913728447905792
author Bayrooti, Jasmine
Ek, Carl Henrik
Prorok, Amanda
author_facet Bayrooti, Jasmine
Ek, Carl Henrik
Prorok, Amanda
contents Learning complex robot behavior through interactions with the environment necessitates principled exploration. Effective strategies should prioritize exploring regions of the state-action space that maximize rewards, with optimistic exploration emerging as a promising direction aligned with this idea and enabling sample-efficient reinforcement learning. However, existing methods overlook a crucial aspect: the need for optimism to be informed by a belief connecting the reward and state. To address this, we propose a practical, theoretically grounded approach to optimistic exploration based on Thompson sampling. Our model structure is the first that allows for reasoning about joint uncertainty over transitions and rewards. We apply our method on a set of MuJoCo and VMAS continuous control tasks. Our experiments demonstrate that optimistic exploration significantly accelerates learning in environments with sparse rewards, action penalties, and difficult-to-explore regions. Furthermore, we provide insights into when optimism is beneficial and emphasize the critical role of model uncertainty in guiding exploration.
format Preprint
id arxiv_https___arxiv_org_abs_2410_04988
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Efficient Model-Based Reinforcement Learning Through Optimistic Thompson Sampling
Bayrooti, Jasmine
Ek, Carl Henrik
Prorok, Amanda
Machine Learning
Robotics
Learning complex robot behavior through interactions with the environment necessitates principled exploration. Effective strategies should prioritize exploring regions of the state-action space that maximize rewards, with optimistic exploration emerging as a promising direction aligned with this idea and enabling sample-efficient reinforcement learning. However, existing methods overlook a crucial aspect: the need for optimism to be informed by a belief connecting the reward and state. To address this, we propose a practical, theoretically grounded approach to optimistic exploration based on Thompson sampling. Our model structure is the first that allows for reasoning about joint uncertainty over transitions and rewards. We apply our method on a set of MuJoCo and VMAS continuous control tasks. Our experiments demonstrate that optimistic exploration significantly accelerates learning in environments with sparse rewards, action penalties, and difficult-to-explore regions. Furthermore, we provide insights into when optimism is beneficial and emphasize the critical role of model uncertainty in guiding exploration.
title Efficient Model-Based Reinforcement Learning Through Optimistic Thompson Sampling
topic Machine Learning
Robotics
url https://arxiv.org/abs/2410.04988