Generative Counterfactual Augmentation for Bias Mitigation

Jason Uwaeze1, Pranav Kulkarni,2, Vladimir Braverman3, Michael A. Jacobs4, Vishwa Parekh4,

1Rice University    2University of Maryland    3Johns Hopkins University    4UTHealth Houston

1ju6@rice.edu, 4vishwa.s.parekh@uth.tmc.edu

📌 Conference: International Conference on Conmputer Vision, 2025 (Poster)

📌 Presented at Computer Vision for Automated Medical Diagnosis (CVAMD) workshop 2025, Hawaii, United States.

Paper Code Dataset Models BibTex
Architecture Diagram

Figure: Overview of the GCA Framework

Abstract

Deep learning (DL) models trained for chest x-ray (CXR) classification can encode protected demographic attributes and exhibit bias towards underrepresented patient populations. In this work, we propose Generative Counterfactual Augmentation (GCA), a framework for mitigating algorithmic bias through demographic-complete augmentation of training data. We use a StyleGAN3-based synthesis network and SVM-guided latent space traversal to generate structured age and sex counterfactuals for each CXR while preserving disease features. We extensively evaluate GCA for training DL models with the RSNA Pneumonia dataset using controlled underdiagnosis bias injection across age- and sex-groups at varying rates. Our results show up to 23% reduction in FNR disparity, with a mean reduction of 9%, across varying rates of underdiagnosis bias. When evaluated with the external CheXpert and MIMIC-CXR datasets, we observe a consistent FNR reduction and improved model generalizability. We demonstrate that GCA is an effective strategy for mitigating algorithmic bias in DL models for medical imaging, ensuring trustworthiness in clinical settings.

Introduction

TODO: Add Introduction

Methodology

TODO: Add Methodology

Results

TODO: Add Results

Image Gallery

Demo Video

Demo Videos List

Conclusion

GCA offers a scalable and effective framework for mitigating algorithmic bias in DL models through structured counterfactual generation. Our findings suggest that GCA not only improves model fairness and robustness but also has the potential to be adapted for other imaging modalities and tasks, ensuring trustworthiness in clinical settings.

BibTeX

        
        
          @inproceedings{uwaeze2025generative,
            title={Generative Counterfactual Augmentation for Bias Mitigation},
            author={Uwaeze, Jason and Kulkarni, Pranav and Braverman, Vladimir and Jacobs, Michael A and Parekh, Vishwa S},
            booktitle={Proceedings of the IEEE/CVF international conference on computer vision},
            pages={[pages]},
            year={2025},
          }
        
      

Acknowledgement

TODO: Add Acknowledgement if applicable or remove this