LLM2D
CodePDE:一个由大规模语言模型驱动的偏微分方程求解器生成推理框架
CodePDE: An Inference Framework for LLM-driven PDE Solver Generation
作者: Shanda Li, Tanya Marwah, Junhong Shen, Weiwei Sun, Andrej Risteski, Yiming Yang, Ameet Talwalkar
发布日期: 5/14/2025
arXiv ID: oai:arXiv.org:2505.08783v1

摘要

arXiv:2505.08783v1 宣传类型: cross 摘要: 偏微分方程(PDEs)是建模物理系统的基本工具,但求解它们仍然是一个复杂的挑战。传统的数值求解器依赖于专家知识的实现,并且计算成本昂贵,而基于神经网络的求解器需要大量训练数据集,通常缺乏可解释性。在这个工作中,我们将PDE求解重新表述为代码生成任务,并引入了CodePDE,这是第一个使用大型语言模型(LLMs)生成PDE求解器的推理框架。通过利用先进的推理时算法和扩展策略,CodePDE解锁了LLMs在PDE求解中的关键能力:推理、调试、自改进和测试时扩展——所有这些都无需针对特定任务进行调优。CodePDE在一系列代表性PDE问题上的性能超越了人类。我们还呈现了一种系统性的实验分析,分析了由LLMs生成的求解器的准确度、效率和数值方案选择。我们的研究结果突显了LLMs在PDE求解中的潜力和当前限制,提供了新的求解器设计视角和未来模型开发的机会。我们的代码可在 https://github.com/LithiumDA/CodePDE 获取。