Skip to content

Latest commit

 

History

History
56 lines (56 loc) · 2.32 KB

2022-06-28-banerjee22a.md

File metadata and controls

56 lines (56 loc) · 2.32 KB
title booktitle abstract layout series publisher issn id month tex_title firstpage lastpage page order cycles bibtex_author author date address container-title volume genre issued pdf extras
Stability Based Generalization Bounds for Exponential Family Langevin Dynamics
Proceedings of the 39th International Conference on Machine Learning
Recent years have seen advances in generalization bounds for noisy stochastic algorithms, especially stochastic gradient Langevin dynamics (SGLD) based on stability (Mou et al., 2018; Li et al., 2020) and information theoretic approaches (Xu & Raginsky, 2017; Negrea et al., 2019; Steinke & Zakynthinou, 2020). In this paper, we unify and substantially generalize stability based generalization bounds and make three technical contributions. First, we bound the generalization error in terms of expected (not uniform) stability which arguably leads to quantitatively sharper bounds. Second, as our main contribution, we introduce Exponential Family Langevin Dynamics (EFLD), a substantial generalization of SGLD, which includes noisy versions of Sign-SGD and quantized SGD as special cases. We establish data dependent expected stability based generalization bounds for any EFLD algorithm with a O(1/n) sample dependence and dependence on gradient discrepancy rather than the norm of gradients, yielding significantly sharper bounds. Third, we establish optimization guarantees for special cases of EFLD. Further, empirical results on benchmarks illustrate that our bounds are non-vacuous, quantitatively sharper than existing bounds, and behave correctly under noisy labels.
inproceedings
Proceedings of Machine Learning Research
PMLR
2640-3498
banerjee22a
0
Stability Based Generalization Bounds for Exponential Family {L}angevin Dynamics
1412
1449
1412-1449
1412
false
Banerjee, Arindam and Chen, Tiancong and Li, Xinyan and Zhou, Yingxue
given family
Arindam
Banerjee
given family
Tiancong
Chen
given family
Xinyan
Li
given family
Yingxue
Zhou
2022-06-28
Proceedings of the 39th International Conference on Machine Learning
162
inproceedings
date-parts
2022
6
28