Accelerated Online Reinforcement Learning using Auxiliary Start State Distributions

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Mehra, Aman, Capone, Alexandre, Schneider, Jeff
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909677251461120
author Mehra, Aman
Capone, Alexandre
Schneider, Jeff
author_facet Mehra, Aman
Capone, Alexandre
Schneider, Jeff
contents A long-standing problem in online reinforcement learning (RL) is of ensuring sample efficiency, which stems from an inability to explore environments efficiently. Most attempts at efficient exploration tackle this problem in a setting where learning begins from scratch, without prior information available to bootstrap learning. However, such approaches fail to leverage expert demonstrations and simulators that can reset to arbitrary states. These affordances are valuable resources that offer enormous potential to guide exploration and speed up learning. In this paper, we explore how a small number of expert demonstrations and a simulator allowing arbitrary resets can accelerate learning during online RL. We find that training with a suitable choice of an auxiliary start state distribution that may differ from the true start state distribution of the underlying Markov Decision Process can significantly improve sample efficiency. We find that using a notion of safety to inform the choice of this auxiliary distribution significantly accelerates learning. By using episode length information as a way to operationalize this notion, we demonstrate state-of-the-art sample efficiency on a sparse-reward hard-exploration environment.
format Preprint
id arxiv_https___arxiv_org_abs_2507_04606
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Accelerated Online Reinforcement Learning using Auxiliary Start State Distributions
Mehra, Aman
Capone, Alexandre
Schneider, Jeff
Machine Learning
Artificial Intelligence
Robotics
A long-standing problem in online reinforcement learning (RL) is of ensuring sample efficiency, which stems from an inability to explore environments efficiently. Most attempts at efficient exploration tackle this problem in a setting where learning begins from scratch, without prior information available to bootstrap learning. However, such approaches fail to leverage expert demonstrations and simulators that can reset to arbitrary states. These affordances are valuable resources that offer enormous potential to guide exploration and speed up learning. In this paper, we explore how a small number of expert demonstrations and a simulator allowing arbitrary resets can accelerate learning during online RL. We find that training with a suitable choice of an auxiliary start state distribution that may differ from the true start state distribution of the underlying Markov Decision Process can significantly improve sample efficiency. We find that using a notion of safety to inform the choice of this auxiliary distribution significantly accelerates learning. By using episode length information as a way to operationalize this notion, we demonstrate state-of-the-art sample efficiency on a sparse-reward hard-exploration environment.
title Accelerated Online Reinforcement Learning using Auxiliary Start State Distributions
topic Machine Learning
Artificial Intelligence
Robotics
url https://arxiv.org/abs/2507.04606