Transfer Learning for Benign Overfitting in High-Dimensional Linear Regression
By: Yeichan Kim, Ilmun Kim, Seyoung Park
Potential Business Impact:
Teaches computers to learn from new data faster.
Transfer learning is a key component of modern machine learning, enhancing the performance of target tasks by leveraging diverse data sources. Simultaneously, overparameterized models such as the minimum-$\ell_2$-norm interpolator (MNI) in high-dimensional linear regression have garnered significant attention for their remarkable generalization capabilities, a property known as benign overfitting. Despite their individual importance, the intersection of transfer learning and MNI remains largely unexplored. Our research bridges this gap by proposing a novel two-step Transfer MNI approach and analyzing its trade-offs. We characterize its non-asymptotic excess risk and identify conditions under which it outperforms the target-only MNI. Our analysis reveals free-lunch covariate shift regimes, where leveraging heterogeneous data yields the benefit of knowledge transfer at limited cost. To operationalize our findings, we develop a data-driven procedure to detect informative sources and introduce an ensemble method incorporating multiple informative Transfer MNIs. Finite-sample experiments demonstrate the robustness of our methods to model and data heterogeneity, confirming their advantage.
Similar Papers
Minimax optimal transfer learning for high-dimensional additive regression
Machine Learning (Stat)
Improves computer learning with new data.
Minimax optimal transfer learning for high-dimensional additive regression
Machine Learning (Stat)
Improves learning from different but related data.
Source-Optimal Training is Transfer-Suboptimal
Machine Learning (Stat)
Improves computer learning by finding the best way to teach it.