Epistemic Uncertainty-Weighted Loss for Visual Bias Mitigation

Abstract

Deep neural networks are highly susceptible to learning biases in visual data. While various methods have been proposed to mitigate such bias, the majority require explicit knowledge of the biases present in the training data in order to mitigate. We argue the relevance of exploring methods which are completely ignorant of the presence of any bias, but are capable of identifying and mitigating them. Furthermore, we propose using Bayesian neural networks with an epistemic uncertainty-weighted loss function to dynamically identify potential bias in individual training samples and to weight them during training. We find a positive correlation between samples subject to bias and higher epistemic uncertainties. Finally, we show the method has potential to mitigate visual bias on a bias benchmark dataset and on a real-world face detection problem, and we consider the merits and weaknesses of our approach.

Cite

Text

Stone et al. "Epistemic Uncertainty-Weighted Loss for Visual Bias Mitigation." IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops, 2022. doi:10.1109/CVPRW56347.2022.00327

Markdown

[Stone et al. "Epistemic Uncertainty-Weighted Loss for Visual Bias Mitigation." IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops, 2022.](https://mlanthology.org/cvprw/2022/stone2022cvprw-epistemic/) doi:10.1109/CVPRW56347.2022.00327

BibTeX

@inproceedings{stone2022cvprw-epistemic,
  title     = {{Epistemic Uncertainty-Weighted Loss for Visual Bias Mitigation}},
  author    = {Stone, Rebecca S. and Ravikumar, Nishant and Bulpitt, Andrew J. and Hogg, David C.},
  booktitle = {IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops},
  year      = {2022},
  pages     = {2897-2904},
  doi       = {10.1109/CVPRW56347.2022.00327},
  url       = {https://mlanthology.org/cvprw/2022/stone2022cvprw-epistemic/}
}