4 papers
Faster-WAM: Do World Action Models Need Deep Action Modules?
Liheng Ma, Rui Heng Yang, Zhanguang Zhang +4
World Action Models (WAMs) couple robot action prediction with video world models. Existing WAMs with shared-backbone and Mixture-of-Transformers designs generally tie the depth of…
CAPE: Context-Aware Diffusion Policy Via Proximal Mode Expansion for Collision Avoidance
Rui Heng Yang, Xuan Zhao, Leo Maxime Brunswic +5
In robotics, diffusion models can capture multi-modal trajectories from demonstrations, making them a transformative approach in imitation learning. However, achieving optimal perf…
Two-Steps Diffusion Policy for Robotic Manipulation via Genetic Denoising
Mateo Clemente, Leo Brunswic, Rui Heng Yang +5
Diffusion models, such as diffusion policy, have achieved state-of-the-art results in robotic manipulation by imitating expert demonstrations. While diffusion models were originall…
Ergodic Generative Flows
Leo Maxime Brunswic, Mateo Clemente, Rui Heng Yang +3
Generative Flow Networks (GFNs) were initially introduced on directed acyclic graphs to sample from an unnormalized distribution density. Recent works have extended the theoretical…