Jun, 2024

基于扩散模型的流场生成预测

TL;DR我们提出了一种几何到流场扩散模型,利用障碍物形状预测障碍物后的流场。该模型基于可学习的马尔可夫转移核函数,从高斯分布中恢复数据分布。马尔可夫过程以障碍物几何形状为条件,在每一步估计需要消除的噪声,通过 U-Net 实现。交叉注意机制将几何形状作为提示集成进来。我们使用包括圆形、椭圆形、矩形和三角形在内的简单障碍物流数据集进行几何到流场扩散模型的训练,并以同一数据集训练 CNN 模型进行比较。对于简单和复杂几何形状的障碍物流场,进行插值和外推的测试。流场生成结果表明几何到流场扩散模型在预测瞬时流场和处理复杂几何形状方面优于 CNN 模型。对模型准确性和场中发散的定量分析表明扩散模型具有很高的鲁棒性,暗示扩散模型隐式地学习了物理规律。