Semantically Structured Mixture-of-Experts for Compositional Robotic Manipulation
For robotic manipulation, this work addresses the trade-off between model performance and computational cost by leveraging semantic task structure for efficient expert routing, enabling better generalization and transfer.
The paper tackles the scalability bottleneck of diffusion-based robotic manipulation policies by introducing a semantically structured Mixture-of-Experts framework (SMoDP) that grounds expert specialization in semantic task structure. SMoDP achieves state-of-the-art performance on multi-task benchmarks with significantly improved parameter efficiency and demonstrates effective compositional transfer to novel tasks.
Diffusion-based policies have established a new standard for precise robotic manipulation but face a critical scalability bottleneck: high-performance models are computationally expensive, while lightweight alternatives often fail to generalize across diverse multi-task environments. Mixture-of-Experts (MoE) architectures offer a promising path to efficiency by activating only a subset of parameters. However, existing MoE routing mechanisms typically rely on low-level noise or latent statistics, ignoring the compositional nature of manipulation tasks. This can fragment reusable behaviors across experts, limiting interpretability and transferability. We introduce Semantically Structured Mixture-of-Experts Diffusion Policy (SMoDP) for compositional robotic manipulation, a framework that grounds expert specialization in semantic task structure. SMoDP leverages a lightweight, inference-time skill predictor, supervised by offline annotations from Vision-Language Models (VLMs), to route action chunks to experts specialized for specific behavioral phases. To ensure robust assignment, we propose a dual contrastive alignment strategy that grounds multi-modal observations in language-defined skill semantics (Inter-modal) while enforcing routing consistency across visually distinct but functionally related behaviors (Intra-modal). Our approach outperforms representative diffusion and MoE-based baselines on multi-task benchmarks with significantly improved parameter efficiency and demonstrates effective compositional transfer to novel tasks through parameter-efficient fine-tuning. Project website: https://deng-cy20.github.io/SMoDP/