Search papers, labs, and topics across Lattice.
PrefPI is an iterative framework that steers pretrained generative robot policies towards out-of-distribution behaviors by learning from relative preferences over self-generated trajectories. It formulates preference learning as preference-conditioned generative modeling, using density ratios amplified by classifier-free guidance to create an implicit preference signal. This process enables significant behavioral shifts, such as increasing object transport height on real hardware with limited preference data.
Real-world robot policies can be steered to novel, out-of-distribution behaviors with just 150 preference labels, dramatically increasing object transport height.
We present PrefPI (Preference-Guided Policy Iteration), an iterative framework for steering pretrained generative robot policies using only relative preferences over self-generated trajectories. Unlike prior preference-learning methods that primarily sharpen modes already represented by the policy, we study steering beyond the initial effective support, where desired behaviors are rarely or never observed under the initial policy. Our key idea is to formulate preference learning as preference-conditioned generative modeling: preferred trajectories define a conditional distribution, whose density ratio with the broader behavior prior provides an implicit preference signal amplified by classifier-free guidance (CFG). Repeating this preference-conditioned modeling and guidance step yields a form of preference-guided policy iteration, turning incremental improvements toward previously inaccessible behaviors. Across diffusion policies and the PI0.5 flow- matching VLA in simulation and the real world, PrefPI produces substantial behavioral shifts with limited feedback. In particular, PrefPI increases object transport height from 10.7 cm to 19.8 cm on real hardware with only 150 preference-labeled trajectories.