about
A potential way to mitigate memorization in Diffusion models? (arxiv.org)
1 point by mildcaseofphd on Jun 1, 2023 | hide | past | pdf | discuss on HN

In plain words: Copying in image generators is usually blamed on repeated training pictures, but the text prompt matters just as much: models without prompts rarely copy, while text-guided ones often do. Randomizing and enriching captions during training cut down on this copying.

Abstract · Understanding and Mitigating Copying in Diffusion Models

Images generated by diffusion models like Stable Diffusion are increasingly widespread. Recent works and even lawsuits have shown that these models are prone to replicating their training data, unbeknownst to the user. In this paper, we first analyze this memorization problem in text-to-image diffusion models. While it is widely believed that duplicated images in the training set are responsible for content replication at inference time, we observe that the text conditioning of the model plays a similarly important role. In fact, we see in our experiments that data replication often does not happen for unconditional models, while it is common in the text-conditional case. Motivated by our findings, we then propose several techniques for reducing data replication at both training and inference time by randomizing and augmenting image captions in the training set.

Gowthami Somepalli, Vasu Singla, Micah Goldblum, Jonas Geiping, Tom Goldstein
arXiv:2305.20086 · cs.LG, cs.CR, cs.CV · submitted May 31, 2023
abstract · pdf · html · 17 pages, preprint. Code is available at https://github.com/somepago/DCR

add comment on HN