用几何框架解释生成模型为何记住数据,区分记忆类型。
A Geometric Framework for Understanding Memorization in Generative Models
- 基于数据流形维度差异,构建记忆分析的几何框架。
- 实验证明:高维数据易过拟合记忆,低维结构则反映分布本质。
- 适用于研究隐私风险或模型可解释性的研究人员。
随着深度生成模型的发展,近期研究发现其在部署时可能重现训练数据样本,引发法律与隐私担忧。为深入理解这一现象,本文提出流形记忆假设(MMH),一个基于流形假设的几何分析框架,通过比较真实数据流形与模型学习流形的维度关系,来量化记忆程度。该框架系统地将记忆分为两类:由过拟合引起的记忆和由底层数据分布驱动的记忆。通过对已有工作的再分析,统一解释了文献中的多种观察结果。在合成数据与至Stable Diffusion规模图像数据集上,通过新开发的检测工具验证了MMH的有效性,并提出了防止生成记忆样本的方法。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。