一种多阶段的流匹配扩散模型训练方法

本发明涉及一种多阶段的流匹配扩散模型训练方法，包括以下步骤：S1获取训练数据集；S2对所述训练数据集中的图像进行预处理；S3确定预训练扩散模型的推理步数；S4设置超参数；S5从较大的窗口数开始，将采样轨迹划分为多个时间窗口；S6逐步减少时间窗口的数量。本发明的主要创新点在于提出了一种多阶段的流模型训练范式，并优化了流模型的学习目标。通过这种多阶段训练方法，逐步减少时间窗口数量，降低了流匹配扩散模型的学习难度。