Bottleneck-Guided Spectral Subgoals For Offline Goal-Conditioned RL
Abstract
Lay Summary
Many artificial intelligence agents need to complete tasks that require many steps, such as moving through a maze or manipulating objects in a kitchen. A common difficulty is that success often depends on passing through a few critical places, such as narrow corridors, doorways, or important transition points. If the agent does not recognize these places, it may waste time moving locally or choose unhelpful intermediate targets. This paper proposes a method that helps an agent identify and use these critical transition points. Instead of asking the agent to choose subgoals based only on time or short-term scores, our method first analyzes the offline data to discover the main regions of the environment and the boundaries between them. The states near these boundaries are treated as keypoints, because they often indicate where the agent must pass in order to make real progress toward the final goal. After finding these keypoints, the agent builds a simple graph that describes which keypoints can be reached from others. At test time, it plans a route through this graph and then uses a low-level controller to move from one keypoint to the next. This separates long-horizon planning from short local control, making the overall task easier. Our theoretical analysis explains why bottleneck keypoints can serve as useful intermediate goals under suitable assumptions, and our experiments show that this idea works well across a range of navigation and manipulation tasks. Overall, the method helps offline agents make better long-term decisions by focusing on the important transition points that structure the environment.