IMLE Policy: Fast and Sample Efficient Visuomotor Policy Learning via Implicit Maximum Likelihood Estimation

2025-02-17 · Krishan Rana, Robert Lee, David Pershouse, Niko Suenderhauf

Recent advances in imitation learning, particularly using generative modelling techniques like diffusion, have enabled policies to capture complex multi-modal action distributions. However, these methods often require large datasets and multiple inference steps for action generation, posing challenges in robotics where the cost for data collection is high and computation resources are limited. To address this, we introduce IMLE Policy, a novel behaviour cloning approach based on Implicit Maximum Likelihood Estimation (IMLE). IMLE Policy excels in low-data regimes, effectively learning from minimal demonstrations and requiring 38\% less data on average to match the performance of baseline methods in learning complex multi-modal behaviours. Its simple generator-based architecture enables single-step action generation, improving inference speed by 97.3\% compared to Diffusion Policy, while outperforming single-step Flow Matching. We validate our approach across diverse manipulation tasks in simulated and real-world environments, showcasing its ability to capture complex behaviours under data constraints. Videos and code are provided on our project page: https://imle-policy.github.io/.

📄 PDF Abstract BibTeX arXiv:2502.12371

Code (0)

등록된 구현이 없습니다.

Tasks

Action GenerationImitation Learning

Methods 이 논문이 사용한 방법론

Diffusion Diffusion models generate samples by gradually removing noise from a signal, and their training objective can be expressed as a reweighted variational lower-bound…

SPEED The monocular depth estimation (MDE) is the task of estimating depth from a single frame. This information is an essential knowledge in many computer vision tasks such as scene…

Consistency Policy: Accelerated Visuomotor Policies via Consistency Distillation

2024-05-13 · Aaditya Prasad, Kevin Lin, Jimmy Wu, Linqi Zhou 외

Many robotic systems, such as mobile manipulators or quadrotors, cannot be equipped with high-end GPUs due to space, weight, and power constraints. These constraints prevent these systems from leveraging recent developme…

GPU

Normalizing Flows are Capable Models for Bi-manual Visuomotor Policy

2025-09-25 · Jialong Li, Simon Kristoffersson Lind, Wenrui Xie, Maj Stenmark 외 arxiv

The field of general-purpose robotics has recently embraced powerful probabilistic diffusion-based models to learn the complex embodiment behaviours. However, existing models often come with significant trade-offs, namel…

Self-Supervised Correspondence in Visuomotor Policy Learning

2019-09-16 · Peter Florence, Lucas Manuelli, Russ Tedrake

In this paper we explore using self-supervised correspondence for improving the generalization performance and sample efficiency of visuomotor policy learning. Prior work has primarily used approaches such as autoencodin…

Imitation Learning

SeFA-Policy: Fast and Accurate Visuomotor Policy Learning with Selective Flow Alignment

2025-11-11 · Rong Xue, Jiageng Mao, Mingtong Zhang, Yue Wang arxiv

Developing efficient and accurate visuomotor policies poses a central challenge in robotic imitation learning. While recent rectified flow approaches have advanced visuomotor policy learning, they suffer from a key limit…

Adversarial Feature Training for Generalizable Robotic Visuomotor Control

2019-09-17 · Xi Chen, Ali Ghadirzadeh, Mårten Björkman, Patric Jensfelt

Deep reinforcement learning (RL) has enabled training action-selection policies, end-to-end, by learning a function which maps image pixels to action outputs. However, it's application to visuomotor robotic policy traini…

Deep Reinforcement Learning Reinforcement Learning Reinforcement Learning (RL)Transfer Learning