arXiv:2605.28858cs.CEcs.LG2026-05

用可微分框架优化流体方程闭合项,让机器学习直接修正物理模型。

An End-to-End PyTorch Interface for Differentiable PDE Solvers: A RANS Model-Correction Study

论文配图:An End-to-End PyTorch Interface for Differentiable PDE Solvers: A RANS Model-Correction Study
图 1 · 摘自论文原文
  • 将PDE求解器转为隐层,实现端到端可微训练
  • 在NASA凸台和涡轮叶片上成功优化湍流模型参数
  • 适合做物理信息神经网络的科研人员快速上手

本文提出一种全可微分的机器学习框架,用于求解受偏微分方程约束的反问题。该方法结合基础可微分PDE求解器与可训练的参数化修正项 $f_ϕ(w)$,通过将PDE重构为隐层,使其可嵌入任意目标函数,并利用PyTorch自动微分机制进行优化。研究以可压缩流的雷诺平均纳维-斯托克斯方程为例,将闭合项或其部分用可训练参数或神经网络建模。第一个应用针对2D NASA壁挂凸台测试案例,基于时间平均大涡模拟数据优化生产项参数;第二个应用在VKI LS-59涡轮叶片上,通过优化可训练空间场重建Spalart-Allmaras湍流粘度。使用可微分BROADCAST求解器从几何生成数据集,验证了框架灵活性,适用于更广泛的物理引导的、含数据驱动成分的PDE约束问题。

原文摘要 · Abstract (English)

This work presents an end-to-end strategy for solving inverse problems constrained by Partial Differential Equations within a fully differentiable Machine Learning framework. The proposed formulation provides a unified and user-friendly methodology applicable to a wide range of problems, from data assimilation to closure modeling. Our approach combines a baseline differentiable PDE solver, which predicts the state w from the nonlinear system $R(w) = 0$, with a generic additive, parametrized, and differentiable correction $f_ϕ(w)$, with trainable parameters $ϕ$. We show how to optimize phi within a fully differentiable Python workflow by reformulating the PDE as an implicit layer, enabling its integration into arbitrary objective functions, while leveraging PyTorch's automatic differentiation graph. The method is demonstrated on the Reynolds-Averaged Navier-Stokes equations for compressible flows, where the closure term, or a portion of it, is modeled using trainable parameters or a Neural Network. The first application considers the 2D NASA Wall-Mounted Hump test case, where a production-term parameter is optimized against time-averaged LES data. A second application is carried out on the VKI LS-59 turbine blade, where the Spalart-Allmaras eddy viscosity field is reconstructed through the optimization of a trainable spatial field. A dataset is generated starting from the VKI LS-59 turbine blade geometry using the differentiable BROADCAST solver with the Spalart-Allmaras turbulence model. The results highlight the flexibility of the framework, showing its applicability beyond turbulence modeling to a broader class of physics-informed PDE-constrained problems with data-driven components.

PDE求解可微分计算湍流建模物理信息网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。