Logo image
RAIGen: Rare Attribute Identification in Text-to-Image Generative Models
Conference proceeding   Open access   Peer reviewed

RAIGen: Rare Attribute Identification in Text-to-Image Generative Models

Silpa Vadakkeeveetil Sreelatha, Dan Wang, Serge Belongie, Muhammad Awais Tanvir Rana and Anjan Dutta
Proceedings of the 43rd International Conference on Machine Learning
Proceedings of the 43rd International Conference on Machine Learning (Seoul, South Korea, 06/07/2026–06/07/2026)
2026

Abstract

Computer Science - Learning Bias identification, Bias mitigation, Fairness, Diffusion models
Text-to-image diffusion models achieve impressive generation quality but inherit and amplify training-data biases, skewing coverage of semantic attributes. Prior work addresses this in two ways. Closed-set approaches mitigate biases in predefined fairness categories (e.g., gender, race), assuming socially salient minority attributes are known a priori. Open-set approaches frame the task as bias identification, highlighting majority attributes that dominate outputs. Both overlook a complementary task: uncovering rare or minority features underrepresented in the data distribution (social, cultural, or stylistic) yet still encoded in model representations. We introduce RAIGen, the first framework, to our knowledge, for label-free rare-attribute discovery in diffusion models, requiring no predefined minority categories. RAIGen leverages Matryoshka Sparse Autoencoders and a novel minority metric combining neuron activation frequency with semantic distinctiveness to identify interpretable neurons whose top-activating images reveal underrepre-sented attributes. Experiments show RAIGen discovers attributes beyond fixed fairness categories in Stable Diffusion, scales to larger models such as SDXL, supports systematic auditing across ar-chitectures, and enables targeted amplification of rare attributes during generation. The project page is available at https://vssilpa.github. io/RAIGen_webpage/.
pdf
2602.06806v22.49 MBDownloadView
Published (Version of record) Open Access CC BY V4.0
url
https://icml.cc/View
Conference website

Metrics

1 Record Views

Details

Logo image

Usage Policy