vix.ing · top · new · best · stats

Memorization Precedes Generation: Learning Unsupervised GANs with Memory Networks

2018/03/05 by Youngjin Kim, Minjung Kim, Kim, Youngjin +3 · 12 citations
Computer Science · Mathematics · #Artificial intelligence #Artificial neural network #Cognitive psychology #Computer science #Digital Media Forensic Detection #Discriminative model #FOS: Computer and information sciences #Fidelity #Forgetting #Generative Adversarial Networks and Image Synthesis #Generator (circuit theory) #MNIST database #Machine Learning (cs.LG) #Machine Learning (stat.ML) #Machine learning #Mathematics #Memorization #Multimodal Machine Learning Applications #Pattern recognition (psychology) #Representation (politics) #Unsupervised learning #cs.LG #stat.ML

paper · pdf · doi:10.48550/arxiv.1803.01500

published in arXiv (Cornell University) (Cornell University) · Published in ICLR 2018

openalex publication_date 2018/03/05 · arxiv created 2018/03/10 · arxiv updated 2018/03/13 · openalex created_date 2025/10/10 · openalex updated_date 2026/08/06

Abstract

We propose an approach to address two issues that commonly occur during training of unsupervised GANs. First, since GANs use only a continuous latent distribution to embed multiple classes or clusters of data, they often do not correctly handle the structural discontinuity between disparate classes in a latent space. Second, discriminators of GANs easily forget about past generated samples by generators, incurring instability during adversarial training. We argue that these two infamous problems of unsupervised GAN training can be largely alleviated by a learnable memory network to which both generators and discriminators can access. Generators can effectively learn representation of training samples to understand underlying cluster distributions of data, which ease the structure discontinuity problem. At the same time, discriminators can better memorize clusters of previously generated samples, which mitigate the forgetting problem. We propose a novel end-to-end GAN model named memoryGAN, which involves a memory network that is unsupervisedly trainable and integrable to many existing GAN models. With evaluations on multiple datasets such as Fashion-MNIST, CelebA, CIFAR10, and Chairs, we show that our model is probabilistically interpretable, and generates realistic image samples of high visual fidelity. The memoryGAN also achieves the state-of-the-art inception scores over unsupervised GAN models on the CIFAR10 dataset, without any optimization tricks and weaker divergences.

Citations

Related