Nov, 2023

基于自蒸馏的微调方法拓展数据有限的扩散模型的表达能力

TL;DR我们提出了自我蒸馏 Fine-Tuning 扩散模型(SDFT),通过利用在大型源数据集上预训练的扩散模型的多样特征,从源模型中提取出更一般的特征(形状、颜色等),少量的领域特定特征(纹理、细节等),以在目标数据集上进行知识传递且不干扰训练过程,以引导有限数据集上扩散模型的生成能力,从而增强了模型的表达能力,并在各种下游任务中显示出更好的生成能力。