Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

CoRDE: Concept-Prior Routed Diffusion Experts for Structural Generalization in Robot Manipulation

About

Diffusion models excel at capturing multi-modal action distributions in robot imitation learning. However, in multi-task and long-horizon scenarios, monolithic architectures lack structural generalization capabilities, suffering from gradient conflicts between distinct semantic sub-stages. While pure data-driven Mixture-of-Experts (MoE) methods introduce labor division, they frequently trigger routing collapse, and instantiating full-scale experts causes parameter explosion and high expansion costs. To address these issues, we propose Concept-prior Routed Diffusion Experts (CoRDE), a structure-guided variational distillation framework. CoRDE extracts semantic distributions from a frozen concept encoder to guide the variational posterior responsibility via a learnable soft mapping matrix. This mechanism introduces an entropy-controlled responsibility inference process that encourages confident routing under reliable semantic predictions while preserving the stochastic diffusion term for behavioral diversity. To overcome parameter inflation, CoRDE employs a parameter-efficient expert pool using Low-Rank Adaptation (LoRA) on a shared frozen backbone. Theoretical analysis shows that the mixture score discrepancy is bounded by responsibility-weighted local expert errors, supporting high-fidelity generation under low-rank expert adaptation. Empirical evaluations confirm that, compared to existing baselines, CoRDE systematically reduces routing collapse, forming robust, semantically aligned expert allocations while achieving superior action quality and incremental learning efficiency.

Haidong Huang, Xixin Zhao, Yaohua Zhou, Jiayu Song, Jiayi Zhang, Jun Ma, Haiyue Zhu, Xiaocong Li• 2026

Related benchmarks

TaskDatasetResultRank
pushingD3IL state-based
Inference Time (ms)2.63
6
Imitation LearningD3IL Avoiding
Task Success Rate97
4
Imitation LearningD3IL Aligning
Task Success Rate88
4
Imitation LearningD3IL Pushing
Task Success Rate93
4
Imitation LearningD3IL Stacking-1
Task Success Rate86
4
Imitation LearningD3IL Stacking-2
Task Success Rate72
4
Imitation LearningD3IL Sorting (Image)
Task Success Rate93
4
Imitation LearningD3IL Stacking (Image)
Task Success Rate85
4
Showing 8 of 8 rows

Other info

Follow for update