让AI画画更公平安全,且不损失图像质量。
RespoDiff: Dual-Module Bottleneck Transformation for Responsible & Faithful T2I Generation
- 双模块设计:一个管责任概念,一个保语义对齐。
- 在未见过的提示下,生成质量提升20%。
- 可无缝接入SDXL等大模型,适合实际部署。
扩散模型在文本到图像生成中实现了高保真与语义丰富性,但公平性与安全性仍面临挑战。现有方法常以牺牲语义保真度和图像质量为代价来提升责任性。本文提出RespoDiff,一种基于扩散模型中间瓶颈表示的双模块变换框架。该方法引入两个可学习模块:一个专注于捕捉并强制执行公平性、安全性等责任概念,另一个则致力于保持与中性提示的语义对齐。为促进双模块协同训练,提出新型评分匹配目标。实验表明,该方法在未见提示下实现责任生成与语义一致性提升20%,同时不损害图像保真度。该框架可无缝集成至SDXL等大规模模型,显著增强生成内容的公平性与安全性。代码将在接受后公开。
原文摘要 · Abstract (English)
The rapid advancement of diffusion models has enabled high-fidelity and semantically rich text-to-image generation; however, ensuring fairness and safety remains an open challenge. Existing methods typically improve fairness and safety at the expense of semantic fidelity and image quality. In this work, we propose RespoDiff, a novel framework for responsible text-to-image generation that incorporates a dual-module transformation on the intermediate bottleneck representations of diffusion models. Our approach introduces two distinct learnable modules: one focused on capturing and enforcing responsible concepts, such as fairness and safety, and the other dedicated to maintaining semantic alignment with neutral prompts. To facilitate the dual learning process, we introduce a novel score-matching objective that enables effective coordination between the modules. Our method outperforms state-of-the-art methods in responsible generation by ensuring semantic alignment while optimizing both objectives without compromising image fidelity. Our approach improves responsible and semantically coherent generation by 20% across diverse, unseen prompts. Moreover, it integrates seamlessly into large-scale models like SDXL, enhancing fairness and safety. Code will be released upon acceptance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。