Sparse probes and murky physics: a case study of interpretability challenges in a foundation model for continuum dynamics
On the map: Mechanistic interpretability → Sparse autoencoder training and evaluation
Abstract (arXiv)
Generative AI emulators are increasingly used in scientific domains where we already have strong theory, benchmarks, and physical intuition. This raises a central evaluation and interpretability question: when a foundation-style model can reproduce known continuum dynamics, what internal mechanism supports that behavior, is the internal behaviour consistent with known physics, and how does it relate to where the emulator succeeds or fails? We investigate a cross-domain foundation model for continuum dynamics, Walrus by Polymathic, using mechanistic interpretability guided by physical principles. We apply a sparse autoencoder (SAE) to probe a selected layer, and address the practical challenge of triaging a large feature set (over 20,000) using enstrophy as a physically grounded metric. As a deliberately simple testbed, we focus on shear flow and compare feature recruitment across multiple shear-flow setups, i.e. parameter values in the numerical simulation. Across setups we find evidence of piecewise consistency, with subsets of features recurring in similar roles, but this structure is intermittent and does not map cleanly onto standard physical decompositions. In parallel, direct comparisons between numerical simulation and the emulator reveal systematic output-level discrepancies, including regimes where energy/structures become too diffuse or too localized. We connect parts of these discrepancies to changes in specific SAE feature usage. Our work highlights open questions for scientific foundation models: how to robustly prioritize mechanistically meaningful features, how to separate stable structure from analysis artifacts (including single-layer and SAE limitations), and how to use established benchmarks to decide when "different" internal representations are genuinely informative rather than merely effective.
Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Placed in Sparse autoencoder training and evaluation, one of 13 directions in Mechanistic interpretability. Builds on 18 of its 37 references that are in the map. First posted on arXiv 10 Jun 2026.
Builds on
18 of 37 references.
Closest ancestors · 2
Same direction, newest first
- 2025Physics Steering: Causal Control of Cross-Domain Concepts in a Physics Foundation Model25 Nov
- 2022Toy Models of Superposition21 Sep
Other references · 16
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,420
- 2024Scaling and evaluating sparse autoencoderscited by 553
- 2019Axial Attention in Multidimensional Transformerscited by 358
- 2024Poseidon: Efficient Foundation Models for PDEscited by 200
- 2024The Well: a Large-Scale Collection of Diverse Physics Simulations for Machine Learningcited by 145
- 2023Multiple Physics Pretraining for Physical Surrogate Modelscited by 98
- 2024Arithmetic Without Algorithms: Language Models Solve Math With a Bag of Heuristicscited by 97
- 2025A Survey on Sparse Autoencoders: Interpreting the Internal Mechanisms of Large Language Modelscited by 76
- 2025Walrus: A Cross-Domain Foundation Model for Continuum Dynamicscited by 36
- 2024An X-Ray Is Worth 15 Features: Sparse Autoencoders for Interpretable Radiology Report Generationcited by 25
- 2025Interpretable Machine Learning in Physics: A Reviewcited by 14
4 more
- 2025Towards mechanistic understanding in a data-driven weather model: internal activations reveal interpretable physical featurescited by 12
- 2023Southern Ocean Dynamics Under Climate Change: New Knowledge Through Physics-Guided Machine Learningcited by 7
- 2025Overtone: Cyclic Patch Modulation for Clean, Efficient, and Flexible Physics Emulatorscited by 5
- 2026OceanCBM: A Concept Bottleneck Model for Mechanistic Interpretability in Ocean Forecastingcited by 1
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026Mechanistic Interpretability and Causal Feature Steering of Neural Quantum States via Sparse Autoencoderscited by 1
- 2026Feature Superposition in Neural Networks: From Theory to Practicecited by 1
- 2026A survey detection channel overrides the pixels in an astronomical foundation model, and biases tomographic mean redshiftscited by 0
- 2026Feature Superposition in Neural Networks: From Theory to Practice6 Sep 2026
- 2026A survey detection channel overrides the pixels in an astronomical foundation model, and biases tomographic mean redshifts23 Aug 2026
- 2026Mechanistic Interpretability and Causal Feature Steering of Neural Quantum States via Sparse Autoencoders1 Jul 2026
Similar, not cited
Close in wording, no citation either way
- 2026Agentic Exploration of PDE Spaces using Latent Foundation Models for Parameterized Simulations
- 2025How does My Model Fail? Automatic Identification and Interpretation of Physical Plausibility Failure Modes with Matryoshka Transcoders
- 2026PhysSAE: Mechanistic Interpretability of PINNs with Sparse Autoencoders
- 2025FunDiff: Diffusion Models over Function Spaces for Physics-Informed Generative Modeling
- 2025Physics-Constrained Fine-Tuning of Flow-Matching Models for Generation and Inverse Problems
- 2025Lost in Latent Space: An Empirical Study of Latent Diffusion Models for Physics Emulation
- 2026Learning to Think in Physics: Breaking Shortcut Learning in Scientific Diffusion via Representation Alignment
- 2026The Observer Effect in World Models: Invasive Adaptation Corrupts Latent Physics