Search papers, labs, and topics across Lattice.
This paper introduces LeaP, a Learnable source Prior that enhances generative robot policies by replacing the standard Gaussian initialization with a proprioception-conditioned diagonal Gaussian over action chunks. By predicting mean and state-adaptive variance through a lightweight MLP, LeaP allows for a more informed and stochastic initialization, enabling the generator to refine actions more effectively. In experiments across 15 RoboTwin manipulation tasks, LeaP achieved an average success rate of 81.6%, significantly outperforming several baselines by up to 25.5 percentage points, demonstrating the importance of the source distribution in generative policy design.
A simple change in the source distribution can boost generative robot policy success rates by over 25%.
Generative robot policies typically begin action generation from an observation-independent standard Gaussian distribution, leaving the choice of source distribution underexplored. This work asks a simple question: where should action generation begin? We propose LeaP, a Learnable source Prior that replaces the standard Gaussian with a proprioception-conditioned diagonal Gaussian over action chunks. Parameterized by a lightweight MLP, LeaP jointly predicts the mean and state-adaptive variance of the source distribution, while keeping the downstream generator architecture and inference solver unchanged. This design provides an observation-informed yet stochastic initialization, allowing the generator to focus on precise action refinement rather than transporting samples from an uninformed noise source. On 15 RoboTwin manipulation tasks, LeaP achieves an average success rate of 81.6%, outperforming four representative baselines -- including deterministic-source methods, a no-prior counterpart, and a diffusion-bridge policy -- by 6.5 to 25.5 percentage points. The same prior consistently improves both flow-matching and diffusion-bridge generators, while using fewer parameters and converging faster. The advantage carries over to real-world deployment, where LeaP attains the best performance. These results suggest that the source distribution is an independent and reusable design axis for generative robot policies, complementary to the choice of generative dynamics.
