FairGen: Controlling Sensitive Attributes for Fair Generations in Diffusion Models via Adaptive Latent Guidance

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Kang, Mintong, Kumar, Vinayshekhar Bannihatti, Roy, Shamik, Kumar, Abhishek, Khosla, Sopan, Narayanaswamy, Balakrishnan Murali, Gangadharaiah, Rashmi
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912663335862272
author Kang, Mintong
Kumar, Vinayshekhar Bannihatti
Roy, Shamik
Kumar, Abhishek
Khosla, Sopan
Narayanaswamy, Balakrishnan Murali
Gangadharaiah, Rashmi
author_facet Kang, Mintong
Kumar, Vinayshekhar Bannihatti
Roy, Shamik
Kumar, Abhishek
Khosla, Sopan
Narayanaswamy, Balakrishnan Murali
Gangadharaiah, Rashmi
contents Text-to-image diffusion models often exhibit biases toward specific demographic groups, such as generating more males than females when prompted to generate images of engineers, raising ethical concerns and limiting their adoption. In this paper, we tackle the challenge of mitigating generation bias towards any target attribute value (e.g., "male" for "gender") in diffusion models while preserving generation quality. We propose FairGen, an adaptive latent guidance mechanism which controls the generation distribution during inference. In FairGen, a latent guidance module dynamically adjusts the diffusion process to enforce specific attributes, while a memory module tracks the generation statistics and steers latent guidance to align with the targeted fair distribution of the attribute values. Furthermore, we address the limitations of existing datasets by introducing the Holistic Bias Evaluation (HBE) benchmark, which covers diverse domains and incorporates complex prompts to assess bias more comprehensively. Extensive evaluations on HBE and Stable Bias datasets demonstrate that FairGen outperforms existing bias mitigation approaches, achieving substantial bias reduction (e.g., 68.5% gender bias reduction on Stable Diffusion 2). Ablation studies highlight FairGen's ability to flexibly control the output distribution at any user-specified granularity, ensuring adaptive and targeted bias mitigation.
format Preprint
id arxiv_https___arxiv_org_abs_2503_01872
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle FairGen: Controlling Sensitive Attributes for Fair Generations in Diffusion Models via Adaptive Latent Guidance
Kang, Mintong
Kumar, Vinayshekhar Bannihatti
Roy, Shamik
Kumar, Abhishek
Khosla, Sopan
Narayanaswamy, Balakrishnan Murali
Gangadharaiah, Rashmi
Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
Text-to-image diffusion models often exhibit biases toward specific demographic groups, such as generating more males than females when prompted to generate images of engineers, raising ethical concerns and limiting their adoption. In this paper, we tackle the challenge of mitigating generation bias towards any target attribute value (e.g., "male" for "gender") in diffusion models while preserving generation quality. We propose FairGen, an adaptive latent guidance mechanism which controls the generation distribution during inference. In FairGen, a latent guidance module dynamically adjusts the diffusion process to enforce specific attributes, while a memory module tracks the generation statistics and steers latent guidance to align with the targeted fair distribution of the attribute values. Furthermore, we address the limitations of existing datasets by introducing the Holistic Bias Evaluation (HBE) benchmark, which covers diverse domains and incorporates complex prompts to assess bias more comprehensively. Extensive evaluations on HBE and Stable Bias datasets demonstrate that FairGen outperforms existing bias mitigation approaches, achieving substantial bias reduction (e.g., 68.5% gender bias reduction on Stable Diffusion 2). Ablation studies highlight FairGen's ability to flexibly control the output distribution at any user-specified granularity, ensuring adaptive and targeted bias mitigation.
title FairGen: Controlling Sensitive Attributes for Fair Generations in Diffusion Models via Adaptive Latent Guidance
topic Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2503.01872