Eradicating Negative Transfer in Multi-Physics Foundation Models via Sparse Mixture-of-Experts Routing
On the map: Mixture-of-experts → MoE theory and gating functions
Abstract (arXiv)
Scaling Scientific Machine Learning (SciML) toward universal foundation models is bottlenecked by negative transfer: the simultaneous co-training of disparate partial differential equation (PDE) regimes can induce gradient conflict, unstable optimization, and plasticity loss in dense neural operators. In particular, broadband open-channel fluid dynamics and boundary-dominated porous media flows impose incompatible spectral and geometric demands on a single dense parameter path. We introduce Shodh-MoE, a sparse-activated latent transformer architecture for multi-physics transport. Shodh-MoE operates on compressed 16^3 physical latents produced by a physics-informed autoencoder with an intra-tokenizer Helmholtz-style velocity parameterization, restricting decoded states to divergence-free velocity manifolds. The model guarantees exact mass conservation, achieving a physically verifiable velocity divergence of ~2.8 x 10^-10 (evaluated post-hoc in FP64) on 128^3 grids. A Top-1 soft-semantic router dynamically assigns localized latent patches to expert subnetworks, enabling specialized parameter paths for distinct physical mechanisms while preserving shared experts for universal symmetries. In a 20,000-step distributed pretraining run over mixed three-dimensional physical tensors, routing telemetry shows autonomous domain bifurcation: held-out validation tokens from the open-channel domain route exclusively to Expert 0, while porous-media tokens route exclusively to Expert 1. The model converges simultaneously across both regimes, achieving latent validation MSEs of 2.46 x 10^-5 and 9.76 x 10^-6, and decoded physical MSEs of 2.48 x 10^-6 and 1.76 x 10^-6. These results support sparse expert routing as a practical architectural mechanism for mitigating multi-physics interference in universal neural operators.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in MoE theory and gating functions, one of 12 directions in Mixture-of-experts. Builds on 11 of its 12 references that are in the map. First posted on arXiv 14 May 2026.
Builds on
11 of 12 references.
References · 11
- 2020Gradient Surgery for Multi-Task Learningcited by 1,256
- 2024Poseidon: Efficient Foundation Models for PDEscited by 200
- 2024The Well: a Large-Scale Collection of Diverse Physics Simulations for Machine Learningcited by 145
- 2024ConFIG: Towards Conflict-free Training of Physics Informed Neural Networkscited by 46
- 2025Walrus: A Cross-Domain Foundation Model for Continuum Dynamicscited by 36
- 2025Load Balancing Mixture of Experts with Similarity Preserving Routerscited by 22
- 2025PhysiX: A Foundation Model for Physics Simulationscited by 22
- 2025DISCO: learning to DISCover an evolution Operator for multi-physics-agnostic predictioncited by 21
- 2025One Model for All Tasks: Leveraging Efficient World Models in Multi-Task Planningcited by 3
- 2026PI-JEPA: Label-Free Surrogate Pretraining for Coupled Multiphysics Simulation via Operator-Split Latent Predictioncited by 3
- 2026HyCOP: Hybrid Composition Operators for Interpretable Learning of PDEscited by 2
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Latent-MoE: Domain-Aware Mixture-of-Experts for PDEs with Multi-Regime Physics
- 2026Learning to Think in Physics: Breaking Shortcut Learning in Scientific Diffusion via Representation Alignment
- 2026Latent Generative Solvers for Generalizable Long-Term Physics Simulation
- 2026Graph Transfer Learning via Shared Latent Geometry: Theory and Applications
- 2025Tensor-Compressed and Fully-Quantized Training of Neural PDE Solvers
- 2025Physics-Constrained Fine-Tuning of Flow-Matching Models for Generation and Inverse Problems
- 2026COSMOS: Soft Mechanism Mixtures with Verifiable Routing for Long-Horizon PDE Forecasting
- 2025Towards Foundational Models for Dynamical System Reconstruction: Hierarchical Meta-Learning via Mixture of Experts