mode collapse
In plain wordsWhen a generator gets stuck producing only a few kinds of output — the same face, the same joke — instead of the full variety it was meant to learn.
- 1
Generative adversarial networks
A failure of generative adversarial networkgenerative adversarial network, n.: Two neural networks trained as rivals: one forges fake images, the other tries to spot the fakes. As both improve, the forger's output becomes realistic enough to fool people. training in which the generator's output loses variety: large regions of its noise input all yield the same or nearly the same sample, so only a few modes (regions of high probability) of the data distribution are ever produced. It arises because the generator is rewarded for fooling the current discriminator, not for diversity: it can chase whichever outputs the discriminator currently misjudges, and the two networks may cycle between modes instead of converging. Partial collapse — reduced variety rather than a single output — is common and is also called mode dropping. Remedies include minibatch discrimination, unrolled objectives, Wasserstein lossesWasserstein GAN, n.: A version of the forger-versus-detective image generator that measures how far its fakes are from real data in a smoother way, making training steadier and easier to monitor., spectral normalization and larger batches; diffusion modelsdiffusion model, n.: An AI that learns to create images, sound or video by practicing noise removal: shown pictures buried under static, it learns to clean them up, then builds new ones starting from pure static. and likelihood-trained models are far less prone to it.
After epoch 40 every sample was the same brown-haired face: classic mode collapse.
- 2
Language models
By extension, the loss of output diversity in a large language modellarge language model, n.: A computer program trained on enormous amounts of text to predict the words that come next. At vast scale, that simple skill lets it chat, answer questions, translate, summarize and write code. after instruction tuninginstruction tuning, n.: Training a language model on many tasks written as plain instructions with correct answers, so it learns to do what a request asks instead of just continuing the text. or reinforcement learning from human feedbackreinforcement learning from human feedback, n.: A way of training AI using people's judgments: humans compare pairs of answers, a scoring model learns their preferences, and the AI is then trained to produce answers that score well.: the model concentrates probability on a few favored answers, phrasings or styles — naming the same 'random' number, writing near-identical stories — even when variety is wanted. It differs from a low sampling temperaturetemperature, n.: A setting that controls how adventurous an AI model's word choices are. Low temperature makes it pick the safest, most likely words; high temperature makes it more varied and surprising, but more error-prone. because the preference persists across prompts and sampling settings.
Origin
The original 2014 GAN paper noted the problem, jokingly calling the extreme case 'the Helvetica scenario', without using the phrase 'mode collapse'; in June 2016 Tim Salimans and colleagues described generators collapsing to a single mode. The noun phrase was common by late 2016, in Unrolled GANs (Luke Metz and colleagues, November 2016) and Ian Goodfellow's NIPS 2016 tutorial. The language-model sense spread after 'Mysteries of mode collapse', a LessWrong post by the pseudonymous researcher janus (November 2022).
Usage note
Not to be confused with model collapsemodel collapse, n.: The gradual decline of AI models trained on the output of earlier AI models: rare details vanish first, then quality and variety erode, like a photocopy of a photocopy., the degradation of models trained on generated data, or with posterior collapse in VAEsvariational autoencoder, n.: A neural network that learns to squeeze data into a compact, slightly fuzzy code and rebuild it, arranged so that decoding random codes produces new, plausible examples..
Related
- generative adversarial networkgenerative adversarial network, n.: Two neural networks trained as rivals: one forges fake images, the other tries to spot the fakes. As both improve, the forger's output becomes realistic enough to fool people.
- Wasserstein GANWasserstein GAN, n.: A version of the forger-versus-detective image generator that measures how far its fakes are from real data in a smoother way, making training steadier and easier to monitor.
- model collapsemodel collapse, n.: The gradual decline of AI models trained on the output of earlier AI models: rare details vanish first, then quality and variety erode, like a photocopy of a photocopy.
- Fréchet inception distanceFréchet inception distance, n.: A score for image generators that compares the overall statistics of their pictures with those of real photos, as seen by an image-recognition network; lower is better.
- reinforcement learning from human feedbackreinforcement learning from human feedback, n.: A way of training AI using people's judgments: humans compare pairs of answers, a scoring model learns their preferences, and the AI is then trained to produce answers that score well.
- temperaturetemperature, n.: A setting that controls how adventurous an AI model's word choices are. Low temperature makes it pick the safest, most likely words; high temperature makes it more varied and surprising, but more error-prone.
- model collapsemodel collapse, n.: The gradual decline of AI models trained on the output of earlier AI models: rare details vanish first, then quality and variety erode, like a photocopy of a photocopy.
Sources
- Metz, L., Poole, B., Pfau, D. & Sohl-Dickstein, J. (2016). Unrolled Generative Adversarial Networks. arXiv:1611.02163 (ICLR 2017). arxiv.org/abs/1611.02163
- janus (2022). Mysteries of mode collapse. LessWrong, 8 November 2022. lesswrong.com/posts/t9svvNPNmFf5Qa3TA/mysteries-of-mode-collapse
Mentioned in
Fréchet inception distanceFréchet inception distance, n.: A score for image generators that compares the overall statistics of their pictures with those of real photos, as seen by an image-recognition network; lower is better., generative adversarial networkgenerative adversarial network, n.: Two neural networks trained as rivals: one forges fake images, the other tries to spot the fakes. As both improve, the forger's output becomes realistic enough to fool people., model collapsemodel collapse, n.: The gradual decline of AI models trained on the output of earlier AI models: rare details vanish first, then quality and variety erode, like a photocopy of a photocopy., entropy collapseentropy collapse, n.: A failure in reinforcement-learning training where an AI quickly becomes overconfident and stops trying different answers, so it explores less and its improvement stalls., Wasserstein GANWasserstein GAN, n.: A version of the forger-versus-detective image generator that measures how far its fakes are from real data in a smoother way, making training steadier and easier to monitor.