Aminian, Gholamali; Masiha, Saeed; Toni, Laura; Rodrigues, Miguel RD; (2024) Learning Algorithm Generalization Error Bounds via Auxiliary Distributions. IEEE Journal on Selected Areas in Information Theory 10.1109/jsait.2024.3391900. (In press).

Preview

Text Rodrigues_Learning Algorithm Generalization Error Bounds via Auxiliary Distributions_AAM.pdf Download (688kB) | Preview

Abstract

Generalization error bounds are essential for comprehending how well machine learning models work. In this work, we suggest a novel method, i.e., the Auxiliary Distribution Method, that leads to new upper bounds on expected generalization errors that are appropriate for supervised learning scenarios. We show that our general upper bounds can be specialized under some conditions to new bounds involving the α-Jensen-Shannon, α-Rényi (0<α<1) information between a random variable modeling the set of training samples and another random variable modeling the set of hypotheses. Our upper bounds based on α-Jensen-Shannon information are also finite. Additionally, we demonstrate how our auxiliary distribution method can be used to derive the upper bounds on excess risk of some learning algorithms in the supervised learning context and the generalization error under the distribution mismatch scenario in supervised learning algorithms, where the distribution mismatch is modeled as α-Jensen-Shannon or α-Rényi divergence between the distribution of test and training data samples distributions. We also outline the conditions for which our proposed upper bounds might be tighter than other earlier upper bounds.

Download activity - last month

Export as CSVJSONXML

Download activity - last 12 months

Export as CSVJSONXML

Downloads by country - last 12 months

Export as JSONCSVXML

Archive Staff Only

View Item