English

NIMMGen: Learning Neural-Integrated Mechanistic Digital Twins with LLMs

Machine Learning 2026-02-23 v1 Artificial Intelligence Computation and Language

Abstract

Mechanistic models encode scientific knowledge about dynamical systems and are widely used in downstream scientific and policy applications. Recent work has explored LLM-based agentic frameworks to automatically construct mechanistic models from data; however, existing problem settings substantially oversimplify real-world conditions, leaving it unclear whether LLM-generated mechanistic models are reliable in practice. To address this gap, we introduce the Neural-Integrated Mechanistic Modeling (NIMM) evaluation framework, which evaluates LLM-generated mechanistic models under realistic settings with partial observations and diversified task objectives. Our evaluation reveals fundamental challenges in current baselines, ranging from model effectiveness to code-level correctness. Motivated by these findings, we design NIMMgen, an agentic framework for neural-integrated mechanistic modeling that enhances code correctness and practical validity through iterative refinement. Experiments across three datasets from diversified scientific domains demonstrate its strong performance. We also show that the learned mechanistic models support counterfactual intervention simulation.

Keywords

Cite

@article{arxiv.2602.18008,
  title  = {NIMMGen: Learning Neural-Integrated Mechanistic Digital Twins with LLMs},
  author = {Zihan Guan and Rituparna Datta and Mengxuan Hu and Shunshun Liu and Aiying Zhang and Prasanna Balachandran and Sheng Li and Anil Vullikanti},
  journal= {arXiv preprint arXiv:2602.18008},
  year   = {2026}
}

Comments

19 pages, 6 figures

R2 v1 2026-07-01T10:43:52.610Z