Search papers, labs, and topics across Lattice.
This study critiques the prevailing methods for safety alignment in text-to-image diffusion models, revealing that high utility claims are often based on coarse global metrics that overlook fine-grained semantic accuracy. By employing structured evaluations, the authors demonstrate that safety-aligned models significantly underperform in semantic fidelity, particularly in object counts and relationships. To address this issue, they introduce StructureAware Geometric Regularization (SAGE), which effectively enhances structured utility while preserving safety and achieving a notable improvement in performance metrics.
Safety-aligned text-to-image models may appear effective, but they suffer substantial semantic fidelity losses that standard metrics fail to capture.
Safety alignment of text-to-image (T2I) diffusion models aims to suppress harmful generations while preserving utility on benign prompts. Recent methods often appear to deliver high safety with high utility, but this conclusion rests largely on coarse global utility metrics (e.g., FID, CLIPScore) that are insensitive to fine-grained semantic correctness, creating an illusion of high utility. We show that when utility is measured with structured evaluation, this illusion breaks: on TIFA (Text-to-Image Faithfulness evaluation with Question Answering), safety-aligned models suffer substantial drops in semantic fidelity, including failures in object counts, attributes, and relationships. To diagnose the source of this gap, we analyze the text-encoder prompt embedding space and uncover semantic collapse, a contraction of embedding spread coupled with distortion of inter-prompt similarity structure, which strongly correlates with structured utility loss. Guided by this insight, we propose StructureAware Geometric Regularization (SAGE), a safety alignment objective that explicitly preserves embedding spread and inter-prompt relational structure during adaptation. Our method restores structured utility (TIFA +5.0% over prior state-of-the-art) while maintaining strong safety performance and competitive coarse-grained utility scores. Our source code and trained models are available at https://adeelyousaf.github.io/SAGE_ECCV26_Project_Page/.
