Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces Closed-Form Diffusion Policies (CFDP), a novel approach to imitation learning that eliminates the need for extensive offline training by utilizing closed-form scores derived from demonstration datasets. The method allows for real-time inference on mobile CPUs, achieving competitive performance against traditional neural diffusion policies that require significant training time. Key findings reveal that CFDP not only accelerates the data collection and deployment process but also serves as a versatile tool for editing pre-trained policies, enhancing their functionality without additional training overhead.
Training-free imitation learning is now possible with CFDP, achieving real-time performance that rivals traditional methods without the long training delays.
While diffusion-based policies have impressive performance and expressivity, their long offline training slows down the data collection and policy deployment loop. We introduce Closed-Form Diffusion Policies, a class of training-free diffusion-based policies for imitation learning using the closed-form score derived from the demonstration dataset. We deploy CFDP with real-time inference with a mobile CPU in hardware experiments, showing it can successfully perform imitation directly from the dataset in milliseconds and with faster inference than neural diffusion policies. In experiments on imitation learning benchmarks, we show that CFDP is competitive against neural baselines that require hours of training, providing a favorable tradeoff between training time and performance. Finally, we show how closed-form diffusion policies act as a composable primitive that enables data-driven inference-time editing of pre-trained neural diffusion policies, including policy guidance and novel demonstration augmentation.