Rethinking Memorization Mitigation in Diffusion Models: Reinforcing Text Conditioning
On the map: Diffusion models → Image generation, text-to-image and editing
Abstract (arXiv)
Text-to-image diffusion models have achieved remarkable progress in image synthesis, yet can exhibit memorization by closely reproducing individual training examples. Effective mitigation must preserve useful prompt information to guide alternative depictions. We introduce a training-free method that redistributes cross-attention with Gaussian smoothing before reinforcing content-token contributions and attenuating padding contributions, without additional denoiser evaluations. With this intervention, stronger content conditioning can improve prompt alignment at comparable training-image similarity. A local analysis identifies when reinforcement preserves shared value information while redistribution reduces localized attention mass. On Stable Diffusion v1.4 and v2.0, all evaluated smoothing widths lie on the empirical Pareto frontiers for training-image similarity versus both prompt alignment and image preference. A configuration selected on Stable Diffusion reduces template reproduction in DeepFloyd IF without further tuning. These findings support jointly controlling conditioning allocation and strength to generate prompt-consistent alternatives.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Image generation, text-to-image and editing, one of 14 directions in Diffusion models. Builds on 30 of its 30 references that are in the map. First posted on arXiv 1 Oct 2026.
Builds on
30 of 30 references.
Closest ancestors · 7
Same direction, newest first
- 2024Towards Understanding the Working Mechanism of Text-to-Image Diffusion Model24 May
- 2023Linguistic Binding in Diffusion Models: Enhancing Attribute Correspondence through Attention Map Alignment15 Jun
- 2023Attend-and-Excite: Attention-Based Semantic Guidance for Text-to-Image Diffusion Models31 Jan
- 2022Training-Free Structured Diffusion Guidance for Compositional Text-to-Image Synthesis9 Dec
- 2022Diffusion Art or Digital Forgery? Investigating Data Replication in Diffusion Models7 Dec
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understanding23 May
- 2022Hierarchical Text-Conditional Image Generation with CLIP Latents13 Apr
Other references · 23
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
- 2020Denoising Diffusion Implicit Modelscited by 9,655
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2022Classifier-Free Diffusion Guidancecited by 5,717
- 2022LAION-5B: An open large-scale dataset for training next generation image-text modelscited by 4,131
- 2022Prompt-to-Prompt Image Editing with Cross Attention Controlcited by 2,409
- 2022Reproducible scaling laws for contrastive language-image learningcited by 1,163
- 2023ImageReward: Learning and Evaluating Human Preferences for Text-to-Image Generationcited by 1,010
- 2023Extracting Training Data from Diffusion Modelscited by 723
- 2023Understanding and Mitigating Copying in Diffusion Modelscited by 205
11 more
- 2022A Self-Supervised Descriptor for Image Copy Detectioncited by 175
- 2024Detecting, Explaining, and Mitigating Memorization in Diffusion Modelscited by 94
- 2023A Reproducible Extraction of Training Images from Diffusion Modelscited by 60
- 2024Unveiling and Mitigating Memorization in Text-to-image Diffusion Models through Cross Attentioncited by 57
- 2024Towards Memorization-Free Diffusion Modelscited by 49
- 2021The 2021 Image Similarity Dataset and Challengecited by 44
- 2024A Geometric Framework for Understanding Memorization in Generative Modelscited by 40
- 2024Finding NeMo: Localizing Neurons Responsible For Memorization in Diffusion Modelscited by 36
- 2024Classifier-Free Guidance inside the Attraction Basin May Cause Memorizationcited by 21
- 2024Exploring Local Memorization in Diffusion Models via Bright Ending Attentioncited by 19
- 2025Adjusting Initial Noise to Mitigate Memorization in Text-to-Image Diffusion Modelscited by 7
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2024Memorized Images in Diffusion Models share a Subspace that can be Located and Deleted
- 2025Finding DoRI: Discovery of Retained Images in Diffusion Models
- 2026Mitigating Memorization in Text-to-Image Diffusion via Region-Aware Prompt Augmentation and Multimodal Copy Detection
- 2026You Don't Need All That Attention: Surgical Memorization Mitigation in Text-to-Image Diffusion Models
- 2026Memorization In Stable Diffusion Is Unexpectedly Driven by CLIP Embeddings
- 2026Forgetting is Competition: Rethinking Unlearning as Representation Interference in Diffusion Models
- 2025Redistribute Ensemble Training for Mitigating Memorization in Diffusion Models
- 2026Prompt Reinjection: Alleviating Prompt Forgetting in Multimodal Diffusion Transformers for Text-to-Image Generation