paper-with-me

홈 › Papers

An Inversion-based Measure of Memorization for Diffusion Models

2024-05-09 · Zhe Ma, Qingming Li, Xuhong Zhang, Tianyu Du, Ruixiao Lin, Zonghui Wang, Shouling Ji, Wenzhi Chen

The past few years have witnessed substantial advances in image generation powered by diffusion models. However, it was shown that diffusion models are vulnerable to training data memorization, raising concerns regarding copyright infringement and privacy invasion. This study delves into a rigorous analysis of memorization in diffusion models. We introduce an inversion-based measure of memorization, InvMM, which searches for a sensitive latent noise distribution accounting for the replication of an image. For accurate estimation of the memorization score, we propose an adaptive algorithm that balances the normality and sensitivity of the inverted distribution. Comprehensive experiments, conducted on both unconditional and text-guided diffusion models, demonstrate that InvMM is capable of detecting heavily memorized images and elucidating the effect of various factors on memorization. Additionally, we discuss how memorization differs from membership. In practice, InvMM serves as a useful tool for model developers to reliably assess the risk of memorization, thereby contributing to the enhancement of trustworthiness and privacy-preserving capabilities of diffusion models.

📄 PDF Abstract BibTeX arXiv:2405.05846

Code (1)

Maryeon/InvMM 공식 구현 pytorch

Tasks

Image GenerationMemorizationPrivacy Preserving

Methods 이 논문이 사용한 방법론

SET Dynamic Sparse Training method where weight mask is updated randomly periodically
Diffusion Diffusion models generate samples by gradually removing noise from a signal, and their training objective can be expressed as a reweighted variational lower-bound…

Similar Papers 제목 키워드 기반

Latent Diffusion Inversion Requires Understanding the Latent Space

2025-11-25 · Mingxing Rao, Bowen Qu, Daniel Moyer arxiv

The recovery of training data from generative models ("model inversion") has been extensively studied for diffusion models in the data domain as a memorization/overfitting phenomenon. Latent diffusion models (LDMs), whic…

On the role of memorization in learned priors for geophysical inverse problems

2026-03-20 · Ali Siahkoohi, Davide Sabeddu arxiv

Learned priors based on deep generative models offer data-driven regularization for seismic inversion, but training them requires a dataset of representative subsurface models -- a resource that is inherently scarce in g…

Memorization, Emergence, and Explaining Reversal Failures: A Controlled Study of Relational Semantics in LLMs

2026-01-06 · Yihua Zhu, Qianying Liu, Jiaxin Wang, Fei Cheng 외 arxiv

Autoregressive LLMs perform well on relational tasks that require linking entities via relational words (e.g., father/son, friend), but it is unclear whether they learn the logical semantics of such relations (e.g., symm…

Filtering Memorization from Parameter-Space in Diffusion Models

2026-05-11 · Yu Zhe, Yang Jiayan, Wei Junhao, Yu-Lin Tsai 외 arxiv

Low-Rank Adaptation (LoRA) has become a widely used mechanism for customizing diffusion models, enabling users to inject new visual concepts or styles through lightweight parameter updates. However, LoRAs can memorize tr…

On the Memorization of Consistency Distillation for Diffusion Models

2026-04-26 · Bingqing Jiang, Difan Zou arxiv

Diffusion models are central to modern generative modeling, and understanding how they balance memorization and generalization is critical for reliable deployment. Recent work has shown that memorization in diffusion mod…