A Geometric Framework for Understanding Memorization in Generative Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Ross, Brendan Leigh, Kamkari, Hamidreza, Wu, Tongzi, Hosseinzadeh, Rasa, Liu, Zhaoyan, Stein, George, Cresswell, Jesse C., Loaiza-Ganem, Gabriel
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916651197267968
author Ross, Brendan Leigh
Kamkari, Hamidreza
Wu, Tongzi
Hosseinzadeh, Rasa
Liu, Zhaoyan
Stein, George
Cresswell, Jesse C.
Loaiza-Ganem, Gabriel
author_facet Ross, Brendan Leigh
Kamkari, Hamidreza
Wu, Tongzi
Hosseinzadeh, Rasa
Liu, Zhaoyan
Stein, George
Cresswell, Jesse C.
Loaiza-Ganem, Gabriel
contents As deep generative models have progressed, recent work has shown them to be capable of memorizing and reproducing training datapoints when deployed. These findings call into question the usability of generative models, especially in light of the legal and privacy risks brought about by memorization. To better understand this phenomenon, we propose the manifold memorization hypothesis (MMH), a geometric framework which leverages the manifold hypothesis into a clear language in which to reason about memorization. We propose to analyze memorization in terms of the relationship between the dimensionalities of (i) the ground truth data manifold and (ii) the manifold learned by the model. This framework provides a formal standard for "how memorized" a datapoint is and systematically categorizes memorized data into two types: memorization driven by overfitting and memorization driven by the underlying data distribution. By analyzing prior work in the context of the MMH, we explain and unify assorted observations in the literature. We empirically validate the MMH using synthetic data and image datasets up to the scale of Stable Diffusion, developing new tools for detecting and preventing generation of memorized samples in the process.
format Preprint
id arxiv_https___arxiv_org_abs_2411_00113
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle A Geometric Framework for Understanding Memorization in Generative Models
Ross, Brendan Leigh
Kamkari, Hamidreza
Wu, Tongzi
Hosseinzadeh, Rasa
Liu, Zhaoyan
Stein, George
Cresswell, Jesse C.
Loaiza-Ganem, Gabriel
Machine Learning
As deep generative models have progressed, recent work has shown them to be capable of memorizing and reproducing training datapoints when deployed. These findings call into question the usability of generative models, especially in light of the legal and privacy risks brought about by memorization. To better understand this phenomenon, we propose the manifold memorization hypothesis (MMH), a geometric framework which leverages the manifold hypothesis into a clear language in which to reason about memorization. We propose to analyze memorization in terms of the relationship between the dimensionalities of (i) the ground truth data manifold and (ii) the manifold learned by the model. This framework provides a formal standard for "how memorized" a datapoint is and systematically categorizes memorized data into two types: memorization driven by overfitting and memorization driven by the underlying data distribution. By analyzing prior work in the context of the MMH, we explain and unify assorted observations in the literature. We empirically validate the MMH using synthetic data and image datasets up to the scale of Stable Diffusion, developing new tools for detecting and preventing generation of memorized samples in the process.
title A Geometric Framework for Understanding Memorization in Generative Models
topic Machine Learning
url https://arxiv.org/abs/2411.00113