发现文生图模型偏见根源并提出轻量级修复方案
FairFlow: Demystifying and Mitigating Stereotype Bias in Text-to-Image Diffusion Transformers

- 定位文生图模型中关键的偏见传播层,揭示其分阶段作用机制
- 在特定层级注入公平方向,实现性别、种族等多维度偏见中和
- 无需重训练,推理开销极低,适合大规模部署系统
多模态扩散变换器(MM-DiTs)已成为现代文生图系统的核心架构,但其存在严重的对齐漏洞,即使在无害提示下也会系统性地表现出严重刻板印象偏见,带来算法歧视风险。现有缓解策略多针对旧式U-Net结构,对MM-DiTs的精准修复仍是开放挑战。本文通过机制分析发现,偏见表征在深度上并非均匀分布,而是由少数稀疏层作为内部语义绑定枢纽所介导。这些枢纽呈现阶段性传播:早期枢纽建立易受偏见影响的结构模板,中期枢纽从文本条件中主动提取核心刻板概念,晚期枢纽通过视觉自注意力全局固化偏见。基于此架构洞察,我们提出FairFlow——一种内在、机制引导的缓解框架。该框架通过稀疏导向方式,在限定推理窗口内学习属性特异的公平方向,并仅注入至识别出的语义枢纽。在FLUX.1-dev与Stable Diffusion~3上的评估表明,FairFlow有效中和了性别、种族及交叉性设置下的刻板偏见,实现公平性与保真度的最优平衡。具有近乎零的推理开销,且对复杂提示保持鲁棒性,为大规模部署的MM-DiT系统提供轻量实用的偏见缓解方案。代码与数据集将在论文录用后公开。
原文摘要 · Abstract (English)
Multimodal diffusion transformers (MM-DiTs) have emerged as the prevalent backbone for modern text-to-image generation systems. However, they exhibit critical alignment vulnerabilities, systematically manifesting severe stereotype biases even under benign prompts. This poses a significant risk of algorithmic discrimination in deployed systems. Since most existing mitigation strategies were tailored for legacy U-Net architectures, the precise remediation of these vulnerabilities in MM-DiTs remains a critical open challenge. In this work, we first investigate the root cause of this vulnerability via mechanistic analysis. We reveal that bias representations in MM-DiTs are not uniformly distributed across depth, but are mediated by a sparse set of layers functioning as internal semantic binding hubs. These hubs exhibit a stage-wise propagation driving bias manifestation: early hubs establish the structural templates susceptible to bias, middle hubs actively extract core stereotypical concepts from textual conditioning, and late hubs globally solidify these biases through visual self-attention. Leveraging these architectural insights, we propose FairFlow, an intrinsic, mechanism-guided mitigation framework. FairFlow acts as an internal regulator by employing sparse steering: it learns attribute-specific fair directions and injects them exclusively at the identified semantic hubs within a constrained inference window. Evaluations on FLUX.1-dev and Stable Diffusion~3 demonstrate that FairFlow effectively neutralizes these stereotypical vulnerabilities across gender, race, and intersectional settings, achieving an optimal fairness-fidelity balance. With near-zero inference overhead and robustness to complex prompts, FairFlow provides a lightweight and practical bias mitigation for large-scale deployed MM-DiT systems. Code and datasets will be publicly released upon acceptance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。