PDE-Controller: LLMs for Autoformalization and Reasoning of PDEs

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Soroco, Mauricio, Song, Jialin, Xia, Mengzhou, Emond, Kye, Sun, Weiran, Chen, Wuyang
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912423565328384
author Soroco, Mauricio
Song, Jialin
Xia, Mengzhou
Emond, Kye
Sun, Weiran
Chen, Wuyang
author_facet Soroco, Mauricio
Song, Jialin
Xia, Mengzhou
Emond, Kye
Sun, Weiran
Chen, Wuyang
contents While recent AI-for-math has made strides in pure mathematics, areas of applied mathematics, particularly PDEs, remain underexplored despite their significant real-world applications. We present PDE-Controller, a framework that enables large language models (LLMs) to control systems governed by partial differential equations (PDEs). Our approach enables LLMs to transform informal natural language instructions into formal specifications, and then execute reasoning and planning steps to improve the utility of PDE control. We build a holistic solution comprising datasets (both human-written cases and 2 million synthetic samples), math-reasoning models, and novel evaluation metrics, all of which require significant effort. Our PDE-Controller significantly outperforms prompting the latest open source and GPT models in reasoning, autoformalization, and program synthesis, achieving up to a 62% improvement in utility gain for PDE control. By bridging the gap between language generation and PDE systems, we demonstrate the potential of LLMs in addressing complex scientific and engineering challenges. We release all data, model checkpoints, and code at https://pde-controller.github.io/.
format Preprint
id arxiv_https___arxiv_org_abs_2502_00963
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle PDE-Controller: LLMs for Autoformalization and Reasoning of PDEs
Soroco, Mauricio
Song, Jialin
Xia, Mengzhou
Emond, Kye
Sun, Weiran
Chen, Wuyang
Machine Learning
While recent AI-for-math has made strides in pure mathematics, areas of applied mathematics, particularly PDEs, remain underexplored despite their significant real-world applications. We present PDE-Controller, a framework that enables large language models (LLMs) to control systems governed by partial differential equations (PDEs). Our approach enables LLMs to transform informal natural language instructions into formal specifications, and then execute reasoning and planning steps to improve the utility of PDE control. We build a holistic solution comprising datasets (both human-written cases and 2 million synthetic samples), math-reasoning models, and novel evaluation metrics, all of which require significant effort. Our PDE-Controller significantly outperforms prompting the latest open source and GPT models in reasoning, autoformalization, and program synthesis, achieving up to a 62% improvement in utility gain for PDE control. By bridging the gap between language generation and PDE systems, we demonstrate the potential of LLMs in addressing complex scientific and engineering challenges. We release all data, model checkpoints, and code at https://pde-controller.github.io/.
title PDE-Controller: LLMs for Autoformalization and Reasoning of PDEs
topic Machine Learning
url https://arxiv.org/abs/2502.00963