AI for Science 系列(二):基于 AI 框架的 CFD 工具组件!赛桨 v1.0 Beta API 介绍以及典型案例分享!
AI for Science 被广泛认为是下一代科研范式,可以有效处理多维度、多模态、多场景下的模拟和真实数据,解决复杂推演计算问题,加速新科学问题发 。百度飞桨科学计算工具组件赛桨 PaddleScience 是国内首个公开且可应用于 CFD(Computational Fluid Dynamics,计算流体力学)领域的工具,提供端到端应用 API,致力于解决科学计算类任务。赛桨综合数学计算与物理数据相结合的处理方法,提供物理机理约束的 PINNs(Physics Informed Neural Networks 物理信息神经网络)加速求解偏微分方程,解决计算流体力学中的仿真分析。本篇文章将重点介绍赛桨 PaddleScience v1.0 Beta 的典型案例及 API 使用示例。
赛桨提供的典型案例包含使用 AI 方法进行顶盖驱动方腔流(LDC)、达西流、2D&3D 圆柱绕流的流场预测及涡激振动(VIV)。所有案例都基于泛化的微分方程、PINNs 求解器、控制体、网络定义以及可视化等多种接口。下面我们将详细讲解赛桨 PaddleScience v1.0 Beta 中提供的计算流体力学案例及科学计算 API 的功能及使用。
图 1 PaddleScience v1.0 Beta 产品全景
场景案例详解
前言
在流体领域,流体力学作为比较传统的物理学科,广泛应用于航空航天、船舶工业以及建筑、水利、能源等工程领域。
在航空工程和造船工业中,如飞行器和船舶的外形设计、操控性、稳定性等问题成为流体力学中广泛的研究课题,并促使流体力学得到了很大的发展。
在水利工程中,如大型水利枢纽和水力发电站的设计与建造、洪峰的预报工作、河流泥沙等问题都与流体力学紧密关联。
在动力机械制造工业中,如何提高水力及蒸汽涡轮、喷气发动机、压缩机和水泵等动力机械的性能,与叶片、导流片及其他零件设计形状的正确性有关。
随着工程问题的深入,流体力学已经逐渐与其他学科互相交叉渗透,形成新的交叉学科或边缘科学,如磁流体力学、物理-化学流体力学、生物流变学等等。
围绕不同的流体问题,当前流体力学分析主要基于数值计算。典型的方法有直接数值模拟 (DNS) 、雷诺平均方法 (RANS)、大涡模拟方法 (LES)、分离涡模拟 (DES) 以及格子玻尔兹曼法(LBM)等求解流体本构 N-S (Navier-Stokes) 方程。由于物理问题的复杂度,数值方法往往有很多局限性,如计算效率等。PaddleScience 的 PINNs 科学计算求解器,将物理信息融入神经网络,针对流体力学问题提供新的解决思路。本章节中,我们将介绍 2D&3D 圆柱绕流及涡激振动的案例,展示 PaddleScience 的基础科研能力。
图 2 不可压缩 N-S(Navier-Stokes)方程
2D 非定常圆柱绕流
圆柱绕流作为经典的 CFD 问题,在不同的雷诺数下,涡脱落产生的卡门涡街类型不同,是能够综合体现层流、湍流过渡的典型问题。针对该问题,采用 PINNs 方法,并基于连续时间的 2D 不可压、非定常 NS 方程作为约束深度学习神经网络的物理规则,将传统的 CFD 求解转换为神经网络参数的优化问题。同时,为了加速训练的收敛时间,提高预测精度,采用半监督方式,从开源 CFD 工具 OpenFOAM 的结果中记录边界位置处约 200 个测点在不同时刻的流场信息,与 N-S 方程、初边值条件等共同形成了网络优化的损失函数。对雷诺数 Re=100 工况,定义约 110W 个时空训练点(t, x, y),并采用 NVIDIA V100-32G 单卡训练约 8 小时,结果如下图所示。基于 PINNs 方法构建的网络能够完整的模拟卡门涡的周期性脱落,且预测的流场结果与 OpenFOAM 相对误差小于 5%(除边界层中个别点外),可满足工程需求。
图 3 Re=100 的 2D 圆柱绕流结果
(左:OpenFOAM 结果,右:PINNs 方法结果)
同时,基于训练完成的模型,利用推理功能,8ms 内即可完成百万级空间位置在 30 个时间步的结果预测,相比于同样网格数量且固定求解配置的 OpenFOAM 计算过程,速度提升近 12000 倍,非常利于后期在线评估、优化等工程应用。
案例链接:
https://github.com/PaddlePaddle/PaddleScience/tree/develop/examples/cylinder/2d_unsteady_continuous
3D 非定常圆柱绕流
除 2D 圆柱绕流外,赛桨 PaddleScience 同时提供 3D 圆柱绕流案例,基于 PINNs 方法,求解 3D 非定常不可压缩 Navier-Stokes 方程,完成了无模化 Re=100 的圆柱绕流问题求解。在该问题中,采用 OpenFOAM 的结果作为基准,选择流场稳定的相对 0 时刻,并记录有限时间内特定测点的数据。采用离散时间的 PINNs 神经网络进行训练,其模型数量与所选取的时间步长相关,在 t_start 至 t_end 内,定义离散时间步长 dt,采用(t_end-t_start)/(T*dt)个模型进行训练,其中 T 为涡脱周期。对比 OpenFOAM 的理论值,相对误差在 5%以内。x 轴上的流场速度变化如下:
图 4 3D 圆柱绕流 x 轴流场速度
案例链接:
https://github.com/PaddlePaddle/PaddleScience/tree/develop/examples/cylinder/3d_unsteady_discrete
涡激振动
涡激振动(VIV)是一种典型的流致振动,是流体经过结构后稳定涡脱频率与结构固有频率综合的流固耦合问题。是海洋工程中立管、输油管路等大跨度结构的主要损伤源,但由于结构复杂,无法有效测量刚度、阻尼等本构特征,导致损伤探测及预防难度较大。相比基于 PINNs 方法实现 2D 圆柱绕流正问题分析,解决“反问题”是 PINNs 方法的一个亮点,可通过部分实验数据“逆向”预测流场中结构的刚度、阻尼等本构特征,从而对实际工程中的复杂结构的疲劳损伤及破坏的预测及预防提出指导。
从工程落地的角度,赛桨 PaddleScience 从“半实物仿真”的技术路线出发,构建了涡激振动试验台架以及模型驱动试验装备的电控闭环,创新性地探索并论证深度学习模型与物理装备的虚实结合的技术可行性。通过对涡激振动开展分阶段分析,首先基于加速度计及激光设备等传感器采集结构振动振幅与结构升力,对系统进行弹簧-振子单自由度等效。基于实测的 160 组位移及受力数据,训练过程中最小化“振幅”、“升力”、“方程”等共同组成的损失函数。采用 V100-16G 单卡训练约 0.5h,预测结果如下图(右上)。基于 PINNs 预测的结构振动振幅、结构升力结果与试验数据的相对误差均在 2%以内。同时基于“反问题”方式分析得到的结构刚度、阻尼分别为 1.092964 与 4.1e-6,相比真实值 1.09 与 0,相对误差均小于 2%。
涡激振动主要基于试验与深度学习模型结合的方式进行,通过构建风洞试验装备,在第一阶段完成结构刚度、阻尼的预测,在第二阶段则基于得到的结构刚度、阻尼等属性,进行流场重构以及升阻力的预测。过程中基于赛桨提供的泛化 PDE 接口,对涡激振动中流固耦合方程也重新进行了整合,定义新的网络与求解过程,具体流程如下图所示。
图 5 VIV 试验及深度学习模型联合验
案例链接:
https://github.com/PaddlePaddle/PaddleScience/tree/develop/examples/fsi
API 功能及使用示例
上述案例充分利用了赛桨 PaddleScience v1.0 Beta 提供的 API。本部分将着重介绍涉及的主要 API 接口及使用示例方法。
图 6 PaddleScience 科学计算工具组件设计架构
科学计算问题定义
科学计算问题定义包含三部分:方程定义(PDE)、计算域定义(Geometry)、初边值条件定义(IC/BC)。下图展示了如何定义在一个方形区域内求解二维非定常 Navier-Stokes 方程问题。具体而言,该问题数学上由(图 b) 定义,包含方程、初边值条件及计算域信息,代码上每一条信息由相应接口描述(图 a 及图 c)。图中展示了一部分数学信息和接口的对应关系,该对应关系由同样的颜色方框表示。
图 7 PaddleScience 科学计算问题定义
本章节将介绍关于科学计算问题定义相关的 API。
偏微分方程(PDE)
赛桨支持调用预置方程接口及自定义方程接口。
预置方程接口:方程定义内置在赛桨中,用户直接调用即可,仅需配置相关参数(如维度、物理属性等)。
自定义方程接口:用户使用 Python SymPy 定义自变量、因变量及方程,通过“add_equation”接口将方程加入 PDE 模块。
对于高维偏微分方程,连续时间方法将时间和空间均视为网络的输入;离散时间方法首先使用数值(隐式)方法对时间离散,得到仅包含空间变量的方程,进而使用 PINNs 方法求解该方程,3D N-S 方程及使用隐式离散方法得到的方程如下,其中 n 时刻的状态(3 个方向的速度)已知,求解 n+1 时刻的状态(3 个方向的速度)。
图 8 使用隐式方法对时间离散后得到的方程
如下代码展示如何定义一个非定常 N-S 方程,以及如何使用连续时间方法和离散时间方法,进行方程离散。仅需通过改变参数 "time_method" 的取值即可选取对应的方法。
计算域(Geometry)
赛桨提供了常用的计算域形状,如立方体、立方体除去圆柱等几何外形,及 VTK 可视化支持,同时提供 " add_boundary " 接口定义边界。
初边值条件(IC/BC)
边值条件模块预置了 Dirichlet/Neumann/Robin 边值条件,仅需定义边值条件,并通过 "set_bc" 接口将其指定到对应边界即可。对于 PDE,以类似方式增加了初值条件模块。
深度学习方法求解模块
上一章节介绍了如何使用赛桨定义科学计算问题,本章节继续介绍深度学习方法求解上述科学计算问题所需使用到的模块。
网络(Network)
赛桨支持全连接网络(FC)以及初始化网络权重的接口 initialize,支持从文件读取权重进行初始化,以及使用飞桨预置的初始化函数(paddle.nn.initializer)进行初始化。
损失函数(Loss)
赛桨提供了灵活的 Loss 设置方式,允许通过设置独立的权重系数配置多任务 Loss (Loss = w_1 * Loss_equation + w_2 * Loss_ic+ w_3 * loss_bc +w_4 * Loss_data)。
优化器(Optimizer)
赛桨支持 Adam 优化器。
求解器(Solver)及求解模式
求解器提供了控制功能,在训练和推理阶段可以分别使用。
启动自动并行模式
使用自动并行技术实现分布式计算内嵌在赛桨中,无需特别设置,使用如下代码运行程序即可实现数据自动并行。
启动动态图/静态图/自动微分模式
赛桨后端支持动态图和静态图模式,可以通过简单的接口进行切换。另外,在单机静态图模式下,赛桨提供接口启用高阶自动微分。
下一期我们将对支持 PaddleScience 的底层框架技术,如自动微分机制、编译器等功能进行详细介绍,敬请期待~
引用:
[1]《IDC perspective:AI for Science 市场研究》报告正式启动.https://mp.weixin.qq.com/s/gtEbuSULI5fzCIvbDKhkfA
拓展阅读:
相关地址:
1.飞桨 AI for Science 共创计划:
https://www.paddlepaddle.org.cn/science
2.飞桨 PPISG-Science 小组:
https://www.paddlepaddle.org.cn/specialgroupdetail?id=9
评论