The pick-to-learn algorithm: empowering compression for tight generalization bounds and improved post-training performance
Author(s)
Paccagnan, Dario
Campi, Marco
Garatti, Simone
Type
Conference Paper
Abstract
Generalization bounds are valuable both for theory and applications. On the one hand, they shed light on the mechanisms that underpin the learning processes; on the
other, they certify how well a learned model performs against unseen inputs. In this work we build upon a recent breakthrough in compression theory (Campi & Garatti,
2023) to develop a new framework yielding tight generalization bounds of wide practical applicability. The core idea is to embed any given learning algorithm into
a suitably-constructed meta-algorithm (here called Pick-to-Learn, P2L) in order to instill desirable compression properties. When applied to the MNIST classification
dataset and to a synthetic regression problem, P2L not only attains generalization bounds that compare favorably with the state of the art (test-set and PAC-Bayes bounds), but it also learns models with better post-training performance
other, they certify how well a learned model performs against unseen inputs. In this work we build upon a recent breakthrough in compression theory (Campi & Garatti,
2023) to develop a new framework yielding tight generalization bounds of wide practical applicability. The core idea is to embed any given learning algorithm into
a suitably-constructed meta-algorithm (here called Pick-to-Learn, P2L) in order to instill desirable compression properties. When applied to the MNIST classification
dataset and to a synthetic regression problem, P2L not only attains generalization bounds that compare favorably with the state of the art (test-set and PAC-Bayes bounds), but it also learns models with better post-training performance
Date Issued
2023-12-10
Date Acceptance
2023-12-01
Citation
Advances in neural information processing systems, 2023, 36
ISBN
9781713899921
ISSN
1049-5258
Journal / Book Title
Advances in neural information processing systems
Volume
36
Copyright Statement
© 2023 The Author(s).
Source
37th Annual Conference on Neural Information Processing Systems (NeurIPS 2023)
Publication Status
Published
Start Date
2023-12-10
Finish Date
2023-12-16
Coverage Spatial
New Orleans, LA, USA