建立可比的物理方程求解器评测框架,揭示不同求解器在计算成本和稳定性上的巨大差异。
Mosaic: A Benchmark Suite for Differentiable Physics Solvers

- 将多种物理求解器封装为统一接口的容器组件,支持跨语言与自动微分策略
- 14个求解器测试显示计算开销差10倍,部分因结构不兼容无法用于实际任务
- 实测表明梯度准确性不是主要瓶颈,内存、数值稳定性和配置兼容性才是关键
可微分偏微分方程(PDE)求解器支撑着模型内嵌训练、基于梯度的最优控制与反问题求解,但获取正确可用的梯度成本尚无系统记录。集成难度、计算开销、梯度精度与数值条件性在不同求解器间差异显著,只能通过试错发现。我们提出Mosaic,一个可扩展的可微分PDE求解器基准测试框架,标准化求解器梯度访问。每个求解器以容器化组件(Tesseract)形式封装,暴露统一梯度API,无论语言或自动微分策略如何,均支持研究者评估、比较并构建非平凡物理求解器。对流体动力学、结构力学和热传导领域共14个求解器的评估显示:计算开销相差一个数量级,雅可比条件数差异显著,且存在结构性不兼容导致部分求解器完全无法用于真实任务。尽管如此,所有能生成梯度的求解器均收敛至相似最优解,表明实际障碍在于内存限制、数值稳定性与配置兼容性,而非梯度精度本身。Mosaic开源,地址见https://github.com/pasteurlabs/mosaic。
原文摘要 · Abstract (English)
Differentiable partial differential equation (PDE) solvers underpin solver-in-the-loop ML training, gradient-based optimal control, and inverse problems, yet the practical cost of obtaining correct, usable gradients from a given solver on a given problem is largely undocumented. Integration effort, computational cost, gradient accuracy, and numerical conditioning vary widely across solvers and are discoverable only by trial and error. We introduce Mosaic, an extensible benchmarking framework for differentiable PDE solvers that standardizes access to solver gradients. Each solver is packaged as a containerized component (Tesseract) exposing a uniform gradient API regardless of language or automatic differentiation (AD) strategy, enabling researchers to evaluate, compare, and build on non-trivial physical solvers. Our evaluation of 14 solvers across fluid dynamics, structural mechanics, and heat transfer demonstrates that the benchmark surfaces practically relevant differences: order-of-magnitude variation in computational cost and Jacobian conditioning, alongside structural incompatibilities that eliminate solvers from realistic tasks entirely. Despite this variation, all solvers that produce gradients converge to similar optima, indicating that the practical barriers are memory limits, numerical stability, and setup compatibility rather than gradient accuracy alone. Mosaic is open-source and available at https://github.com/pasteurlabs/mosaic.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。