A Phase Transition in Diffusion Models Reveals the Hierarchical Nature of Data

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Sclocchi, Antonio, Favero, Alessandro, Wyart, Matthieu
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909439650430976
author Sclocchi, Antonio
Favero, Alessandro
Wyart, Matthieu
author_facet Sclocchi, Antonio
Favero, Alessandro
Wyart, Matthieu
contents Understanding the structure of real data is paramount in advancing modern deep-learning methodologies. Natural data such as images are believed to be composed of features organized in a hierarchical and combinatorial manner, which neural networks capture during learning. Recent advancements show that diffusion models can generate high-quality images, hinting at their ability to capture this underlying compositional structure. We study this phenomenon in a hierarchical generative model of data. We find that the backward diffusion process acting after a time $t$ is governed by a phase transition at some threshold time, where the probability of reconstructing high-level features, like the class of an image, suddenly drops. Instead, the reconstruction of low-level features, such as specific details of an image, evolves smoothly across the whole diffusion process. This result implies that at times beyond the transition, the class has changed, but the generated sample may still be composed of low-level elements of the initial image. We validate these theoretical insights through numerical experiments on class-unconditional ImageNet diffusion models. Our analysis characterizes the relationship between time and scale in diffusion models and puts forward generative models as powerful tools to model combinatorial data properties.
format Preprint
id arxiv_https___arxiv_org_abs_2402_16991
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle A Phase Transition in Diffusion Models Reveals the Hierarchical Nature of Data
Sclocchi, Antonio
Favero, Alessandro
Wyart, Matthieu
Machine Learning
Disordered Systems and Neural Networks
Computer Vision and Pattern Recognition
Understanding the structure of real data is paramount in advancing modern deep-learning methodologies. Natural data such as images are believed to be composed of features organized in a hierarchical and combinatorial manner, which neural networks capture during learning. Recent advancements show that diffusion models can generate high-quality images, hinting at their ability to capture this underlying compositional structure. We study this phenomenon in a hierarchical generative model of data. We find that the backward diffusion process acting after a time $t$ is governed by a phase transition at some threshold time, where the probability of reconstructing high-level features, like the class of an image, suddenly drops. Instead, the reconstruction of low-level features, such as specific details of an image, evolves smoothly across the whole diffusion process. This result implies that at times beyond the transition, the class has changed, but the generated sample may still be composed of low-level elements of the initial image. We validate these theoretical insights through numerical experiments on class-unconditional ImageNet diffusion models. Our analysis characterizes the relationship between time and scale in diffusion models and puts forward generative models as powerful tools to model combinatorial data properties.
title A Phase Transition in Diffusion Models Reveals the Hierarchical Nature of Data
topic Machine Learning
Disordered Systems and Neural Networks
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2402.16991