You Only Live Once: Single-Life Reinforcement Learning
This addresses the challenge of autonomous adaptation in real-world scenarios like disaster relief, where agents must perform tasks once without supervision, though it is incremental as it builds on existing RL methods.
The paper tackles the problem of single-life reinforcement learning (SLRL), where an agent must complete a task successfully in a single trial without interventions, and finds that standard episodic RL algorithms struggle in this setting. The proposed QWALE algorithm improves success rates by 20-60% on continuous control problems by using distribution matching to recover from novel states.
Reinforcement learning algorithms are typically designed to learn a performant policy that can repeatedly and autonomously complete a task, usually starting from scratch. However, in many real-world situations, the goal might not be to learn a policy that can do the task repeatedly, but simply to perform a new task successfully once in a single trial. For example, imagine a disaster relief robot tasked with retrieving an item from a fallen building, where it cannot get direct supervision from humans. It must retrieve this object within one test-time trial, and must do so while tackling unknown obstacles, though it may leverage knowledge it has of the building before the disaster. We formalize this problem setting, which we call single-life reinforcement learning (SLRL), where an agent must complete a task within a single episode without interventions, utilizing its prior experience while contending with some form of novelty. SLRL provides a natural setting to study the challenge of autonomously adapting to unfamiliar situations, and we find that algorithms designed for standard episodic reinforcement learning often struggle to recover from out-of-distribution states in this setting. Motivated by this observation, we propose an algorithm, $Q$-weighted adversarial learning (QWALE), which employs a distribution matching strategy that leverages the agent's prior experience as guidance in novel situations. Our experiments on several single-life continuous control problems indicate that methods based on our distribution matching formulation are 20-60% more successful because they can more quickly recover from novel states.