Action Manifold Smoothing: A Lipschitz Pathway Perspective on High-Dimensional Reinforcement Learning
Abstract
Lay Summary
Modern AI agents that learn to walk, run, or manipulate objects often collapse when the task becomes physically complex—imagine teaching a robot dog with 38 joints to run. These failures are usually treated as bugs to patch, but we found a deeper reason: the AI is trying to draw a smooth, continuous map of "what every action is worth," using only a handful of scattered data points. In high dimensions, this map is full of false peaks and cliffs, which mislead the agent and cause learning to derail. Our solution, Action Manifold Smoothing, gently averages the agent's value estimates over a small neighborhood of nearby actions, rather than trusting a single noisy point. This is analogous to a surveyor estimating ground elevation from several nearby measurements instead of one footstep. With this simple change, two standard learning algorithms that previously failed completely on a simulated 38-joint robot dog now succeed within a single day of training. The result suggests that respecting the geometry of how value changes with action— rather than adding more tricks—is the key to scaling reinforcement learning to high-dimensional physical control.