arXiv:2411.16079cs.CVcs.AI2024-11

用扩散模型生成反偏见图像,提升分类器鲁棒性

Debiasing Classifiers by Amplifying Bias with Latent Diffusion and Large Language Models

论文配图:Debiasing Classifiers by Amplifying Bias with Latent Diffusion and Large Language Models
图 1 · 摘自论文原文
  • 用偏差分类器的高损失样本指导扩散模型生成反偏见图像
  • 在多个基准数据集上实现当前最佳去偏效果
  • 无需训练即可生成,适合关注模型公平性的研究者

神经网络在图像分类中常因学习到错误关联而产生偏差,影响泛化能力。现有方法依赖属性标签或生成对抗网络(GAN)缓解偏差。本文提出DiffuBias,一种基于文本到图像生成的新框架,通过预训练扩散模型和图像描述模型,利用偏差分类器$f_B$的前$K$个最高损失样本生成具有冲突偏差的图像,从而增强分类器鲁棒性,且生成阶段无需微调。实验表明,该方法在多个基准数据集上达到当前最优去偏性能。此外,我们对比了多种生成模型的碳排放与能耗,强调计算效率的重要性。

原文摘要 · Abstract (English)

Neural networks struggle with image classification when biases are learned and misleads correlations, affecting their generalization and performance. Previous methods require attribute labels (e.g. background, color) or utilizes Generative Adversarial Networks (GANs) to mitigate biases. We introduce DiffuBias, a novel pipeline for text-to-image generation that enhances classifier robustness by generating bias-conflict samples, without requiring training during the generation phase. Utilizing pretrained diffusion and image captioning models, DiffuBias generates images that challenge the biases of classifiers, using the top-$K$ losses from a biased classifier ($f_B$) to create more representative data samples. This method not only debiases effectively but also boosts classifier generalization capabilities. To the best of our knowledge, DiffuBias is the first approach leveraging a stable diffusion model to generate bias-conflict samples in debiasing tasks. Our comprehensive experimental evaluations demonstrate that DiffuBias achieves state-of-the-art performance on benchmark datasets. We also conduct a comparative analysis of various generative models in terms of carbon emissions and energy consumption to highlight the significance of computational efficiency.

去偏见扩散模型图像生成

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。