Algorithm- and Data-Dependent Generalization Bounds for Diffusion Models

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Dupuis, Benjamin, Shariatian, Dario, Haddouche, Maxime, Durmus, Alain, Simsekli, Umut
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866915785457270784
author Dupuis, Benjamin
Shariatian, Dario
Haddouche, Maxime
Durmus, Alain
Simsekli, Umut
author_facet Dupuis, Benjamin
Shariatian, Dario
Haddouche, Maxime
Durmus, Alain
Simsekli, Umut
contents Score-based generative models (SGMs) have emerged as one of the most popular classes of generative models. A substantial body of work now exists on the analysis of SGMs, focusing either on discretization aspects or on their statistical performance. In the latter case, bounds have been derived, under various metrics, between the true data distribution and the distribution induced by the SGM, often demonstrating polynomial convergence rates with respect to the number of training samples. However, these approaches adopt a largely approximation theory viewpoint, which tends to be overly pessimistic and relatively coarse. In particular, they fail to fully explain the empirical success of SGMs or capture the role of the optimization algorithm used in practice to train the score network. To support this observation, we first present simple experiments illustrating the concrete impact of optimization hyperparameters on the generalization ability of the generated distribution. Then, this paper aims to bridge this theoretical gap by providing the first algorithmic- and data-dependent generalization analysis for SGMs. In particular, we establish bounds that explicitly account for the optimization dynamics of the learning algorithm, offering new insights into the generalization behavior of SGMs. Our theoretical findings are supported by empirical results on several datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2506_03849
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Algorithm- and Data-Dependent Generalization Bounds for Diffusion Models
Dupuis, Benjamin
Shariatian, Dario
Haddouche, Maxime
Durmus, Alain
Simsekli, Umut
Machine Learning
Score-based generative models (SGMs) have emerged as one of the most popular classes of generative models. A substantial body of work now exists on the analysis of SGMs, focusing either on discretization aspects or on their statistical performance. In the latter case, bounds have been derived, under various metrics, between the true data distribution and the distribution induced by the SGM, often demonstrating polynomial convergence rates with respect to the number of training samples. However, these approaches adopt a largely approximation theory viewpoint, which tends to be overly pessimistic and relatively coarse. In particular, they fail to fully explain the empirical success of SGMs or capture the role of the optimization algorithm used in practice to train the score network. To support this observation, we first present simple experiments illustrating the concrete impact of optimization hyperparameters on the generalization ability of the generated distribution. Then, this paper aims to bridge this theoretical gap by providing the first algorithmic- and data-dependent generalization analysis for SGMs. In particular, we establish bounds that explicitly account for the optimization dynamics of the learning algorithm, offering new insights into the generalization behavior of SGMs. Our theoretical findings are supported by empirical results on several datasets.
title Algorithm- and Data-Dependent Generalization Bounds for Diffusion Models
topic Machine Learning
url https://arxiv.org/abs/2506.03849