SolidMark: Evaluating Image Memorization in Generative Models

Nicky Kriplani, Minh Pham, Gowthami Somepalli, Chinmay Hegde, Niv Cohen

Keywords: Memorization, Diffusion Models, Metrics

TL;DR: We present a method to evaluate fine-grained pixel-level memorization in diffusion models.

Abstract:

Recent works have shown that diffusion models are able to memorize training images and emit them at generation time. However, the metrics used to evaluate memorization and its mitigation techniques suffer from dataset-dependent biases and struggle to detect whether a given specific image has been memorized or not.

This paper begins with a comprehensive exploration of issues surrounding memorization metrics in diffusion models. Then, to mitigate these issues, we introduce SolidMark, a novel evaluation method that provides a per-image memorization score. We then re-evaluate existing memorization mitigation techniques and show that SolidMark is capable of evaluating fine-grained pixel-level memorization. Finally, we release a variety of models based on SolidMark to facilitate further research for understanding memorization phenomena in generative models.

Submission Number: 20