融合物理知识的KAN模型在湍流流动预测及CFD(计算流体动力学)集成中的应用

《International Journal of Mechanical Sciences》:Physics-Infused KAN for Turbulent Flow Prediction and CFD Integration

【字体: 时间:2026年01月09日 来源:International Journal of Mechanical Sciences 9.4

编辑推荐:

  湍流场预测模型PI-KAN通过融合物理信息神经网络与Kolmogorov-Arnold网络,结合Transformer编码器高效提取复杂几何特征,显著降低计算参数(仅46,500个可调参数)并加速CFD求解器收敛,验证其在多工况下的高精度与泛化能力。

  
陈欧阳|赵传祺|刘宏|张慧英|马天鸽|徐胜利
中国大连理工大学能源与动力工程学院,教育部海洋能源利用与节能重点实验室,大连,116024

摘要

计算流体动力学(CFD)已成为模拟湍流流动不可或缺的工具,然而求解雷诺平均纳维-斯托克斯(RANS)方程仍然耗时、内存需求高且计算成本高昂。为应对这一挑战,提出了一种新的预测模型——物理信息Kolmogorov-Arnold网络(PI-KAN)模型,该模型能够有效地从CFD结果中学习湍流场表示。该模型结合了物理信息神经网络(PINN)框架和注意力机制与Kolmogorov-Arnold网络(KAN)。在PI-KAN中,为了高效提取复杂几何信息,编码器策略性地整合了KAN和多头注意力机制,形成了Kansformer架构。这使得PI-KAN能够直接根据给定的流动条件和几何形状预测流场,同时结合物理约束以提高准确性并降低计算成本。验证表明,PI-KAN仅使用46,500个可训练参数即可实现高精度预测,而MLP需要548,460个参数。当其预测结果作为CFD求解器的初始条件时,收敛过程显著加快,并且该模型对未见过的流动配置具有很强的泛化能力。这些结果表明,PI-KAN可以作为加速复杂工程应用中CFD模拟的有效且准确的替代模型。

引言

随着数学理论[1]、数据科学[2]和高性能计算[3]的发展,计算流体动力学(CFD)在航空航天[4]、能源与动力科学[5,6]以及交通运输行业[7]中发挥着关键作用。尽管其在机械工程中的广泛应用至关重要,从航空航天湍流到能源系统优化,但复杂的模拟需要数十亿个网格点和数千个高性能计算核心,从而导致巨大的计算成本和延长的设计周期[[8], [9], [10], [11], [12]]。CFD技术已逐渐发展成为大型设备数字工程的基础和关键支持工具[13]。CFD技术主要通过求解非线性纳维-斯托克斯(NS)方程来获取流场信息,这需要大量的计算资源[14]。因此,迫切需要一种高效的流场预测方法,特别是在当前探索AI和GPU技术以缓解这些计算挑战的情况下[15,16]。
为了解决计算挑战,数据驱动的深度学习方法在流体力学领域取得了显著进展[[17], [18], [19], [20], [21]]。这些方法通过学习大型数据集来加速流场预测。卷积神经网络(CNN)和生成对抗网络(GAN)在流场重建和湍流建模方面表现出色,提高了模拟精度[[22], [23], [24]]。实证研究表明,数据驱动的深度学习在机械工程应用中显著提高了计算效率,例如双模燃烧器中的流场[25,26]、多孔介质中的流动与传输[27,28]、多个钝体周围的流动[29,30]、将神经网络与物理求解器耦合的流场预测[31]以及海洋工程中的速度流场[32,33],将模拟时间缩短了几倍,并解决了高维数据问题。与传统CFD方法相比,数据驱动模型显著提高了计算效率,初步研究验证了人工智能在加速机械科学领域科学计算方面的有效性[[34], [35], [36], [37], [38]]。
尽管效率有所提高,但数据驱动模型往往缺乏物理约束,可能导致物理结果不一致[39,40]。物理损失函数解决了这一问题[41]。机械科学领域的最新研究表明,物理信息神经网络通过自适应优化,在流场预测应用中将预测误差降低了多达30%,例如多孔介质中的流体流动[42,43]、氢泄漏中的空间扩散[44,45]、两相流中的不稳定性[46,47]、层流和湍流火焰中的反应性流动[48,49]以及二维圆柱流动[50,51]。物理信息神经网络(PINNs)将流体力学方程嵌入损失函数中,确保了物理一致性并提高了性能[[52], [53], [54]]。
在工业中,流场的边界往往很复杂,例如不同的翼型或航空发动机几何形状。将复杂的几何特征信息作为训练数据来预测变几何形状的流场是一个重要的考虑因素[55]。一些研究人员使用CNN或Transformer从机翼图像中提取几何特征[[56], [57], [58]],然后将提取的几何参数与雷诺数和攻角作为输入,传递给多层感知器(MLP)网络来预测流场[59]。CNN和Transformer已成为图像分割和扩散概率模型等各种视觉应用的基石,但实证研究表明,这些网络仍然局限于线性建模,在流体力学预测中缺乏可解释性[[60], [61], [62]]。
Kolmogorov-Arnold表示定理类似于MLP,用于近似多变量连续函数[[63], [64], [65]]。然而,浅层Kolmogorov网络在流体力学模拟等复杂任务中的效果较差[66]。最近,刘等人[67]提出了Kolmogorov-Arnold网络(KAN),通过学习激活函数来提高非线性建模的性能[68]。KAN使用B样条作为激活函数,所需参数较少,并且与偏微分方程的数值解保持一致,使其适用于机械工程中的流场预测[69,70]。将KAN与U-Net和Transformer结合使用,可以增强变几何形状流场的非线性特征表示和可解释性[71,72]。
将KAN与PINN结合使用以替代传统的MLP有望获得更好的结果[73,74]。本文提出了PI-KAN模型,该模型将物理信息与KAN架构相结合。KAN网络进一步与Transformer编码器结合,以提取流场的复杂几何特征。使用带有障碍物的后退步流动作为示例案例,与以往的研究[75]相比,PI-KAN显示出更好的训练稳定性和收敛性。为了提高流场预测的可靠性,将预测值作为初始条件,以促进Navier-Stokes方程的Fluent求解器的迭代收敛。这种方法不仅加速了CFD求解器的收敛速度,还提高了其泛化性能。
本文的结构如下:第2节介绍了本研究的背景和采用的神经网络架构。第3节介绍了用于训练PI-KAN的数据集设置。第4节展示了PI-KAN神经网络模型的训练和测试结果。第5节表明,PI-KAN预测为Fluent提供了高质量的初始条件,并比较了有无深度学习整合时Fluent的收敛时间和迭代次数。第6节给出了结论。

部分摘录

基于深度学习的流场预测模型

第2节详细探讨了基于深度学习的流场预测模型。首先介绍了受Kolmogorov-Arnold表示定理启发的Kolmogorov-Arnold网络(KAN)模型,然后讨论了如何利用这种架构构建有效的流场预测模型。讨论逐步进行,解释了模型的各个组成部分,包括

数据库建立方法和预处理

在图6中,为了评估模型对复杂流动模式的预测能力,选择了五种不同的阶梯障碍模型(椭圆1、椭圆2、圆形、正方形1、正方形2)进行阶梯流动的训练和测试。雷诺数范围从510000到810000,共计155个案例。变障碍模型的阶梯高度为

= 25毫米,h1 = 94毫米,计算区域为w1 = 50毫米(在阶梯前方)到w2 = 508毫米(在阶梯后方)。为了确保可靠性

PI-KAN训练和测试的结果与讨论

第4节展示了PI-KAN模型在训练和测试阶段的结果和性能。首先检查了不同的模型配置及其相应的性能,以确定模型的最佳设置。然后进入测试阶段,使用未参与训练的流场数据评估PI-KAN的预测能力。本节提供了对该模型的见解

人工智能辅助的数值模拟

第5节探讨了将人工智能(AI)与计算流体动力学(CFD)相结合,以展示其在提高流场计算效率和准确性方面的潜力。首先介绍了AI+CFD集成模型,其中PI-KAN模型用于为CFD求解器提供高保真度的初始条件。然后在新生成的自适应细化网格上对该方法进行了测试,结果表明该模型保持了其强大的加速能力

结论

本研究介绍了物理信息Kolmogorov-Arnold网络(PI-KAN),这是一种用于预测稳态条件下二维不可压缩湍流场的新模型。PI-KAN基于Kolmogorov-Arnold网络(KAN)架构构建,结合了物理信息损失函数和动态权重,从而提高了预测准确性和在不同流动场景下的泛化能力。此外,Kansformer编码器能够高效提取

CRediT作者贡献声明

陈欧阳:写作 – 审稿与编辑,撰写原始稿件,可视化,验证,监督,软件,资源管理,项目管理,方法论,调查,形式分析,数据管理,概念化。赵传祺:可视化。刘宏:写作 – 审稿与编辑,监督,项目管理,资金获取。张慧英:写作 – 审稿与编辑。马天鸽:可视化。徐胜利:项目管理,资金获取。
相关新闻
生物通微信公众号
微信
新浪微博
  • 搜索
  • 国际
  • 国内
  • 人物
  • 产业
  • 热点
  • 科普
  • 急聘职位
  • 高薪职位

知名企业招聘

热点排行

    今日动态 | 人才市场 | 新技术专栏 | 中国科学人 | 云展台 | BioHot | 云讲堂直播 | 会展中心 | 特价专栏 | 技术快讯 | 免费试用

    版权所有 生物通

    Copyright© eBiotrade.com, All Rights Reserved

    联系信箱:

    粤ICP备09063491号