Research

A Class of Geometric Structures in Transfer Learning: Minimax Bounds and Optimality

AISTATS

Authors

Published on

03/30/2022

We study the problem of transfer learning, observing that previous efforts to understand its information-theoretic limits do not fully exploit the geometric structure of the source and target domains. In contrast, our study first illustrates the benefits of incorporating a natural geometric structure within a linear regression model, which corresponds to the generalized eigenvalue problem formed by the Gram matrices of both domains. We next establish a finite-sample minimax lower bound, propose a refined model interpolation estimator that enjoys a matching upper bound, and then extend our framework to multiple source domains and generalized linear models. Surprisingly, as long as information is available on the distance between the source and target parameters, negative-transfer does not occur. Simulation studies show that our proposed interpolation estimator outperforms state-ofthe-art transfer learning methods in both moderate- and high-dimensional settings.

Please cite our work using the BibTeX below.

@misc{https://doi.org/10.48550/arxiv.2202.11685,
  doi = {10.48550/ARXIV.2202.11685},
  
  url = {https://arxiv.org/abs/2202.11685},
  
  author = {Zhang, Xuhui and Blanchet, Jose and Ghosh, Soumyadip and Squillante, Mark S.},
  
  keywords = {Machine Learning (cs.LG), Methodology (stat.ME), Machine Learning (stat.ML), FOS: Computer and information sciences, FOS: Computer and information sciences},
  
  title = {A Class of Geometric Structures in Transfer Learning: Minimax Bounds and Optimality},
  
  publisher = {arXiv},
  
  year = {2022},
  
  copyright = {Creative Commons Attribution 4.0 International}
}
Close Modal