A Unified Analysis of Generalization and Sample Complexity for Semi-Supervised Domain Adaptation

TL;DR

This paper provides a comprehensive theoretical analysis of semi-supervised domain adaptation, focusing on domain alignment methods like MMD and adversarial training, and derives bounds on generalization and sample complexity.

Contribution

It introduces new generalization bounds and sample complexity analysis for domain alignment techniques, including neural networks, in semi-supervised domain adaptation.

Findings

Sample complexity scales quadratically with network size.

Robustness improves with more labeled target data.

Theoretical bounds are supported by experiments.

Abstract

Domain adaptation seeks to leverage the abundant label information in a source domain to improve classification performance in a target domain with limited labels. While the field has seen extensive methodological development, its theoretical foundations remain relatively underexplored. Most existing theoretical analyses focus on simplified settings where the source and target domains share the same input space and relate target-domain performance to measures of domain discrepancy. Although insightful, these analyses may not fully capture the behavior of modern approaches that align domains into a shared space via feature transformations. In this paper, we present a comprehensive theoretical study of domain adaptation algorithms based on domain alignment. We consider the joint learning of domain-aligning feature transformations and a shared classifier in a semi-supervised setting. We first derive generalization bounds in a broad setting, in terms of covering numbers of the relevant function classes. We then extend our analysis to characterize the sample complexity of domain-adaptive neural networks employing maximum mean discrepancy (MMD) or adversarial objectives. Our results rely on a rigorous analysis of the covering numbers of these architectures. We show that, for both MMD-based and adversarial models, the sample complexity admits an upper bound that scales quadratically with network depth and width. Furthermore, our analysis suggests that in semi-supervised settings, robustness to limited labeled target data can be achieved by scaling the target loss proportionally to the square root of the number of labeled target samples. Experimental evaluation in both shallow and deep settings lends support to our theoretical findings.