SPEAR: A Spectral-Disentangled MoE Neural Operator with Knowledge-Guided Expert Aggregation for Large-Scale PDE Pretraining
On the map: Mixture-of-experts → MoE theory and gating functions
Abstract (arXiv)
Large-scale pre-training has improved the generalization of neural operators across diverse PDEs. However, existing PDE foundation models still struggle with heterogeneous dynamics, where shared representations may cause knowledge interference, while mixture-of-experts (MoE) architectures suffer from increasing expert redundancy. We propose SPEAR, a spectral-disentangled MoE neural operator with knowledge-guided expert aggregation for large-scale PDE pre-training. SPEAR decouples latent features into low- and high-frequency components, enabling shared modeling of transferable dynamics and specialized learning of PDE-specific patterns. To address expert redundancy, we design a knowledge-guided expert aggregation strategy that measures expert similarity from dataset-specific learned knowledge and routing preferences, enabling the identification and consolidation of similar experts. Experiments on twelve PDE datasets and multiple downstream benchmarks demonstrate superior performance in pre-training, fine-tuning, and transfer learning. Furthermore, our aggregation strategy reduces the number of experts by 50% while maintaining or improving prediction accuracy, achieving a balance between model efficiency and generalization for PDE foundation models.
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 29 of its 34 references that are in the map. First posted on arXiv 2 Oct 2026.
Builds on
29 of 34 references.
Closest ancestors · 3
Same direction, newest first
- 2026NESTOR: A Nested MOE-based Neural Operator for Large-Scale PDE Pre-Training25 Feb
- 2025Mixture-of-Experts Operator Transformer for Large-Scale PDE Pre-Training29 Oct
- 2025FreqMoE: Dynamic Frequency Enhancement for Neural PDE Solvers11 May
Other references · 26
- 2017Attention Is All You Needcited by 60,361
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
- 2016Gaussian Error Linear Units (GELUs)cited by 3,502
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,073
- 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,908
- 2019DeepONet: Learning nonlinear operators for identifying differential equations based on the universal approximation theorem of operatorscited by 2,018
- 2018Group Normalizationcited by 1,524
- 2021Choose a Transformer: Fourier or Galerkincited by 343
14 more
- 2023GNOT: A General Neural Operator Transformer for Operator Learningcited by 336
- 2022Transformer for Partial Differential Equations' Operator Learningcited by 283
- 2021Adaptive Fourier Neural Operators: Efficient Token Mixers for Transformerscited by 248
- 2022Towards Multi-spatiotemporal-scale Generalized PDE Modelingcited by 197
- 2024DPOT: Auto-Regressive Denoising Operator Transformer for Large-Scale PDE Pre-Trainingcited by 151
- 2023Merge, Then Compress: Demystify Efficient SMoE with Hints from Its Routing Policycited by 108
- 2023Multiple Physics Pretraining for Physical Surrogate Modelscited by 98
- 2023CFDBench: A Large-Scale Benchmark for Machine Learning Methods in Fluid Dynamicscited by 46
- 2022Mitigating spectral bias for the multiscale operator learningcited by 44
- 2024Unisolver: PDE-Conditional Transformers Towards Universal Neural PDE Solverscited by 35
- 2024Retraining-Free Merging of Sparse MoE via Hierarchical Clusteringcited by 35
- 2025Sub-MoE: Efficient Mixture-of-Expert LLMs Compression via Subspace Expert Mergingcited by 28
- 2024OmniArch: Building Foundation Model For Scientific Computingcited by 9
- 2025Expert Merging: Model Merging with Unsupervised Expert Alignment and Importance-Guided Layer Chunkingcited by 8
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
- 2025Symphony-MoE: Harmonizing Disparate Pre-trained Models into a Coherent Mixture-of-Experts
- 2026SD-MoE: Spectral Decomposition for Effective Expert Specialization
- 2025TT-LoRA MoE: Unifying Parameter-Efficient Fine-Tuning and Sparse Mixture-of-Experts
- 2022Parameter-Efficient Mixture-of-Experts Architecture for Pre-trained Language Models
- 2026Spectral Manifold Regularization for Stable and Modular Routing in Deep MoE Architectures
- 2022Task-Specific Expert Pruning for Sparse Mixture-of-Experts
- 2026Spectral-MoE: Heterogeneous Per-Layer Expert Utilization in Dense-to-MoE Conversion