Option Discovery in Hierarchical Reinforcement Learning using Spatio-Temporal Clustering
This addresses the challenge of efficient and reusable skill learning in reinforcement learning for AI agents, though it appears incremental as it builds on existing clustering and prediction methods.
The paper tackles the problem of automating skill acquisition in hierarchical reinforcement learning by identifying abstract states and actions, which simplifies and speeds up learning algorithms and enables generalization across tasks without relearning policies. The result is a framework that works without an exact environment model, using sample trajectories and scaling to complex tasks with large state spaces via state aggregation.
This paper introduces an automated skill acquisition framework in reinforcement learning which involves identifying a hierarchical description of the given task in terms of abstract states and extended actions between abstract states. Identifying such structures present in the task provides ways to simplify and speed up reinforcement learning algorithms. These structures also help to generalize such algorithms over multiple tasks without relearning policies from scratch. We use ideas from dynamical systems to find metastable regions in the state space and associate them with abstract states. The spectral clustering algorithm PCCA+ is used to identify suitable abstractions aligned to the underlying structure. Skills are defined in terms of the sequence of actions that lead to transitions between such abstract states. The connectivity information from PCCA+ is used to generate these skills or options. These skills are independent of the learning task and can be efficiently reused across a variety of tasks defined over the same model. This approach works well even without the exact model of the environment by using sample trajectories to construct an approximate estimate. We also present our approach to scaling the skill acquisition framework to complex tasks with large state spaces for which we perform state aggregation using the representation learned from an action conditional video prediction network and use the skill acquisition framework on the aggregated state space.