一种流匹配生成模型的训练方法、推理方法及相关装置

本发明实施例提供了一种流匹配生成模型的训练方法、推理方法及相关装置，用于提升训练后的流匹配生成模型所预测的动作序列的准确性。本发明实施例方法包括：获取噪声、时刻t及第一动作的环境特征，第一动作的环境特征至少包括第一动作的观测值；将噪声、时刻t及环境特征输入至初始化的流匹配生成模型，以得到输出的t+1时刻的条件概率路径的预测速度场向量；利用预设的损失函数计算预测速度场向量与真实速度场向量之间的损失，预设的损失函数包括第一损失函数和第二损失函数中的至少一项，以及第三损失函数；利用损失和反向传播算法，对初始化的流匹配生成模型进行训练，直至流匹配生成模型收敛为止，以得到训练后的流匹配生成模型。