MoE^2: A Mixture-of-Mixtures of Experts for Ensemble-Free Domain Generalization

Read full text
Authors: Association for Artificial Intelligence 2026, Omar Abdelaziz, Ahmad Abdel-Qader, Ahmed Radwan, Mohamed Shehata, Mahmoud Soliman
Cited by: 0

Abstract

Domain Generalization (DG) requires models to generalize across unseen data distributions. Kernel-based theory reveals a No-Free-Lunch problem: any model with a fixed representation is fundamentally sub-optimal for all possible shifts. While large ensembles mitigate this, they are computationally expensive and remain static once trained, inheriting the same theoretical limitation. We introduce MoE$^2$ (Mixture-of-Mixtures of Experts), a framework that uses a single frozen backbone to dynamically synthesize a bespoke adapter for each input, allowing it to continuously adapt its effective kernel. We provide a theoretical grounding for this process, proving our routing mechanism is a principled non-parametric estimator for the optimal Bayes mixture of experts. We derive a generalization bound that cleanly separates the router's estimation error from the reduction in a kernel-mismatch penalty achieved via synthesis. MoE$^2$ matches or exceeds state-of-the-art ensemble baselines on major DG benchmarks while using only a single, compact model. MoE$^2$ thus provides a theoretically-grounded and lightweight alternative to large-scale ensembles for robust domain generalization.

Fields of Study & Subjects

Domain Adaptation and Few-Shot Learning Advanced Neural Network Applications Adversarial Robustness in Machine Learning

Publication details

Journal / source
Underline Science Inc.
Publication date
7 January 2026
Language
EN