REVIEW 5 major objections 6 minor 21 references
Interpreting CFD Surrogates through Sparse Autoencoders
T0 review · 5 major / 6 minor · reviewed 2026-08-06 · deepseek-v4-flash
Pith's one-line read Training a sparse autoencoder on a frozen CFD surrogate's node embeddings surfaces vorticity-aligned features that beat PCA and embedding-norm baselines at highlighting flow structures.
desk verdict Credible first transfer of SAEs to CFD surrogate interpretability, but the quantitative eval is unverifiable as written and the disentanglement/trustworthiness claims outrun the evidence. read the letter →
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
The reading
What carries the argument
The load-bearing object is a sparse autoencoder trained on the frozen node embeddings $\{h^L_{i,t}\}$ of the pretrained surrogate. The encoder is a linear map followed by a ReLU, $z = \sigma((h - b_{\mathrm{dec}}) W_{\mathrm{enc}} + b_{\mathrm{enc}})$ with $W_{\mathrm{enc}} \in \mathbb{R}^{d_{\mathrm{in}} \times d_{\mathrm{hid}}}$ and expansion factor $\kappa = 8$; the decoder reconstructs via $\hat h = z W_{\mathrm{dec}} + b_{\mathrm{dec}}$ with $W_{\mathrm{dec}} \in \mathbb{R}^{d_{\mathrm{hid}} \times d_{\mathrm{in}}}$, whose rows are renormalized to unit $\ell^2$ norm after each step. The objective $L_{\mathrm{SAE}} = \|\hat h - h\|_2^2 + \lambda \|z\|_1$ balances reconstruction fidelity against sparsity, and the normalized decoder rows constitute the feature dictionary. Relevance scores—variance, mean absolute, and entropy—rank the dictionary dimensions; taking the top $K$ globally or per time step and summing each node's activations gives mesh-space saliency maps $a_{i,t} = \sum_{d \in K_t} z^{(d)}_{i,t}$, linking abstract latent coordinates back to spatial flow structures.
What would settle it
Recompute Table 2 with an explicitly specified vorticity estimator, such as the discrete curl of the surrogate's predicted velocity field, and with alternative physical masks (pressure gradient, enstrophy, boundary-layer distance); if SAE saliency no longer outperforms PCA or embedding-norm under any of these masks, the paper's claim that SAE features align with physics would lose its quantitative support.
Extended reading notes
Core claim
On the paper's own terms, the central discovery is that the 128-dimensional node embeddings of a trained MeshGraphNet are not irreducibly entangled: an overcomplete sparse autoencoder with an eightfold expansion factor ($d_{\mathrm{hid}} = 1024$) and an L1 sparsity penalty learns decoder rows that act as individual feature detectors, each firing at a distinct mesh region. Aggregating the top salient dimensions—ranked by variance, mean absolute activation, or entropy—produces a node-level saliency map, and the first two rankings match the top-10% high-vorticity nodes better than embedding-norm, PCA, or random baselines (F1 0.60 versus 0.55, 0.49, and 0.09). The paper takes this as evidence that the surrogate's latent space contains disentangled, physically meaningful concepts that can be extracted post-hoc without modifying the predictive model.
Load-bearing premise
The quantitative ranking of all methods assumes that the top 10% vorticity nodes, computed by a definition the paper leaves unresolved in a dangling citation, are the correct ground truth for what an interpretable feature should highlight.
Editorial extensions
If this is right
- A frozen graph surrogate's node embeddings can be decomposed into a sparse dictionary whose individual atoms localize to distinct flow features, so interpretability does not have to be engineered into the architecture.
- SAE-based saliency with variance or mean-absolute scoring beats embedding-norm, PCA, and random baselines at highlighting high-vorticity nodes, providing a quantitative post-hoc trust signal.
- Because global and time-local top-$K$ selections oscillate with the vortex-shedding period (measured by Jaccard similarity), the same dictionary can separate persistent flow structures from transient shedding events without retraining.
- The pipeline is agnostic to the scoring criterion's granularity: variance and mean-abs track coherent structures while entropy tracks short-lived fine-scale features, so users can tune the explanation's temporal resolution.
Reading between the lines
- A natural next test the paper does not run is causal: amplify or suppress a single dictionary atom in the frozen surrogate and check whether the predicted flow field changes most in that atom's region; positive results would upgrade the dictionary from correlational to causal interpretability.
- The time series of individual atom activations could itself be used to detect vortex-shedding events or regime changes automatically, turning the SAE into a flow-diagnostics tool rather than only a saliency map.
- The quantitative evaluation targets only vorticity; extending the same comparison to pressure-gradient, shear, or boundary-layer masks would show whether SAE features are broadly physical or specifically vorticity-tuned.
- Training the same SAE pipeline on a different surrogate or mesh geometry would test the claimed model-agnosticism more strongly than the single CylinderFlow dataset used here.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper proposes a post-hoc interpretability framework for graph-based CFD surrogates. The authors train a sparse autoencoder on frozen node embeddings from a pretrained MeshGraphNet surrogate, rank the resulting dictionary dimensions by variance, mean absolute activation, or entropy, and project the most active dimensions back to the mesh. The evaluation consists of qualitative visualizations (Figures 1-3) and a quantitative comparison (Section 5.4, Table 2) of node-level saliency against a vorticity mask, with the claim that SAE-based saliency outperforms embedding-norm, PCA, and random baselines and thereby identifies monosemantic concepts aligned with physical phenomena.
Significance. If the evaluation were fully specified and the claims appropriately scoped, the paper would make a useful contribution by demonstrating that sparse dictionary learning can be applied post-hoc to physics-based surrogate models and can yield spatially meaningful node saliency. The work has clear strengths: it uses a public benchmark (CylinderFlow), evaluates multiple ranking criteria, compares against several baselines, and its quantitative target (vorticity) is external to the surrogate, so the core comparison is not circular. However, as it stands, the evidence does not support the stronger claims about monosemanticity and trustworthiness, and the central quantitative result rests on an unresolved citation and lacks statistical grounding. The approach is promising and the missing pieces are addressable, so major revision is appropriate rather than rejection.
major comments (5)
- [Section 5.4, Table 2] The ground truth for the quantitative evaluation is defined only as 'the top 10% vorticity nodes (calculated using the method described in [?])'. Because this citation is unresolved, the discrete vorticity estimator (finite-difference scheme, mesh curl, filtering, thresholding) is unspecified, so Table 2 cannot be reproduced and a different vorticity estimator could change the reported ranking. Specify the vorticity computation exactly, or replace it with a fully specified reference.
- [Section 3.2 and Section 5.4] The SAE is trained on 'the collection of node embeddings {h^L_i,t} from testing data' (Section 3.2), and the evaluation in Section 5.4 appears to use the same test rollouts. If the dictionary and the top-K selection are derived from the same snapshots that are scored, the results may reflect in-sample selection rather than generalizable saliency. State the exact data split used for SAE training, top-K selection, and evaluation, and if the split differs, describe it explicitly.
- [Abstract and Section 6 vs. Section 5.4] The abstract and conclusion claim that the method identifies 'monosemantic concepts' and 'enhance[s] explainability and trustworthiness', but the only quantitative result is spatial overlap of a node-level saliency map with a vorticity mask. Such overlap can be produced by mixed, non-monosemantic features and does not by itself establish trustworthiness. Since Section 3.2 states that the construction enables causal ablations, report at least one per-feature purity metric (e.g., activation sparsity per atom, footprint overlap) or a causal intervention test, or restrict the claims to spatial localization only.
- [Table 2] The reported scores for SAE(variance) and SAE(mean_abs) are identical to three decimal places, and SAE(entropy) is identical to embedding-norm, which suggests the comparison has low sensitivity. No error bars, per-timestep distributions, or significance tests are given, so the reader cannot judge whether the differences between methods are meaningful. Report variability across test cases and timesteps and a statistical test (e.g., paired bootstrap or Wilcoxon) for the Table 2 comparisons.
- [Section 4.3 and Section 4.5] Salient dimensions are selected by variance, mean-abs, or entropy of SAE activations, which are generic informativeness criteria unrelated to vorticity or any other physical target. Consequently, the observed vortex-region alignment (Table 2) may be a property of the node-level aggregation a_{i,t} in Section 4.5 rather than of the dictionary atoms. The paper should separately evaluate the contribution of the dictionary (vs. the aggregation rule) and, if the claim is about physical concepts, select or rank dimensions by a physics-aware criterion or justify why generic informativeness should yield physical alignment.
minor comments (6)
- [Section 4.5 and Table 2] Report the actual values of η and K used for the quantitative results; the text gives only visualization budgets (η = 20, 85, 300).
- [Figure 1 caption] The caption contains a typo, 'T op-50'; also define 'Top-50' as K=50 in the caption or text.
- [Table 1] Table 1 uses z* and p_{b,d} without defining z*; align the notation with Eq. (4).
- [Section 4.2] The SAE training description omits learning rate, number of epochs, and batch size; include these details for reproducibility.
- [Introduction] The speedup claim '11x-290x' cites Beale and Majda (1985), which is a vortex-methods paper and not a surrogate-model benchmark; this citation appears inappropriate.
- [Section 5.3] The 'spatial disjointness' claim is based on visual inspection of three selected dimensions; a quantitative overlap metric would strengthen the disentanglement statement.
Circularity Check
No significant circularity: the SAE saliency is selected by activation statistics and scored against an external vorticity mask; the unresolved ground-truth citation is a reproducibility concern, not circularity.
full rationale
The paper's central derivation is not circular. The sparse autoencoder is trained post-hoc on frozen MGN node embeddings using an ℓ1-sparsity objective (Eq. 6), and its hyperparameters (κ = 8, λ = 3×10^-4) are fixed independently of the evaluation target. The salient latent dimensions are ranked by activation statistics alone—variance, mean absolute activation, and entropy (Table 1)—not by alignment with vorticity. The quantitative benchmark in Section 5.4 then compares each method's top-η mesh nodes against an external ground-truth mask of high-vorticity nodes derived from the CFD simulation field, not from the SAE's own activations or from the surrogate's outputs. Baselines such as embedding-norm, PCA, and random node sampling are independent comparators, so Table 2 is a fair external comparison rather than a fitted-input prediction. The unresolved citation '[?]' for the vorticity computation is a genuine reproducibility gap, and the abstract's 'monosemantic' and 'trustworthiness' phrasing goes beyond what a spatial-overlap metric can establish, but these are correctness and scope concerns, not circularity. Self-citations in the introduction ([Hu et al., 2023], [Hu et al., 2024]) are illustrative context about surrogate models and CFD cost and do not carry the derivation. Accordingly, the paper earns a low circularity score despite its evaluation weaknesses.
Assumptions & free parameters
free parameters (5)
- SAE expansion factor kappa =
8
- Sparsity penalty lambda =
3e-4
- Top-K number of salient dimensions K =
50
- Sampling budget eta for top nodes =
20, 85, 300 (Fig. 2); 100 (Fig. 3)
- Entropy histogram bins B =
not reported
assumptions (4)
- domain assumption The final-layer node embeddings h^L_i,t of the pretrained MGN contain sufficient information to recover meaningful physical concepts.
- domain assumption Vorticity alignment is a valid proxy for interpretability and trustworthiness of a CFD surrogate.
- domain assumption The pretrained MeshGraphNets surrogate is accurate enough that its embeddings reflect the true flow physics.
- ad hoc to paper Sparse dictionary learning on ReLU activations yields features that are monosemantic in the embedding space.
Cite this review
Pith. "Pith review of Interpreting CFD Surrogates through Sparse Autoencoders." pith.science (2026). https://pith.science/paper/5D7SKVSC
@misc{pith2026250716069,
author = {Pith},
title = {Pith review of: Interpreting CFD Surrogates through Sparse Autoencoders},
year = {2026},
howpublished = {\url{https://pith.science/paper/5D7SKVSC}},
note = {Machine review of arXiv:2507.16069}
}
read the original abstract
Learning-based surrogate models have become a practical alternative to high-fidelity CFD solvers, but their latent representations remain opaque and hinder adoption in safety-critical or regulation-bound settings. This work introduces a posthoc interpretability framework for graph-based surrogate models used in computational fluid dynamics (CFD) by leveraging sparse autoencoders (SAEs). By obtaining an overcomplete basis in the node embedding space of a pretrained surrogate, the method extracts a dictionary of interpretable latent features. The approach enables the identification of monosemantic concepts aligned with physical phenomena such as vorticity or flow structures, offering a model-agnostic pathway to enhance explainability and trustworthiness in CFD applications.
Figures
Reference graph
Works this paper leans on
-
[1]
Data-Driven Insights into Jet Turbulence: Explainable AI Approaches
[Amico et al., 2025] Enrico Amico, Lorenzo Matteucci, and Gioacchino Cafiero. Data-driven insights into jet tur- bulence: Explainable ai approaches. arXiv preprint arXiv:2503.02126,
work page Pith review arXiv 2025
-
[2]
arXiv preprint arXiv:2408.05147,
-
[5]
Scaling and evaluating sparse autoencoders
[Gao et al., 2024] Leo Gao, Tom Dupr’e la Tour, Henk Tillman, Gabriel Goh, Rajan Troll, Alec Radford, Ilya Sutskever, Jan Leike, and Jeffrey Wu. Scaling and evaluating sparse autoencoders. arXiv preprint arXiv:2406.04093,
arXiv 2024
-
[7]
Comparative study of machine learning tech- niques for post-combustion carbon capture systems
[Hu et al., 2024] Yeping Hu, Bo Lei, Yash Girish Shah, Jose Cadena, Amar Saini, Grigorios Panagakos, and Phan Nguyen. Comparative study of machine learning tech- niques for post-combustion carbon capture systems. Fron- tiers in Artificial Intelligence, 7:1441934,
work page 2024
-
[8]
Gemma scope: Open sparse autoencoders everywhere all at once on gemma
[Lieberum et al., 2024] Tom Lieberum, Senthooran Raja- manoharan, Arthur Conmy, Lewis Smith, Nicolas Son- nerat, Vikrant Varma, J ´anos Kram ´ar, Anca Dragan, Ro- hin Shah, and Neel Nanda. Gemma scope: Open sparse autoencoders everywhere all at once on gemma
work page 2024
-
[10]
[Makhzani and Frey, 2013] Alireza Makhzani and Bren- dan Frey. K-sparse autoencoders. arXiv preprint arXiv:1312.5663,
arXiv 2013
-
[11]
[Marks et al., 2024] Luke Marks, Alasdair Paren, David Krueger, and Fazl Barez. Enhancing neural network inter- pretability with feature-aligned sparse autoencoders.arXiv preprint arXiv:2411.01220,
arXiv 2024
-
[12]
Michaud, Max Tegmark, and Christian Schroeder de Witt
[Mudide et al., 2024] Anish Mudide, Joshua Engels, Eric J. Michaud, Max Tegmark, and Christian Schroeder de Witt. Efficient dictionary learning with switch sparse autoen- coders. arXiv preprint arXiv:2410.08201,
arXiv 2024
Show all 21 references
-
[13]
Diab, and Virginia Smith
[Muhamed et al., 2024] Aashiq Muhamed, Mona T. Diab, and Virginia Smith. Decoding dark matter: Specialized sparse autoencoders for interpreting rare concepts in foun- dation models. arXiv preprint arXiv:2411.00743,
2024 arXiv
-
[14]
Uncertainty quantification and polynomial chaos techniques in computational fluid dy- namics
[Najm, 2009] Habib N Najm. Uncertainty quantification and polynomial chaos techniques in computational fluid dy- namics. Annual review of fluid mechanics , 41(1):35–52,
2009
-
[17]
Jumping ahead: Improving reconstruction fidelity with jumprelu sparse au- toencoders
[Rajamanoharan et al., 2024b] Senthooran Rajamanoharan, Tom Lieberum, Nicolas Sonnerat, Arthur Conmy, Vikrant Varma, J´anos Kram ´ar, and Neel Nanda. Jumping ahead: Improving reconstruction fidelity with jumprelu sparse au- toencoders. arXiv preprint arXiv:2407.14435,
-
[18]
Sparse autoencoders for scien- tifically rigorous interpretation of vision models
[Stevens et al., 2025] Samuel Stevens, Wei-Lun Chao, Tanya Berger-Wolf, and Yu Su. Sparse autoencoders for scien- tifically rigorous interpretation of vision models. arXiv preprint arXiv:2502.06755,
2025
-
[19]
Universal sparse autoencoders: Interpretable cross-model concept alignment
[Thasarathan et al., 2025] Harrish Thasarathan, Julian Forsyth, Thomas Fel, Matthew Kowal, and Kon- stantinos Derpanis. Universal sparse autoencoders: Interpretable cross-model concept alignment. arXiv preprint arXiv:2502.03714,
2025
-
[20]
Artificial intelligence explainability requirements of the ai act and metrics for measuring compliance
[Walke et al., 2023] Fabian Walke, Lars Bennek, and Till J Winkler. Artificial intelligence explainability requirements of the ai act and metrics for measuring compliance. In International Conference on Wirtschaftsinformatik, pages 113–129. Springer,
2023
-
[21]
An explainable deep learning model based on hydrological principles for flood simulation and forecasting
[Xiang et al., 2025] Xin Xiang, Shenglian Guo, Chenglong Li, and Yun Wang. An explainable deep learning model based on hydrological principles for flood simulation and forecasting. EGUsphere [preprint], 2025
2025
-
[1985]
Sparse autoencoders find highly interpretable features in language models
[Cunningham et al., 2023] Hoagy Cunningham, Aidan Ewart, Logan Riggs, Robert Huben, and Lee Sharkey. Sparse autoencoders find highly interpretable features in language models. arXiv preprint arXiv:2309.08600,
2023 arXiv
-
[2009]
Learning mesh- based simulation with graph networks
[Pfaff et al., 2020] Tobias Pfaff, Meire Fortunato, Alvaro Sanchez-Gonzalez, and Peter Battaglia. Learning mesh- based simulation with graph networks. In International conference on learning representations,
2020
-
[2020]
Improving dictionary learning with gated sparse autoen- coders
[Rajamanoharan et al., 2024a] Senthooran Rajamanoharan, Arthur Conmy, Lewis Smith, Tom Lieberum, Vikrant Varma, J ´anos Kram ´ar, Rohin Shah, and Neel Nanda. Improving dictionary learning with gated sparse autoen- coders. arXiv preprint arXiv:2404.16014,
-
[2023]
High order accurate vortex methods with explicit velocity kernels
[Beale and Majda, 1985] J Thomas Beale and Andrew Ma- jda. High order accurate vortex methods with explicit velocity kernels. Journal of Computational Physics , 58(2):188–208,
1985
-
[2024]
Graph learning in physical-informed mesh-reduced space for real-world dynamic systems
[Hu et al., 2023] Yeping Hu, Bo Lei, and Victor M Castillo. Graph learning in physical-informed mesh-reduced space for real-world dynamic systems. In Proceedings of the 29th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, pages 4166–4174,
2023
-
[2025]
Interpretable fine-tuning for graph neural net- work surrogate models
[Barwey and Maulik, 2023] Shivam Barwey and Romit Maulik. Interpretable fine-tuning for graph neural net- work surrogate models. arXiv preprint arXiv:2311.07548,
2023 arXiv
Reviewed August 6, 2026 · model on record in the stance chip above.
Discussion (0). Continue with ORCID to comment.