PAC-Bayesian Generalization Bounds for Adversarial Generative Models
Sokhna Diarra Mbacke, Florence Clerc, Pascal Germain
Abstract
We extend PAC-Bayesian theory to generative models and develop generalization bounds for models based on the Wasserstein distance and the total variation distance. Our first result on the Wasserstein distance assumes the instance space is bounded, while our second result takes advantage of dimensionality reduction. Our results naturally apply to Wasserstein GANs and Energy-Based GANs, and our bounds provide new training objectives for these two. Although our work is mainly theoretical, we perform numerical experiments showing non-vacuous generalization bounds for Wasserstein GANs on synthetic datasets.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 45986b0f-bfcd-41c9-993a-9abfafc3888aCited by top-tier papers4
- Statistical Guarantees for Variational Autoencoders using PAC-Bayesian TheorySokhna Diarra Mbacke, Florence Clerc, Pascal GermainNeurIPS 2023 · 22 citations
- Generalization and Distributed Learning of GFlowNetsTiago Silva, Amauri H. Souza, Omar Rivasplata, Vikas Garg et al.ICLR 2025
- Proof-of-Authorship for Diffusion-based AI Generated ContentDe Zhang Lee, Han Fang, Ee-Chien ChangCCS 2026
- Generalization in VAE and Diffusion Models: A Unified Information-Theoretic AnalysisQi Chen, Jierui Zhu, Florian ShkurtiICLR 2025
Builds on9
- The Intrinsic Dimension of Images and Its Impact on LearningPhillip Pope, Chen Zhu, Ahmed Abdelkader, Micah Goldblum et al.ICLR 2021 · 381 citations
- Distributional Sliced-Wasserstein and Applications to Generative ModelingKhai Nguyen, Nhat Ho, Tung Pham, Hung BuiICLR 2021 · 111 citations
- PAC-Bayes Analysis Beyond the Usual BoundsOmar Rivasplata, Ilja Kuzborskij, Csaba Szepesvári, John Shawe-TaylorNeurIPS 2020 · 101 citations
- Random Matrix Theory Proves that Deep Learning Representations of GAN-data Behave as Gaussian MixturesMohamed El Amine Seddik, Cosme Louart, Mohamed Tamaazousti, Romain CouilletICML 2020 · 78 citations
- Online PAC-Bayes LearningMaxime Haddouche, Benjamin GuedjNeurIPS 2022 · 33 citations
Related papers
- Shedding a PAC-Bayesian Light on Adaptive Sliced-Wasserstein DistancesRuben Ohana, Kimia Nadjahi, Alain Rakotomamonjy, Liva RalaivolaICML 2023 · 7 citations
- Learning via Wasserstein-Based High Probability Generalisation BoundsPaul Viallard, Maxime Haddouche, Umut Simsekli, Benjamin GuedjNeurIPS 2023 · 16 citations
- Integral Probability Metrics PAC-Bayes BoundsRon Amit, Baruch Epstein, Shay Moran, Ron MeirNeurIPS 2022 · 25 citations
- Information-Theoretic Generalization Bounds for VAEs: A Role of Encoder and Latent VariableFutoshi Futami, Masahiro FujisawaICML 2026
- Universal generalization guarantees for Wasserstein distributionally robust modelsTam Le, Jérôme MalickICLR 2025
