Identifying Memorization of Diffusion Models through $p$-Laplace Analysis: Estimators, Bounds and Applications

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Brokman, Jonathan, Gershon, Itay, Giloni, Amit, Hofman, Omer, Vainshtein, Roman, Kojima, Hisashi, Gilboa, Guy
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915816414380032
author Brokman, Jonathan
Gershon, Itay
Giloni, Amit
Hofman, Omer
Vainshtein, Roman
Kojima, Hisashi
Gilboa, Guy
author_facet Brokman, Jonathan
Gershon, Itay
Giloni, Amit
Hofman, Omer
Vainshtein, Roman
Kojima, Hisashi
Gilboa, Guy
contents Diffusion models, today's leading image generative models, estimate the score function, i.e. the gradient of the log probability of (perturbed) data samples, without direct access to the underlying probability distribution. This work investigates whether the estimated score function can be leveraged to compute higher-order differentials, namely the p-Laplace operators. We show that these operators can be employed to identify memorized training data. We propose a numerical p-Laplace approximation based on the learned score functions, showing its effectiveness in identifying key features of the probability landscape. Furthermore, theoretical error-bounds to these estimators are proven and demonstrated numerically. We analyze the structured case of Gaussian mixture models, and demonstrate that the results carry-over to text-conditioned image generative models (text-to-image), where memorization identification based on the p-Laplace operator is performed for the first time, showing its advantage on 500 memorized prompts ($\sim$3000 generated images) in a post-generation regime, especially when the conditioning text is unavailable.
format Preprint
id arxiv_https___arxiv_org_abs_2505_08246
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Identifying Memorization of Diffusion Models through $p$-Laplace Analysis: Estimators, Bounds and Applications
Brokman, Jonathan
Gershon, Itay
Giloni, Amit
Hofman, Omer
Vainshtein, Roman
Kojima, Hisashi
Gilboa, Guy
Computer Vision and Pattern Recognition
Numerical Analysis
Diffusion models, today's leading image generative models, estimate the score function, i.e. the gradient of the log probability of (perturbed) data samples, without direct access to the underlying probability distribution. This work investigates whether the estimated score function can be leveraged to compute higher-order differentials, namely the p-Laplace operators. We show that these operators can be employed to identify memorized training data. We propose a numerical p-Laplace approximation based on the learned score functions, showing its effectiveness in identifying key features of the probability landscape. Furthermore, theoretical error-bounds to these estimators are proven and demonstrated numerically. We analyze the structured case of Gaussian mixture models, and demonstrate that the results carry-over to text-conditioned image generative models (text-to-image), where memorization identification based on the p-Laplace operator is performed for the first time, showing its advantage on 500 memorized prompts ($\sim$3000 generated images) in a post-generation regime, especially when the conditioning text is unavailable.
title Identifying Memorization of Diffusion Models through $p$-Laplace Analysis: Estimators, Bounds and Applications
topic Computer Vision and Pattern Recognition
Numerical Analysis
url https://arxiv.org/abs/2505.08246