English

DiffInject: Revisiting Debias via Synthetic Data Generation using Diffusion-based Style Injection

Computer Vision and Pattern Recognition 2024-06-11 v1 Artificial Intelligence Machine Learning

Abstract

Dataset bias is a significant challenge in machine learning, where specific attributes, such as texture or color of the images are unintentionally learned resulting in detrimental performance. To address this, previous efforts have focused on debiasing models either by developing novel debiasing algorithms or by generating synthetic data to mitigate the prevalent dataset biases. However, generative approaches to date have largely relied on using bias-specific samples from the dataset, which are typically too scarce. In this work, we propose, DiffInject, a straightforward yet powerful method to augment synthetic bias-conflict samples using a pretrained diffusion model. This approach significantly advances the use of diffusion models for debiasing purposes by manipulating the latent space. Our framework does not require any explicit knowledge of the bias types or labelling, making it a fully unsupervised setting for debiasing. Our methodology demonstrates substantial result in effectively reducing dataset bias.

Keywords

Cite

@article{arxiv.2406.06134,
  title  = {DiffInject: Revisiting Debias via Synthetic Data Generation using Diffusion-based Style Injection},
  author = {Donggeun Ko and Sangwoo Jo and Dongjun Lee and Namjun Park and Jaekwang Kim},
  journal= {arXiv preprint arXiv:2406.06134},
  year   = {2024}
}

Comments

10 pages (including supplementary), 3 figures, SynData4CV@CVPR 24 (Workshop)

R2 v1 2026-06-28T16:59:22.636Z