cs.LGOct 7, 2026

Cross-Domain Pretraining for Steady-State Neural CFD Surrogates

Authors: Anthony Zhou, Amir Barati Farimani, Shirley Ho, Rudy Morel

Organizations: Carnegie Mellon University · New York University · Polymathic AI · Princeton University · Flatiron Institute, Center for Computational Astrophysics · Flatiron Institute, Center for Computational Mathematics · Flatiron Institute, Scientific Computing Core

Abstract

Neural surrogates for computational fluid dynamics (CFD) have the potential to greatly enhance engineering innovation through accelerating simulation. However, the primary limitation for neural surrogates is the lack of generalization to geometries and applications beyond the training set, which is significant given the diversity of engineering scenarios. Currently, this is addressed by generating a new dataset for a specific application; however, this requires running costly numerical solvers. In this work, we take a step toward addressing this by studying neural surrogates trained across different geometries, boundary conditions, and fidelities. We find that cross-domain pretraining improves zero- and few-shot performance on held-out datasets relative to both training from scratch and transferring from domain-specific experts. In particular, finetuning a pretrained, cross-domain model can achieve 2-3x lower errors at the same sample size and use 8x fewer samples to achieve the same error, compared to training from scratch. This benefit is architecture agnostic and improves with model size and pretraining dataset diversity. Furthermore, we study how and why cross-domain pretraining works in CFD surrogates, and find that simply pooling steady-state datasets is both sufficient and effective. Given the high cost of generating CFD data, leveraging existing datasets through cross-domain pretraining will likely be a valuable strategy as future surrogates expand to tackle new problems and use cases.

Figures & tables

Appendix figures & tables35 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. How Does Distribution Shift Shape Pretraining Gains in Neural PDE Surrogates?

    Sep 17, 2026Pochinapeddi Sai Bhargav, Nithin Somasekharan, Rohit Sunil Kanchi +2Surrogate ModelsComputational Fluid Dynamic

  2. Inpainting physics: self-supervised learning for context-driven fluid simulation

    May 9, 2026Jonas Weidner, Yeray Martin-Ruisanchez, Daniel Rueckert +2Computational Fluid DynamicFluid Dynamics

  3. When Does Equivariance Help? Canonical Alignment in Neural Fluid Surrogates

    May 12, 2026Patryk Rygiel, Julian Suk, Kak Khee Yeung +2EquivarianceComputational Fluid Dynamic