Enregistré dans:
Détails bibliographiques
Auteurs principaux: Shi, Lianghe, Wu, Meng, Zhang, Huijie, Zhang, Zekai, Tao, Molei, Qu, Qing
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:https://arxiv.org/abs/2509.16499
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866917169181229056
author Shi, Lianghe
Wu, Meng
Zhang, Huijie
Zhang, Zekai
Tao, Molei
Qu, Qing
author_facet Shi, Lianghe
Wu, Meng
Zhang, Huijie
Zhang, Zekai
Tao, Molei
Qu, Qing
contents The widespread use of diffusion models has led to an abundance of AI-generated data, raising concerns about model collapse -- a phenomenon in which recursive iterations of training on synthetic data lead to performance degradation. Prior work primarily characterizes this collapse via variance shrinkage or distribution shift, but these perspectives miss practical manifestations of model collapse. This paper identifies a transition from generalization to memorization during model collapse in diffusion models, where models increasingly replicate training data instead of generating novel content during iterative training on synthetic samples. This transition is directly driven by the declining entropy of the synthetic training data produced in each training cycle, which serves as a clear indicator of model degradation. Motivated by this insight, we propose an entropy-based data selection strategy to mitigate the transition from generalization to memorization and alleviate model collapse. Empirical results show that our approach significantly enhances visual quality and diversity in recursive generation, effectively preventing collapse.
format Preprint
id arxiv_https___arxiv_org_abs_2509_16499
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle A Closer Look at Model Collapse: From a Generalization-to-Memorization Perspective
Shi, Lianghe
Wu, Meng
Zhang, Huijie
Zhang, Zekai
Tao, Molei
Qu, Qing
Machine Learning
The widespread use of diffusion models has led to an abundance of AI-generated data, raising concerns about model collapse -- a phenomenon in which recursive iterations of training on synthetic data lead to performance degradation. Prior work primarily characterizes this collapse via variance shrinkage or distribution shift, but these perspectives miss practical manifestations of model collapse. This paper identifies a transition from generalization to memorization during model collapse in diffusion models, where models increasingly replicate training data instead of generating novel content during iterative training on synthetic samples. This transition is directly driven by the declining entropy of the synthetic training data produced in each training cycle, which serves as a clear indicator of model degradation. Motivated by this insight, we propose an entropy-based data selection strategy to mitigate the transition from generalization to memorization and alleviate model collapse. Empirical results show that our approach significantly enhances visual quality and diversity in recursive generation, effectively preventing collapse.
title A Closer Look at Model Collapse: From a Generalization-to-Memorization Perspective
topic Machine Learning
url https://arxiv.org/abs/2509.16499