Jan, 2024

基于高斯混合模型和负高斯混合梯度的扩散模型

TL;DR利用高斯混合模型作为特征条件引导去噪过程,构建了一种基于高斯混合模型的条件机制,证明了该条件机制在特征上的条件潜在分布相较于类别上的条件潜在分布产生较少的缺陷生成,通过两种基于高斯混合模型的扩散模型的实验结果支持上述发现,并提出负高斯混合梯度作为一种新的梯度函数,通过额外分类器在扩散模型训练中应用,提高了训练稳定性,并从理论上证明了负高斯混合梯度和地球移动距离(Wasserstein 距离)在学习由低维流形支持的分布时具有相同的优势作为一种更合理的代价函数。