Solving Parameter-Robust Avoid Problems with Unknown Feasibility using Reinforcement Learning
Organizations: Department of Aeronautics and Astronautics, MIT · MIT Lincoln Laboratory
Abstract
Recent advances in deep reinforcement learning (RL) have achieved strong results on high-dimensional control tasks, but applying RL to reachability problems raises a fundamental mismatch: reachability seeks to maximize the set of states from which a system remains safe indefinitely, while RL optimizes expected returns over a user-specified distribution. This mismatch can result in policies that perform poorly on low-probability states that are still within the safe set. A natural alternative is to frame the problem as a robust optimization over a set of initial conditions that specify the initial state, dynamics and safe set, but whether this problem has a solution depends on the feasibility of the specified set, which is unknown a priori. We propose Feasibility-Guided Exploration (FGE), a method that simultaneously identifies a subset of feasible initial conditions under which a safe policy exists, and learns a policy to solve the reachability problem over this set of initial conditions. Empirical results demonstrate that FGE learns policies with over 50% more coverage than the best existing method for challenging initial conditions across tasks in the MuJoCo simulator and the Kinetix simulator with pixel observations.
Figures & tables
Appendix figures & tables17 assets
Supplementary material from the paper’s appendix.
Appendix
| Method | Hyperparameter | Value |
| All | Neural Network Size | [256, 256] |
| Activation Function | ||
| NN Optimizer | Adam | |
| # envs (per collect) | 1024 | |
| # rollout steps | 30 | |
| FGE | 0.02 |
| Method | Hyperparameter | ToyLevels | Dubins | Hopper | Cheetah | FixedWing |
|---|---|---|---|---|---|---|
| All | Policy learning rate | 4e-4 | 5e-5 | 4e-4 | 4e-4 | 4e-4 |
| Value learning rate | 1e-3 | 8e-4 | 8e-4 | 8e-4 | 8e-4 | |
| # PPO epochs | 5000 | 50000 | 50000 | 50000 | 50000 | |
| Truncate timesteps | 199 | 1000 | 1000 | 1000 | 1000 | |
| Batch size | 30 | 30 | 15 | 30 | 30 | |
| Entropy coefficient | 1e-3 | 5e-3 | 2e-3 | 1e-4 | 1e-4 |
Safe RL
Safe reinforcement learning studies how agents can learn and act while respecting safety constraints and limiting harmful outcomes.
See all 213 papers