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