Beyond the Safety Tax: Mitigating Unsafe Text-to-Image Generation via External Safety Rectification

Read full text
Authors: Xiangtao Meng, Yingkai Dong, Ning Yu, Li Wang, Zheng Li, Shanqing Guo
Cited by: 0

Abstract

Text-to-image (T2I) generative models have achieved remarkable visual fidelity, yet remain vulnerable to generating unsafe content.Existing safety defenses typically intervene internally within the generative model, but suffer from severe concept entanglement, leading to degradation of benign generation quality-a trade-off we term the Safety Tax.To overcome this limitation, we advocate a paradigm shift from destructive internal editing to external safety rectification.Following this principle, we propose SafePatch, a structurally isolated safety module that performs external, interpretable rectification without modifying the base model.The core backbone of SafePatch is architecturally instantiated as a trainable clone of the base model's encoder, allowing it to inherit rich semantic priors and maintain representation consistency.To enable interpretable safety rectification, we construct a strictly aligned counterfactual safety dataset (ACS) for differential supervision training.Across nudity and multi-category benchmarks and recent adversarial prompt attacks, SafePatch achieves robust unsafe suppression (7% unsafe on I2P) while preserving image quality and semantic alignment.

Fields of Study & Subjects

Adversarial Robustness in Machine Learning Generative Adversarial Networks and Image Synthesis Advanced Malware Detection Techniques

Publication details

Journal / source
Findings of the Association for Computational Linguistics: ACL 2026
Publisher
Association for Computational Linguistics
Pages
27986–27998
Publication date
1 January 2026
Language
EN