Utilizing Adversarial Examples for Bias Mitigation and Accuracy Enhancement
Pushkar Shukla, Dhruv Srikanth, Lee Cohen, Matthew Turk
TL;DR
The paper targets bias in computer vision by introducing Attribute-Specific Adversarial Counterfactuals (ASACs) and a curriculum-based fine-tuning framework that uses ASACs to debias a target classifier while preserving or improving accuracy. It combines adversarial counterfactual generation (via FGSM/PGD) with a two-stage curriculum and an adversarial loss term to guide post-processing debiasing. Across CelebA and LFW, with multiple backbones, the method achieves improvements in fairness metrics such as $DDP$, $DEO$, and $DEOp$ without sacrificing $ACC$, demonstrating robustness and generalization. The work offers a practical, ethics-conscious approach to bias mitigation that minimizes reliance on generator-based counterfactuals and enhances model interpretability through attribution analysis.
Abstract
We propose a novel approach to mitigate biases in computer vision models by utilizing counterfactual generation and fine-tuning. While counterfactuals have been used to analyze and address biases in DNN models, the counterfactuals themselves are often generated from biased generative models, which can introduce additional biases or spurious correlations. To address this issue, we propose using adversarial images, that is images that deceive a deep neural network but not humans, as counterfactuals for fair model training. Our approach leverages a curriculum learning framework combined with a fine-grained adversarial loss to fine-tune the model using adversarial examples. By incorporating adversarial images into the training data, we aim to prevent biases from propagating through the pipeline. We validate our approach through both qualitative and quantitative assessments, demonstrating improved bias mitigation and accuracy compared to existing methods. Qualitatively, our results indicate that post-training, the decisions made by the model are less dependent on the sensitive attribute and our model better disentangles the relationship between sensitive attributes and classification variables.
