清华团队研发AI驱动PDE智能计算框架并开创预训练有限元新方法
清华新闻网6月2日电 偏微分方程(PDEs)作为刻画复杂物理系统演变规律的核心数学工具,也是航空航天、先进制造、能源装备、生物医学工程等领域开展数值模拟与工程设计的重要基础。传统有限元、有限体积、有限差分等数值方法在过往数十年间取得了显著成就,但在处理复杂几何、强非线性、多尺度、多物理场及高维参数空间等问题时,仍面临计算开销大、重复求解耗时长、数据与物理融合困难等瓶颈。近年来,人工智能与科学计算深度交叉融合形成的AI for Science正在引领科学研究范式的深刻变革。其中,运用人工智能技术求解偏微分方程的AI for PDEs,已成为智能计算力学与科学机器学习领域的核心前沿方向。
近日,清华大学航天航空学院刘应华教授团队围绕AI for PDEs在计算力学中的理论框架、方法体系与工程应用展开系统性研究。一方面,团队全面梳理了AI for PDEs在计算力学领域的发展脉络,整合了物理信息神经网络、深度能量法、神经算子及物理信息神经算子等主流算法,并展望其在固体力学、流体力学和生物力学中的应用前景;另一方面,团队创新性提出预训练有限元方法PFEM,将物理信息神经算子与经典有限元求解器有机结合,探索了面向未来智能仿真的"预训练模型+经典数值求解器"新范式。
在综述研究中,团队指出当前AI for PDEs在计算力学中主要形成三类代表性范式。第一类为物理信息神经网络PINNs,通过将控制方程、边界条件和初始条件嵌入损失函数,使神经网络在满足物理约束的条件下逼近PDE解。第二类为深度能量法DEM,基于变分原理和最小势能原理构建能量泛函,尤其适用于弹性力学、断裂力学等具有能量结构特征的问题。第三类为神经算子和物理信息神经算子,前者学习一类PDE问题从输入函数到解函数之间的映射关系,后者进一步将控制方程引入神经算子训练流程,从而在降低数据需求的同时增强物理一致性。
图1、AI for PDEs智能计算力学框架在AI for Science中的定位
图2、智能计算力学主要方法体系:PINNs、Operator Learning和Physics-informed Neural Operator
综述进一步阐述,AI for PDEs对计算力学的价值体现在两个维度。首先,它可作为传统数值仿真的加速器,通过代理模型、神经算子和混合求解框架降低重复求解成本。其次,它也为科学发现提供了崭新工具,例如从实验或仿真数据中识别未知参数、反演本构关系、提取控制规律等。
基于此,团队创新性提出预训练有限元方法PFEM。该方法以Transolver架构为基础,构建基于物理信息神经算子的预训练模型,并将其预测结果作为经典有限元迭代求解器的初始解。
图3、预训练有限元方法PFEM的整体框架。PFEM由物理驱动预训练阶段和warm-start迭代求解阶段组成:前者通过Transolver-based PINO学习一类PDE问题的快速近似解,后者将该近似解作为有限元求解器初始值,从而获得高精度结果并减少迭代成本
与传统神经算子通常依赖结构化网格不同,PFEM直接采用非结构点云作为输入,将空间坐标、几何信息、材料参数和边界条件统一编码,从而能够更灵活地处理复杂几何问题。与此同时,PFEM通过显式有限元微分构造PDE约束,避免了完全依赖自动微分带来的额外开销。这一设计使PFEM能够自然嵌入现有有限元框架,并为复杂工程结构的快速预测和高精度校正提供统一接口。
图4、PFEM在线弹性验证结果
图5、PFEM在非线性超弹性问题中的验证结果
图6、PFEM在三维均匀化问题中的验证结果
这一系列成果展示了AI for PDEs在智能计算力学中的系统性发展路径:从理论综述、方法分类和应用图谱出发,进一步走向与经典数值方法深度融合的原创算法框架。
综述成果以"人工智能在计算力学偏微分方程中的应用综述"(Artificial intelligence for partial differential equations in computational mechanics: A review)为题,发表于《应用力学评论》(Applied Mechanics Reviews)。预训练有限元以"预训练有限元方法:基于物理信息神经算子的偏微分方程预训练与热启动框架"(Pretrain finite element method: A pretraining and warm-start framework for PDEs via physics-informed neural operators)为题,发表于《固体力学与物理学杂志》(Journal of the Mechanics and Physics of Solids)。
清华大学航天航空学院2024级博士生王一铮为两项工作的独立第一作者,清华大学航天航空学院教授刘应华为相关工作的通讯作者。研究得到国家自然科学基金重点项目的资助。
论文链接:
AI for PDEs综述:
https://doi.org/10.1115/1.4071710
PFEM论文:
https://doi.org/10.1016/j.jmps.2026.106682