paper-with-me

홈 › Papers

Demystifying Foreground-Background Memorization in Diffusion Models

2025-08-16 · Jimmy Z. Di, Yiwei Lu, Yaoliang Yu, Gautam Kamath, Adam Dziedzic, Franziska Boenisch arxiv

Diffusion models (DMs) memorize training images and can reproduce near-duplicates during generation. Current detection methods identify verbatim memorization but fail to capture two critical aspects: quantifying partial memorization occurring in small image regions, and memorization patterns beyond specific prompt-image pairs. To address these limitations, we propose Foreground Background Memorization (FB-Mem), a novel segmentation-based metric that classifies and quantifies memorized regions within generated images. Our method reveals that memorization is more pervasive than previously understood: (1) individual generations from single prompts may be linked to clusters of similar training images, revealing complex memorization patterns that extend beyond one-to-one correspondences; and (2) existing model-level mitigation methods, such as neuron deactivation and pruning, fail to eliminate local memorization, which persists particularly in foreground regions. Our work establishes an effective framework for measuring memorization in diffusion models, demonstrates the inadequacy of current mitigation approaches, and proposes a stronger mitigation method using a clustering approach.

📄 PDF Abstract BibTeX arXiv:2508.12148

Code (0)

등록된 구현이 없습니다.

Similar Papers 제목 키워드 기반

Uncovering Memorization Effect in the Presence of Spurious Correlations

2025-01-01 · Chenyu You, Haocheng Dai, Yifei Min, Jasjeet S. Sekhon 외

Machine learning models often rely on simple spurious features -- patterns in training data that correlate with targets but are not causally related to them, like image backgrounds in foreground classification. This reli…

Memorization

Invariant Learning via Diffusion Dreamed Distribution Shifts

2022-11-18 · Priyatham Kattakinda, Alexander Levine, Soheil Feizi

Though the background is an important signal for image classification, over reliance on it can lead to incorrect predictions when spurious correlations between foreground and background are broken at test time. Training …

image-classificationImage Classification

Few-Shot Generation of Brain Tumors for Secure and Fair Data Sharing

2025-03-31 · Yongyi Shi, Ge Wang

Leveraging multi-center data for medical analytics presents challenges due to privacy concerns and data heterogeneity. While distributed approaches such as federated learning has gained traction, they remain vulnerable t…

Brain Tumor SegmentationData AugmentationFairnessFederated Learning+2

TKG-DM: Training-free Chroma Key Content Generation Diffusion Model

2024-11-23 · CVPR 2025 1 · Ryugo Morita, Stanislav Frolov, Brian Bernhard Moser, Takahiro Shirakawa 외

Diffusion models have enabled the generation of high-quality images with a strong focus on realism and textual fidelity. Yet, large-scale text-to-image models, such as Stable Diffusion, struggle to generate images where …

HYPNOS : Highly Precise Foreground-focused Diffusion Finetuning for Inanimate Objects

2024-10-18 · Oliverio Theophilus Nathanael, Jonathan Samuel Lumentut, Nicholas Hans Muliawan, Edbert Valencio Angky 외

In recent years, personalized diffusion-based text-to-image generative tasks have been a hot topic in computer vision studies. A robust diffusion model is determined by its ability to perform near-perfect reconstruction …

Disentanglement