TR2026-155

OpInf-LLM: Parametric PDE Solving with LLMs via Operator Inference


    •  Wang, Z., Hu, H., Deng, X., Mowlavi, S., Nakahira, Y., "OpInf-LLM: Parametric PDE Solving with LLMs via Operator Inference", Transactions on Machine Learning Research, August 2026.
      BibTeX TR2026-155 PDF
      • @article{Wang2026aug,
      • author = {Wang, Zhuoyuan and Hu, Hanjiang and Deng, Xiyu and Mowlavi, Saviz and Nakahira, Yorie},
      • title = {{OpInf-LLM: Parametric PDE Solving with LLMs via Operator Inference}},
      • journal = {Transactions on Machine Learning Research},
      • year = 2026,
      • month = aug,
      • url = {https://www.merl.com/publications/TR2026-155}
      • }
  • MERL Contact:
  • Research Areas:

    Artificial Intelligence, Computational Sensing, Data Analytics, Dynamical Systems, Machine Learning

Abstract:

Solving diverse partial differential equations (PDEs) is fundamental in science and engineering. Large language models (LLMs) have demonstrated strong capabilities in code generation, symbolic reasoning, and tool use, but reliably solving PDEs across heterogeneous settings remains challenging. Prior work on LLM-based code generation and transformer-based foundation models for PDE learning has shown promising advances. However, a persistent trade-off between execution success rate and numerical accuracy arises, particularly when generalization to unseen parameters and boundary conditions is required. In this work, we propose OpInf-LLM, an LLM parametric PDE solving framework via operator inference. The proposed framework leverages small amounts of solution data to enable accurate prediction of diverse PDE instances, including unseen parameters and configurations, and provides seamless integration with LLMs for natural language task specification and physics-based reasoning of proper feature parameterization. Its low computational demands and unified solution pipeline further enable a high execution success rate across heterogeneous settings, opening new possibilities for generalizable reduced-order modeling in LLM-based PDE solving. Code is available at https://github.com/jacobwang925/opinf-llm.