Skip to content
AI Atlas
PaperActive

Generalization in VAE and Diffusion Models: A Unified Information-Theoretic Analysis

arxiv.org/abs/2506.00849

quality89

Updated 4 h ago · first seen 11 Sept 2026

paper_01M294FRSCJAF5PF68RTG3JK64

Published
11 Sept 2026
T1 · 4 h ago
arXiv
2506.00849
T1 · 4 h ago
Category
cs.LG
T1 · 4 h ago

Abstract

Despite the empirical success of Diffusion Models (DMs) and Variational Autoencoders (VAEs), their generalization performance remains theoretically underexplored, especially lacking a full consideration of the shared encoder-generator structure. Leveraging recent information-theoretic tools, we propose a unified theoretical framework that provides guarantees for the generalization of both the encoder and generator by treating them as randomized mappings. This framework further enables (1) a refined analysis for VAEs, accounting for the generator's generalization, which was previously overlooked; (2) illustrating an explicit trade-off in generalization terms for DMs that depends on the diffusion time $T$; and (3) providing computable bounds for DMs based solely on the training data, allowing the selection of the optimal $T$ and the integration of such bounds into the optimization process to improve model performance. Empirical results on both synthetic and real datasets illustrate the validity of the proposed theory.

Authors 3

Qi Chen, Jierui Zhu, Florian Shkurti

Specification

Official page

Source:arXiv (Atom API + RSS)T1observed 4 h agohigh

Arxiv announce type
replace

Source:arXiv (Atom API + RSS)T1observed 4 h agohigh

arXiv id
2506.00849

Source:arXiv (Atom API + RSS)T1observed 4 h agohigh

Categories
cs.LG, cs.AI

Source:arXiv (Atom API + RSS)T1observed 4 h agohigh

PDF

Source:arXiv (Atom API + RSS)T1observed 4 h agohigh

Primary category
cs.LG

Source:arXiv (Atom API + RSS)T1observed 4 h agohigh

Published
11 Sept 2026

Source:arXiv (Atom API + RSS)T1observed 4 h agohigh

Each value shows its source, tier and observation time. Conflicting claims are kept side by side and flagged — never averaged. How AI Atlas records facts →

Provenance

Attributed facts

9

Source tiers

T19

Freshest observation

4 h ago

Conflicts

None