cs.LGMay 21, 2026

Understanding Multimodal Failure in Action-Chunking Behavioral Cloning

Authors: Lorenzo MazzaMassimiliano DatresAriel RodriguezSebastian BodenstedtGitta KutyniokStefanie Speidel

Organizations: NCT/UCC Dresden, UKDD Dresden, TU Dresden, DKFZ Heidelberg. · Ludwig-Maximilians-Universität München, Munich Center for Machine Learning (MCML). · BMFTR Research Hub 6G-Life · Cluster of Excellence CeTI. · University of Tromsø. · German Aerospace Center (DLR). · HZDR Dresden.

Abstract

Behavioral cloning becomes difficult when the same observation admits several valid actions. We study this problem for action-chunking policies and show that different multimodal parameterizations fail in different ways. For latent-variable policies, posterior-prior regularization makes deployment-time sampling more reliable, but excessive regularization removes the action-conditioned information needed to distinguish demonstrated modes. Reducing this regularization can preserve mode information, but then success depends on whether the 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 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 tasks and robotic simulation benchmarks support these mechanisms.

Explore similar work

CardsList