Jun, 2024

分数融合:通过 Kullback-Leibler 重心融合基于分数的生成模型

TL;DR研究融合预训练生成模型以增强目标生成模型训练的问题,提出使用 KL 散度加权重心作为最优融合机制,其中重心权重经过最优训练以最小化适合目标群体的适当损失。研究表明,通过使用基于扩散得分方法训练辅助生成模型时,可以高效地计算最优 KL 重心权重。此外,展示了当辅助模型适合其自身任务并且辅助任务组合能很好捕捉目标时,我们的融合方法在总变异距离的样本复杂度上是无维度限制的。我们提供了融合算法的简洁计算实现,并通过涉及混合模型和图像数据集的数值实验在低数据情况下验证其效率。