mlsanityData to 7 Oct 2026

AutoSurrogate: An LLM-Driven Multi-Agent Framework for Autonomous Construction of Deep Learning Surrogate Models in Subsurface Flow

arXiv 2604.11945 ↗First posted 13 Apr 2026cs.LG cs.AI cs.MAcited by 0

On the map: LLM agents and tool use → Web, search, and research agents

Abstract (arXiv)

High-fidelity numerical simulation of subsurface flow is computationally intensive, especially for many-query tasks such as uncertainty quantification and data assimilation. Deep learning (DL) surrogates can significantly accelerate forward simulations, yet constructing them requires substantial machine learning (ML) expertise - from architecture design to hyperparameter tuning - that most domain scientists do not possess. Furthermore, the process is predominantly manual and relies heavily on heuristic choices. This expertise gap remains a key barrier to the broader adoption of DL surrogate techniques. For this reason, we present AutoSurrogate, a large-language-model-driven multi-agent framework that enables practitioners without ML expertise to build high-quality surrogates for subsurface flow problems through natural-language instructions. Given simulation data and optional preferences, four specialized agents collaboratively execute data profiling, architecture selection from a model zoo, Bayesian hyperparameter optimization, model training, and quality assessment against user-specified thresholds. The system also handles common failure modes autonomously, including restarting training with adjusted configurations when numerical instabilities occur and switching to alternative architectures when predictive accuracy falls short of targets. In our setting, a single natural-language sentence can be sufficient to produce a deployment-ready surrogate model, with minimum human intervention required at any intermediate stage. We demonstrate the utility of AutoSurrogate on a 3D geological carbon storage modeling task, mapping permeability fields to pressure and CO2 saturation fields over 31 timesteps. Without any manual tuning, AutoSurrogate is able to outperform expert-designed baselines and domain-agnostic AutoML methods, demonstrating strong potential for practical deployment.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Web, search, and research agents, one of 16 directions in LLM agents and tool use. Builds on 28 of its 69 references that are in the map. First posted on arXiv 13 Apr 2026.

Builds on

28 of 69 references.

References · 28

  1. 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,355
  2. 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,146
  3. 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,806
  4. 2019Optuna: A Next-generation Hyperparameter Optimization Frameworkcited by 2,455
  5. 2024OpenHands: An Open Platform for AI Software Developers as Generalist Agentscited by 956
  6. 2023A Comprehensive Overview of Large Language Modelscited by 639
  7. 2019Explaining Machine Learning Classifiers through Diverse Counterfactual Explanationscited by 612
  8. 2022Scientific Machine Learning through Physics-Informed Neural Networks: Where we are and What's nextcited by 603
  9. 2021U-FNO -- An enhanced Fourier neural operator-based deep-learning model for multiphase flowcited by 292
  10. 2019Physics-Constrained Deep Learning for High-dimensional Surrogate Modeling and Uncertainty Quantification without Labeled Datacited by 261
  11. 2020On Hyperparameter Optimization of Machine Learning Algorithms: Theory and Practicecited by 181
  12. 2025A Comprehensive Survey of Self-Evolving AI Agents: A New Paradigm Bridging Foundation Models and Lifelong Agentic Systemscited by 177
16 more
  1. 2025AI Agents vs. Agentic AI: A Conceptual Taxonomy, Applications and Challengescited by 167
  2. 2024The Landscape of Emerging AI Agent Architectures for Reasoning, Planning, and Tool Calling: A Surveycited by 164
  3. 2019A deep-learning-based surrogate model for data assimilation in dynamic subsurface flow problemscited by 59
  4. 2018Deep convolutional encoder-decoder networks for uncertainty quantification of dynamic multiphase flow in heterogeneous mediacited by 56
  5. 2024Large Language Models for Constructing and Optimizing Machine Learning Workflows: A Surveycited by 19
  6. 2020Efficient Uncertainty Quantification for Dynamic Subsurface Flow with Surrogate by Theory-guided Neural Networkcited by 17
  7. 2020Theory-guided Auto-Encoder for Surrogate Construction and Inverse Modelingcited by 12
  8. 2021Surrogate and inverse modeling for two-phase flow in porous media via theory-guided convolutional neural networkcited by 5
  9. 2023Graph Network Surrogate Model for Subsurface Flow Optimizationcited by 4
  10. 2024Deep Learning Framework for History Matching CO2 Storage with 4D Seismic and Monitoring Well Datacited by 3
  11. 2024Accelerated training of deep learning surrogate models for surface displacement and flow, with application to MCMC-based history matching of CO2 storage operationscited by 3
  12. 2025InsightX Agent: An LMM-based Agentic Framework with Integrated Tools for Reliable X-ray NDT Analysiscited by 3
  13. 2025Recurrent Transformer U-Net Surrogate for Flow Modeling and Data Assimilation in Subsurface Formations with Faultscited by 3
  14. 2026Veri-Sure: A Contract-Aware Multi-Agent Framework with Temporal Tracing and Formal Verification for Correct RTL Code Generationcited by 3
  15. 2023Explaining and visualizing black-box models through counterfactual pathscited by 2
  16. 2025AeroGPT: Leveraging Large-Scale Audio Model for Aero-Engine Bearing Fault Diagnosiscited by 2

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Agentic Exploration of PDE Spaces using Latent Foundation Models for Parameterized Simulations
  2. 2025SURGIN: SURrogate-guided Generative INversion for subsurface multiphase flow with quantified uncertainty
  3. 2025Automating Data-Driven Modeling and Analysis for Engineering Applications using Large Language Model Agents
  4. 2026LAViG-FLOW: Latent Autoregressive Video Generation for Fluid Flow Simulations
  5. 2025Fine-tuning a Large Language Model for Automating Computational Fluid Dynamics Simulations
  6. 2026AutoFOAM: The Self-Refining Autonomous OpenFOAM Agent
  7. 2025Foam-Agent: A Large Language Model-Based Multi-Agent Framework for Automating Computational Fluid Dynamics Workflows
  8. 2026KISS - Knowledge Infrastructure for Scientific Simulation: A Scaffolding for Agentic Earth Science