Search papers, labs, and topics across Lattice.
This paper introduces MATE, a Multi-Modal MoE framework designed to enhance trajectory prediction for robotic manipulation by effectively integrating visual, linguistic, and trajectory inputs. By employing a novel cross-modal cosine router and techniques like temperature-controlled routing, MATE mitigates the challenges of modality interference and expert collapse, particularly in data-scarce environments. Experimental results demonstrate a 4.75% improvement in success rates on the LIBERO benchmark and practical applicability in real-world robotic tasks, showcasing the framework's robustness and efficiency.
MATE achieves a 4.75% boost in robotic manipulation success rates under data scarcity by leveraging specialized expert routing across heterogeneous inputs.
Robotic manipulation requires the effective integration of heterogeneous inputs, including visual observations, language instructions, and trajectory representations, to generate accurate actions. Existing transformer-based policies typically process these heterogeneous modalities within a shared parameter space, which often leads to modality interference and inefficient representation learning, especially in data-scarce scenarios. While Mixture-of-Experts (MoE) offers a scalable solution through expert specialization, conventional routing mechanisms are often sensitive to such cross-modal representation discrepancies, resulting in unstable expert assignment and expert collapse. In this work, we propose MATE (Multi-ModAl TrajEctory Policies), a novel trajectory prediction framework built upon MoE. Specifically, we introduce a Multi-Modal MoE architecture to achieve fine-grained sub-token feature decoupling, and design a cross-modal cosine router for stable and scale-invariant expert assignment across heterogeneous modalities. We further employ temperature-controlled routing and stochastic noise injection to improve expert balance and prevent premature routing collapse under scarce demonstrations. Experiments on the LIBERO benchmark show that our MATE consistently outperforms prior work under data scarcity. It achieves a 4.75% improvement in average success rate over the trajectory-guided counterpart. Real-world experiments on robotic ping-pong also suggest that the predicted trajectories can provide useful guidance for downstream robotic execution, further indicating the practical feasibility of our algorithm.
Tsinghua AI