Generalization Analysis of Machine Learning Algorithms via the Worst-Case Data-Generating Probability Measure - Inria - Institut national de recherche en sciences et technologies du numérique
Communication Dans Un Congrès Année : 2024

Generalization Analysis of Machine Learning Algorithms via the Worst-Case Data-Generating Probability Measure

Résumé

In this paper, the worst-case probability measure over the data is introduced as a tool for characterizing the generalization capabilities of machine learning algorithms. More specifically, the worst-case probability measure is a Gibbs probability measure and the unique solution to the maximization of the expected loss under a relative entropy constraint with respect to a reference probability measure. Fundamental generalization metrics, such as the sensitivity of the expected loss, the sensitivity of the empirical risk, and the generalization gap are shown to have closed-form expressions involving the worst-case data-generating probability measure. Existing results for the Gibbs algorithm, such as characterizing the generalization gap as a sum of mutual information and lautum information, up to a constant factor, are recovered. A novel parallel is established between the worst-case data-generating probability measure and the Gibbs algorithm. Specifically, the Gibbs probability measure is identified as a fundamental commonality of the model space and the data space for machine learning algorithms.
Fichier principal
Vignette du fichier
AAAI24camera_Hal.pdf (253.37 Ko) Télécharger le fichier

Dates et versions

hal-04353957 , version 1 (19-12-2023)

Licence

Identifiants

Citer

Xinying Zou, Samir M. Perlaza, Iñaki Esnaola, Eitan Altman. Generalization Analysis of Machine Learning Algorithms via the Worst-Case Data-Generating Probability Measure. AAAI 2024 - Conference on Artificial Intelligence, Feb 2024, Vancouver, Canada. pp.17271-17279, ⟨10.1609/aaai.v38i15.29674⟩. ⟨hal-04353957⟩
132 Consultations
138 Téléchargements

Altmetric

Partager

More