Understanding Multimodal Failure in Action-Chunking Behavioral Cloning
On the map: Offline RL and imitation learning from data → Robot manipulation imitation learning
Abstract (arXiv)
Behavioral cloning becomes challenging when the same observation admits several valid actions. We study how generative behavioral-cloning policies represent such multimodal expert behavior and identify different bottlenecks across model parameterizations. For latent-variable policies, preserving demonstrated modes requires action-conditioned information in the latent representation. Excessive posterior-prior regularization can suppress this information and prevent the policy from distinguishing demonstrated modes. Weaker or aggregate regularization can preserve mode information, but shifts the challenge to ensuring that the deployment-time prior covers the relevant latent regions. For action-space generative policies, multimodality is constrained by the smoothness of the base-to-action transport: a map with a small Lipschitz constant cannot assign substantial probability to many well-separated modes. Covering many modes therefore requires either sharp transitions in base space or off-support bridge regions in action space. Experiments on synthetic multimodal navigation and a physical-robot bimodal manipulation task support these mechanisms. In contrast, our analysis reveals limited conditional multimodality in standard robotic simulation benchmarks, where deterministic regression remains competitive.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in Robot manipulation imitation learning, one of 13 directions in Offline RL and imitation learning from data. Builds on 16 of its 43 references that are in the map. First posted on arXiv 21 May 2026.
Builds on
16 of 43 references.
Closest ancestors · 3
Same direction, newest first
- 2025Quantization-Free Autoregressive Action Transformer18 Mar
- 2023Learning Fine-Grained Bimanual Manipulation with Low-Cost Hardware23 Apr
- 2019Relay Policy Learning: Solving Long-Horizon Tasks via Imitation and Reinforcement Learning25 Oct
Other references · 13
- 2024OpenVLA: An Open-Source Vision-Language-Action Modelcited by 2,777
- 2024π0: A Vision-Language-Action Flow Model for General Robot Controlcited by 2,297
- 2025π0.5: a Vision-Language-Action Model with Open-World Generalizationcited by 1,653
- 2024Octo: An Open-Source Generalist Robot Policycited by 1,427
- 2022Diffusion Policies as an Expressive Policy Class for Offline Reinforcement Learningcited by 543
- 2025SmolVLA: A Vision-Language-Action Model for Affordable and Efficient Roboticscited by 507
- 2017Wasserstein Auto-Encoderscited by 371
- 2023Goal-Conditioned Imitation Learning using Score-based Diffusion Policiescited by 229
- 2016Variational Lossy Autoencodercited by 175
- 2024Behavior Generation with Latent Actionscited by 159
- 2025The Pitfalls of Imitation Learning when Actions are Continuouscited by 25
- 2026Lipschitz regularity in Flow Matching and Diffusion Models: sharp sampling rates and functional inequalitiescited by 6
1 more
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026MARS Policy: Multimodality Only When It Matters
- 2026Measuring the Stability Assumption Behind Action Chunking
- 2026Identifying Habit, Physics, and Nuisance in Robot World Models
- 2025Improving Generative Behavior Cloning via Self-Guidance and Adaptive Chunking
- 2026Reuse Before You Retrieve: Diagnosing Headroom and Complementarity for Test-Time Augmentation of Embodied Multimodal Policies
- 2024How Generalizable Is My Behavior Cloning Policy? A Statistical Approach to Trustworthy Performance Evaluation
- 2026IMPACT-VLA: Interaction-aware Multimodal Propagation Attribution via Counterfactual Trajectories for Vision-Language-Action Policies
- 2026Conditional Trajectory Peaks: Single-Pass Multimodal Policies over Action Chunks