4 个月前

BiaSwap:通过偏差定制的样本交换增强去除数据集偏差

BiaSwap:通过偏差定制的样本交换增强去除数据集偏差

摘要

深度神经网络通常基于数据集中固有的虚假相关性做出决策,在面对无偏数据分布时无法进行有效的泛化。尽管先前的方法预定义了数据集偏差的类型以防止网络学习这些偏差,但在实际数据集中识别偏差类型往往非常困难。本文提出了一种新颖的针对偏差的增强方法——BiaSwap,该方法能够在无需监督偏差类型的情况下学习去偏表示。假设偏差对应于易于学习的属性,我们根据有偏分类器利用这些属性作为捷径的程度对训练图像进行排序,并以无监督的方式将其划分为偏差引导样本和反偏差样本。随后,我们将图像翻译模型中的风格转换模块与这种有偏分类器的类激活图(Class Activation Maps)相结合,从而主要转移由分类器学到的偏差属性。因此,给定一对偏差引导样本和反偏差样本,BiaSwap 生成包含来自反偏差样本的偏差属性的图像,同时保留偏差引导样本中的非相关属性。通过使用这种增强后的图像,BiaSwap 在合成数据集和真实世界数据集上均表现出优于现有基线方法的去偏能力。即使在没有仔细监督的情况下,BiaSwap 也在无偏样本和偏差引导样本上取得了显著的性能表现,这表明模型的泛化能力得到了提升。

基准测试

基准方法指标
action-recognition-on-barBiaSwap
Accuracy: 52.44
facial-attribute-classification-on-bffhqBiaSwap
Bias-Conflicting Accuracy: 58.87

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
BiaSwap:通过偏差定制的样本交换增强去除数据集偏差 | 论文 | HyperAI超神经