Bound to Disagree: Generalization Bounds via Certifiable Surrogates
About
Generalization bounds for deep learning models are typically vacuous, not computable or restricted to specific model classes. In this paper, we tackle these issues by providing new disagreement-based certificates for the gap between the true risk of any two predictors. We then bound the true risk of the predictor of interest via a surrogate model that enjoys tight generalization guarantees, and by evaluating our disagreement bound on an unlabeled dataset.We empirically demonstrate the tightness of the obtained certificates and showcase the versatility of the approach by training surrogate models leveraging three different frameworks: sample compression, model compression and PAC-Bayes theory. Importantly, such guarantees are achieved without modifying the target model, nor adapting the training procedure to the generalization framework.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Generalization bound certification | CIFAR10 (test) | Generalization Bound17.36 | 7 | |
| Robustness Certification | MNIST (test) | Generalization Bound21.5 | 7 | |
| Generalization bound certification | MNIST | Generalization Bound0.0744 | 6 | |
| Generalization bound certification | CIFAR10 | Generalization Bound0.6858 | 6 |