无需反向传播的分类器引导方法,提升图像生成质量与类别准确性
Gradient-Free Classifier Guidance for Diffusion Model Sampling
- 仅在推理时使用预训练分类器,动态调整每步引导标签和强度
- 在ImageNet上实现23.09的最新FD_DINOv2分数,精度达94.3%
- 可无缝集成至现有方法,不增加计算开销,兼顾保真度与多样性
扩散模型在图像生成中展现出强大的学习能力,能有效捕捉训练数据的完整分布。尽管生成样本变化丰富,但存在保真度不足的问题。已有引导采样方法如分类器引导(CG)和无分类器引导(CFG)虽能提高图像质量,但各有局限:CG需反向传播计算梯度,成本高;而CFG虽高效却弱化了类别对齐。本文提出一种无需梯度下降的新型引导方法——无梯度分类器引导(GFCG),仅在推理阶段使用预训练分类器,于每一步时间步动态确定自适应参考类别标签与引导强度。在条件图像生成与文生图任务中,实验表明该方法持续提升类别预测准确率。此外,GFCG可与其它引导方法(如自动引导,ATG)互补,结合后无额外计算开销,显著提升图像保真度并保持多样性。在ImageNet 512×512数据集上,获得23.09的最新FD_DINOv2得分,同时分类精度达到94.3%,优于ATG的90.2%。
原文摘要 · Abstract (English)
Image generation using diffusion models have demonstrated outstanding learning capabilities, effectively capturing the full distribution of the training dataset. They are known to generate wide variations in sampled images, albeit with a trade-off in image fidelity. Guided sampling methods, such as classifier guidance (CG) and classifier-free guidance (CFG), focus sampling in well-learned high-probability regions to generate images of high fidelity, but each has its limitations. CG is computationally expensive due to the use of back-propagation for classifier gradient descent, while CFG, being gradient-free, is more efficient but compromises class label alignment compared to CG. In this work, we propose an efficient guidance method that fully utilizes a pre-trained classifier without using gradient descent. By using the classifier solely in inference mode, a time-adaptive reference class label and corresponding guidance scale are determined at each time step for guided sampling. Experiments on both class-conditioned and text-to-image generation diffusion models demonstrate that the proposed Gradient-free Classifier Guidance (GFCG) method consistently improves class prediction accuracy. We also show GFCG to be complementary to other guided sampling methods like CFG. When combined with the state-of-the-art Autoguidance (ATG), without additional computational overhead, it enhances image fidelity while preserving diversity. For ImageNet 512$\times$512, we achieve a record $\text{FD}_{\text{DINOv2}}$ of 23.09, while simultaneously attaining a higher classification Precision (94.3%) compared to ATG (90.2%)
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。