vix.ing · top · new · best · stats

Characterizing and Understanding the Generalization Error of Transfer Learning with Gibbs Algorithm

2021/11/02 by Yuheng Bu, Bu, Yuheng, Gholamali Aminian +9 · 1 citation
Computer Science · Mathematics · #Domain Adaptation and Few-Shot Learning #FOS: Computer and information sciences #Information Theory (cs.IT) #Machine Learning (cs.LG) #Machine Learning (stat.ML) #Machine Learning and Algorithms #Machine Learning and ELM #cs.IT #cs.LG #math.IT #stat.ML

paper · pdf · doi:10.48550/arxiv.2111.01635

arxiv created 2021/11/02 · openalex publication_date 2021/11/02 · arxiv updated 2021/11/03 · openalex created_date 2021/11/08 · openalex updated_date 2026/07/28

Abstract

We provide an information-theoretic analysis of the generalization ability of Gibbs-based transfer learning algorithms by focusing on two popular transfer learning approaches, α-weighted-ERM and two-stage-ERM. Our key result is an exact characterization of the generalization behaviour using the conditional symmetrized KL information between the output hypothesis and the target training samples given the source samples. Our results can also be applied to provide novel distribution-free generalization error upper bounds on these two aforementioned Gibbs algorithms. Our approach is versatile, as it also characterizes the generalization errors and excess risks of these two Gibbs algorithms in the asymptotic regime, where they converge to the α-weighted-ERM and two-stage-ERM, respectively. Based on our theoretical results, we show that the benefits of transfer learning can be viewed as a bias-variance trade-off, with the bias induced by the source distribution and the variance induced by the lack of target samples. We believe this viewpoint can guide the choice of transfer learning algorithms in practice.

Citations

Cited by

Related