一种基于轻量级条件扩散模型的流体力场仿真预测方法

本发明公开一种基于轻量级条件扩散模型的流体力场仿真预测方法，包括以下步骤：S1、获取流体力场图像数据集，并划分为训练集、验证集和测试集；S2、基于条件扩散神经网络构建流体力场仿真模型的基础架构，通过所述训练集进行模型预训练；S3、基于多层级剪枝对步骤S2预训练后的流体力场仿真模型进行轻量化改进，所述多层级剪枝包括卷积核剪枝和通道剪枝；S4、通过所述训练集对于步骤S3轻量化改进后的流体力场仿真模型中未剪枝参数进行微调，并在微调过程中结合所述验证集监控模型性能变化，待微调完整后，通过所述测试集计算模型的流体力场预测性能指标，筛选出性能最优的流体力场仿真模型进行流体力场的预测生成。