Robustness Implies Generalization via Data-Dependent Generalization Bounds
Kenji Kawaguchi, Zhun Deng, Kyle Luh, Jiaoyang Huang
Abstract
This paper proves that robustness implies generalization via data-dependent generalization bounds. As a result, robustness and generalization are shown to be connected closely in a data-dependent manner. Our bounds improve previous bounds in two directions, to solve an open problem that has seen little development since 2010. The first is to reduce the dependence on the covering number. The second is to remove the dependence on the hypothesis space. We present several examples, including ones for lasso and deep learning, in which our bounds are provably preferable. The experiments on real-world data and theoretical models demonstrate near-exponential improvements in various situations. To achieve these improvements, we do not require additional assumptions on the unknown distribution; instead, we only incorporate an observable and computable property of the training samples. A key technical innovation is an improved concentration bound for multinomial random variables that is of independent interest beyond robustness and generalization.
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 36d262db-fd44-4d96-907d-854f921063acCited by top-tier papers10
- How Does Information Bottleneck Help Deep Learning?Kenji Kawaguchi, Zhun Deng, Xu Ji, Jiaoyang HuangICML 2023 · 117 citations
- Simple Hierarchical Planning with DiffusionChang Chen, Fei Deng, Kenji Kawaguchi, Caglar Gulcehre et al.ICLR 2024 · 79 citations
- Discrete Key-Value BottleneckFrederik Träuble, Anirudh Goyal, Nasim Rahaman, Michael Curtis Mozer et al.ICML 2023 · 25 citations
- Towards Robust Out-of-Distribution Generalization Bounds via SharpnessYingtian Zou, Kenji Kawaguchi, Yingnan Liu, Jiashuo Liu et al.ICLR 2024 · 13 citations
- Rethinking Information-theoretic Generalization: Loss Entropy Induced PAC BoundsYuxin Dong, Tieliang Gong, Hong Chen, Shujian Yu et al.ICLR 2024 · 8 citations
Builds on5
- How Does Mixup Help With Robustness and Generalization?Linjun Zhang, Zhun Deng, Kenji Kawaguchi, Amirata Ghorbani et al.ICLR 2021 · 294 citations
- Adversarial Training Helps Transfer Learning via Better RepresentationsZhun Deng, Linjun Zhang, Kailas Vodrahalli, Kenji Kawaguchi et al.NeurIPS 2021 · 60 citations
- Toward Better Generalization Bounds with Locally Elastic StabilityZhun Deng, Hangfeng He, Weijie J. SuICML 2021 · 51 citations
- Adversarial Robustness with Semi-Infinite Constrained LearningAlexander Robey, Luiz F. O. Chamon, George J. Pappas, Hamed Hassani et al.NeurIPS 2021 · 51 citations
- Robustness between the worst and average caseLeslie Rice, Anna Bair, Huan Zhang, J. Zico KolterNeurIPS 2021 · 26 citations
Related papers
- Generalizability of Neural Networks Minimizing Empirical Risk Based on Expressive PowerLijia Yu, Yibo Miao, Yifan Zhu, Xiao-Shan Gao et al.ICLR 2025
- Generalization Bounds using Data-Dependent Fractal DimensionsBenjamin Dupuis, George Deligiannidis, Umut SimsekliICML 2023 · 17 citations
- Toward Better PAC-Bayes Bounds for Uniformly Stable AlgorithmsSijia Zhou, Yunwen Lei, Ata KabánNeurIPS 2023 · 4 citations
- Relative Deviation Margin BoundsCorinna Cortes, Mehryar Mohri, Ananda Theertha SureshICML 2021 · 16 citations
- Neural Dependencies Emerging from Learning Massive CategoriesRuili Feng, Kecheng Zheng, Kai Zhu, Yujun Shen et al.CVPR 2023
