mlsanityData to 7 Oct 2026

COSMOS: Soft Mechanism Mixtures with Verifiable Routing for Long-Horizon PDE Forecasting

arXiv 2610.04427 ↗First posted 3 Oct 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Neural operators offer an efficient alternative to classical PDE solvers, but most learn a monolithic map per equation family and discretization. Real systems are compositional: transport, diffusion, wave, and reaction processes can act simultaneously. Existing mixture-of-experts operators typically use sparse top-K routing, although concurrent physics is naturally a blend rather than a discrete choice. We propose COSMOS (Cooperative Operator Specialists with Mechanism-level Operator Soft-routing), a soft mechanism-mixture neural operator. Four process-biased specialists remain active at every step and are continuously mixed by a learned gate, with their features fused by a small network. Specialists share a coarse latent grid, while a zero-initialized full-resolution residual restores detail lost through the bottleneck. We also introduce an operator-splitting compositional benchmark with known mixture weights w^star per trajectory. Against family-tuned FNO under 20-step rollouts, an initial 3-seed evaluation suggested gains on diffusion--reaction, parity on Navier--Stokes, and weaker shallow-water performance. An 11-seed audit showed that the diffusion--reaction gain was unstable, motivating caution in small-seed rollout comparisons and precluding a reliable accuracy-win claim on these families. Ablations show that uniform routing or removing the specialist mixture substantially degrades stable-regime rollouts. On the labeled benchmark, dense soft routing yields 2.2× lower error than hard top-1 routing at identical fusion; using generator weights w^star at inference further lowers rollout error to 0.034, diagnosing limitations of the learned gate. However, routing labels do not align with the specialists' intended mechanisms: COSMOS supports compositional accuracy, not mechanism identity.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). First posted on arXiv 3 Oct 2026.

Builds on

No reference list yet.

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025CFO: Learning Continuous-Time PDE Dynamics via Flow-Matched Neural Operators
  2. 2026Autoregression-Free Neural Operators for Time-Dependent PDEs
  3. 2025Compositional Generation for Long-Horizon Coupled PDEs
  4. 2026Latent Generative Solvers for Generalizable Long-Term Physics Simulation
  5. 2026Learning Physical Operators using Neural Operators
  6. 2023Real-time Inference and Extrapolation via a Diffusion-inspired Temporal Transformer Operator (DiTTO)
  7. 2026NESTOR: A Nested MOE-based Neural Operator for Large-Scale PDE Pre-Training
  8. 2026PDEFlow: Autonomous Agentic PDE Pipelines for Neural Operator Learning and Solver-Free Inference