Adaptive Sample Aggregation In Transfer Learning

TL;DR

This paper introduces a unified approach to adaptive sample aggregation in transfer learning, leveraging divergence measures and moduli of transfer to improve target risk, with theoretical guarantees and applicability to complex scenarios.

Contribution

It identifies a unified framework for adaptively aggregating source and target data across multiple divergence measures using a weak modulus of transfer.

Findings

Unified algorithms adapt to various divergence measures.

Faster transfer rates are achievable under certain regimes.

Adaptive procedures work for both weak and strong moduli of transfer.

Abstract

Transfer Learning aims to optimally aggregate samples from a target distribution, with related samples from a so-called source distribution to improve target risk. Multiple procedures have been proposed over the last two decades to address this problem, each driven by one of a multitude of possible divergence measures between source and target distributions. A first question asked in this work is whether there exist unified algorithmic approaches that automatically adapt to many of these divergence measures simultaneously. We show that this is indeed the case for a large family of divergences proposed across classification and regression tasks, as they all happen to upper-bound the same measure of continuity between source and target risks, which we refer to as a weak modulus of transfer. This more unified view allows us, first, to identify algorithmic approaches that are simultaneously adaptive to these various divergence measures via a reduction to particular confidence sets. Second, it allows for a more refined understanding of the statistical limits of transfer under such divergences, and in particular, reveals regimes with faster rates than might be expected under coarser lenses. We then turn to situations that are not well captured by the weak modulus and corresponding divergences: these are situations where the aggregate of source and target data can improve target performance significantly beyond what's possible with either source or target data alone. We show that common such situations -- as may arise, e.g., under certain causal models with spurious correlations -- are well described by a so-called strong modulus of transfer which supersedes the weak modulus. We finally show that the strong modulus also admits adaptive procedures, which achieve near optimal rates in terms of the unknown strong modulus, and therefore apply in more general settings.