mlsanityData to 7 Oct 2026

Discovering Physical Directions in Weight Space: Composing Neural PDE Experts

arXiv 2605.14546 ↗First posted 14 May 2026cs.LGcited by 3

Not on the map. Why?

Abstract (arXiv)

Recent advances in neural operators have made partial differential equation (PDE) surrogate modeling increasingly scalable and transferable through large-scale pretraining and in-context adaptation. However, after a shared operator is fine-tuned to multiple regimes within a continuous physical family, it remains unclear whether the resulting weight-space updates merely form isolated regime experts or reveal reusable physical structure. Starting from a shared family anchor, we fine-tune low- and high-regime endpoint experts and show that their updates can be separated into a family-shared adaptation and a direction aligned with the underlying physical parameter. This separation reinterprets endpoint experts as finite-difference probes of a local physical direction in weight space, explaining why static averaging can interpolate between regimes but attenuates endpoint-specific physics. Building on this perspective, we propose Calibration-Conditioned Merge (CCM), a post-hoc coordinate readout method for composing neural PDE experts along this physical direction. Given physical metadata, a calibrated coordinate mapping, or a short observed rollout prefix, CCM infers the target composition coordinate and deploys a single merged checkpoint for the remaining rollout. We evaluate CCM on the reaction--diffusion system, viscosity-parameterized two-dimensional Navier--Stokes equations, and radial dam-break dynamics. Across these benchmarks, CCM achieves its strongest gains in extrapolative regimes, reducing out-of-distribution rollout error relative to the family anchor by 54.2%, 42.8%, and 13.8%, respectively. Further experiments across FNO scales, a DPOT-style backbone, and ablations confirm that endpoint fine-tuning is not arbitrary checkpoint drift, but reveals a calibratable physical direction for training-free transfer across PDE regimes.

Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Builds on 45 of its 58 references that are in the map. First posted on arXiv 14 May 2026.

Builds on

45 of 58 references.

References · 45

  1. 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,806
  2. 2019DeepONet: Learning nonlinear operators for identifying differential equations based on the universal approximation theorem of operatorscited by 1,965
  3. 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,366
  4. 2022Editing Models with Task Arithmeticcited by 1,076
  5. 2020Learning Mesh-Based Simulation with Graph Networkscited by 840
  6. 2023TIES-Merging: Resolving Interference When Merging Modelscited by 795
  7. 2020Neural Operator: Graph Kernel Network for Partial Differential Equationscited by 722
  8. 2023Language Models are Super Mario: Absorbing Abilities from Homologous Models as a Free Lunchcited by 679
  9. 2021Merging Models with Fisher-Weighted Averagingcited by 613
  10. 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 460
  11. 2022Fourier Neural Operator with Learned Deformations for PDEs on General Geometriescited by 416
  12. 2022PDEBENCH: An Extensive Benchmark for Scientific Machine Learningcited by 396
33 more
  1. 2022Message Passing Neural PDE Solverscited by 368
  2. 2021Choose a Transformer: Fourier or Galerkincited by 333
  3. 2023GNOT: A General Neural Operator Transformer for Operator Learningcited by 321
  4. 2024Transolver: A Fast Transformer Solver for PDEs on General Geometriescited by 299
  5. 2023Geometry-Informed Neural Operator for Large-Scale 3D PDEscited by 293
  6. 2023AdaMerging: Adaptive Model Merging for Multi-Task Learningcited by 248
  7. 2021Factorized Fourier Neural Operatorscited by 237
  8. 2023Neural General Circulation Models for Weather and Climatecited by 232
  9. 2021Multiwavelet-based Operator Learning for Differential Equationscited by 229
  10. 2023Task Arithmetic in the Tangent Space: Improved Editing of Pre-Trained Modelscited by 218
  11. 2023Neural Operators for Accelerating Scientific Simulations and Designcited by 218
  12. 2019Model Fusion via Optimal Transportcited by 217
  13. 2022FourCastNet: Accelerating Global High-Resolution Weather Forecasting using Adaptive Fourier Neural Operatorscited by 188
  14. 2024Poseidon: Efficient Foundation Models for PDEscited by 186
  15. 2024DPOT: Auto-Regressive Denoising Operator Transformer for Large-Scale PDE Pre-Trainingcited by 137
  16. 2024The Well: a Large-Scale Collection of Diverse Physics Simulations for Machine Learningcited by 134
  17. 2024EMR-Merging: Tuning-Free High-Performance Model Mergingcited by 108
  18. 2023In-Context Operator Learning with Data Prompts for Differential Equation Problemscited by 98
  19. 2022Deep transfer operator learning for partial differential equations under conditional shiftcited by 83
  20. 2024Latent Neural Operator for Solving Forward and Inverse PDE Problemscited by 76
  21. 2024Pretraining Codomain Attention Neural Operators for Solving Multiphysics PDEscited by 58
  22. 2024APEBench: A Benchmark for Autoregressive Neural Emulators of PDEscited by 51
  23. 2023Group Equivariant Fourier Neural Operators for Partial Differential Equationscited by 42
  24. 2023MgNO: Efficient Parameterization of Linear Operators via Multigridcited by 40
  25. 2022MAgNet: Mesh Agnostic Neural PDE Solvercited by 28
  26. 2025Walrus: A Cross-Domain Foundation Model for Continuum Dynamicscited by 26
  27. 2024MAP: Low-compute Model Merging with Amortized Pareto Fronts via Quadratic Approximationcited by 20
  28. 2024Diffeomorphism Neural Operator for various domains and parameters of partial differential equationscited by 15
  29. 2025InfiGFusion: Graph-on-Logits Distillation via Efficient Gromov-Wasserstein for Model Fusioncited by 11
  30. 2025Model Merging Scaling Laws in Large Language Modelscited by 10
  31. 2025CALM: Consensus-Aware Localized Merging for Multi-Task Learningcited by 8
  32. 2025InfiFPO: Implicit Model Fusion via Preference Optimization in Large Language Modelscited by 7
  33. 2025An Efficient Graph-Transformer Operator for Learning Physical Dynamics with Manifolds Embeddingcited by 3

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 3.

3 papers.