On Predicting Generalization using GANs
Yi Zhang, Arushi Gupta, Nikunj Saunshi, Sanjeev Arora
Abstract
Research on generalization bounds for deep networks seeks to give ways to predict test error using just the training dataset and the network parameters. While generalization bounds can give many insights about architecture design, training algorithms, etc., what they do not currently do is yield good predictions for actual test error. A recently introduced Predicting Generalization in Deep Learning competition aims to encourage discovery of methods to better predict test error. The current paper investigates a simple idea: can test error be predicted using synthetic data, produced using a Generative Adversarial Network (GAN) that was trained on the same training dataset? Upon investigating several GAN models and architectures, we find that this turns out to be the case. In fact, using GANs pre-trained on standard datasets, the test error can be predicted without requiring any additional hyper-parameter tuning. This result is surprising because GANs have well-known limitations (e.g. mode collapse) and are known to not learn the data distribution accurately. Yet the generated samples are good enough to substitute for test data. Several additional experiments are presented to explore reasons why GANs do well at this task. In addition to a new approach for predicting generalization, the counter-intuitive phenomena presented in our work may also call for a better understanding of GANs' strengths and limitations.
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 d517c8af-8615-475f-bec6-77e9df15ffc2Cited by top-tier papers3
- PAC-Bayesian Generalization Bounds for Adversarial Generative ModelsSokhna Diarra Mbacke, Florence Clerc, Pascal GermainICML 2023 · 12 citations
- CAME: Contrastive Automated Model EvaluationRu Peng, Qiuyang Duan, Haobo Wang, Jiachen Ma et al.ICCV 2023 · 8 citations
- Input Margins Can Predict Generalization TooCoenraad Mouton, Marthinus Wilhelmus Theunissen, Marelie H. DavelAAAI 2024 · 5 citations
Builds on7
- Differentiable Augmentation for Data-Efficient GAN TrainingShengyu Zhao, Zhijian Liu, Ji Lin, Jun-Yan Zhu et al.NeurIPS 2020 · 707 citations
- Fantastic Generalization Measures and Where to Find ThemYiding Jiang, Behnam Neyshabur, Hossein Mobahi, Dilip Krishnan et al.ICLR 2020 · 705 citations
- Evaluating Gradient Inversion Attacks and Defenses in Federated LearningYangsibo Huang, Samyak Gupta, Zhao Song, Kai Li et al.NeurIPS 2021 · 419 citations
- Seeing What a GAN Cannot GenerateDavid Bau, Jun-Yan Zhu, Jonas Wulff, William S. Peebles et al.ICCV 2019 · 342 citations
- Assessing Generalization of SGD via DisagreementYiding Jiang, Vaishnavh Nagarajan, Christina Baek, J. Zico KolterICLR 2022 · 134 citations
Related papers
- High-dimensional Analysis of Synthetic Data SelectionParham Rezaei, Filip Kovacevic, Francesco Locatello, Marco MondelliICLR 2026 · 6 citations
- Towards Universal Fake Image Detectors that Generalize Across Generative ModelsUtkarsh Ojha, Yuheng Li, Yong Jae LeeCVPR 2023
- CNN-Generated Images Are Surprisingly Easy to Spot... for NowSheng-Yu Wang, Oliver Wang, Richard Zhang, Andrew Owens et al.CVPR 2020
- Adversarial Feedback LoopFiras Shama, Roey Mechrez, Alon Shoshan, Lihi Zelnik-ManorICCV 2019 · 23 citations
- Generalization Error Bounds of Gradient Descent for Learning Over-Parameterized Deep ReLU NetworksYuan Cao, Quanquan GuAAAI 2020 · 168 citations
