Robust angle-based transfer learning in high dimensions

TL;DR

This paper introduces a novel angle-based transfer learning method for high-dimensional ridge regression that effectively leverages source models with limited target data, outperforming existing benchmarks.

Contribution

The paper proposes the angleTL method, unifying several transfer learning approaches and providing algorithms for multiple sources, with theoretical analysis and practical validation.

Findings

angleTL outperforms benchmark methods in simulations

The method effectively incorporates multiple source models

Theoretical analysis explains when transfer learning is beneficial

Abstract

Transfer learning aims to improve the performance of a target model by leveraging data from related source populations, which is known to be especially helpful in cases with insufficient target data. In this paper, we study the problem of how to train a high-dimensional ridge regression model using limited target data and existing regression models trained in heterogeneous source populations. We consider a practical setting where only the parameter estimates of the fitted source models are accessible, instead of the individual-level source data. Under the setting with only one source model, we propose a novel flexible angle-based transfer learning (angleTL) method, which leverages the concordance between the source and the target model parameters. We show that angleTL unifies several benchmark methods by construction, including the target-only model trained using target data alone, the source model fitted on source data, and distance-based transfer learning method that incorporates the source parameter estimates and the target data under a distance-based similarity constraint. We also provide algorithms to effectively incorporate multiple source models accounting for the fact that some source models may be more helpful than others. Our high-dimensional asymptotic analysis provides interpretations and insights regarding when a source model can be helpful to the target model, and demonstrates the superiority of angleTL over other benchmark methods. We perform extensive simulation studies to validate our theoretical conclusions and show the feasibility of applying angleTL to transfer existing genetic risk prediction models across multiple biobanks.