Source-linked AI summary
Domain Adaptation with Asymmetrically-Relaxed Distribution Alignment
Yifan Wu, Ezra Winston, Divyansh Kaushik, Zachary Lipton
TL;DR
Unsupervised domain adaptation must transfer from labeled source data to unlabeled target data under distribution drift, but standard alignment can impose target-error costs under label mismatch. The paper proposes asymmetrically relaxed alignment and reports improved target performance under label shift, with comparable performance otherwise.
Problem
Domain-adversarial alignment can reduce two terms of a target-error bound while increasing another, and minimizing the standard objective under label mismatch can impose a positive lower bound on target error.
Source: §3 · pp. 5–6Method
The paper replaces exact latent-distribution matching with asymmetrically relaxed alignment, which is minimized when target-to-source latent density ratios are bounded and target support is contained in source support.
Source: §1 · p. 2Results
The relaxed matching losses improve target performance under label-distribution shift and perform comparably otherwise.
Source: §1 · p. 3Takeaways & Limitations
The approach provides target-performance guarantees under concrete distribution assumptions, while its connectedness assumptions may not cover completely disjoint domains.
Source: §1 · p. 2, §8 · p. 15The relaxed objective can admit alignments with poor target predictions even when a perfect predictive alignment exists, so minimizing it does not ensure that the good alignment is learned.
Source: §3 · p. 6
Abstract
from arXiv · showhide
Domain adaptation addresses the common problem when the target distribution generating our test data drifts from the source (training) distribution. While absent assumptions, domain adaptation is impossible, strict conditions, e.g. covariate or label shift, enable principled algorithms. Recently-proposed domain-adversarial approaches consist of aligning source and target encodings, often motivating this approach as minimizing two (of three) terms in a theoretical bound on target error. Unfortunately, this minimization can cause arbitrary increases in the third term, e.g. they can break down under shifting label distributions. We propose asymmetrically-relaxed distribution alignment, a new approach that overcomes some limitations of standard domain-adversarial algorithms. Moreover, we characterize precise assumptions under which our algorithm is theoretically principled and demonstrate empirical benefits on both synthetic and real datasets.
1 Introduction
The section frames unsupervised domain adaptation as transferring from labeled source data to unlabeled target data under distribution drift, and argues that exact latent-space alignment can harm target accuracy under label shift. It introduces asymmetrically relaxed alignment, theoretical guarantees under stated assumptions, and empirical gains under label shift with comparable performance otherwise.
1 Introduction
- The section notes that prior target-accuracy guarantees generally require assumptions, while support-coverage assumptions can fail and earlier out-of-support guarantees do not cover high-capacity hypothesis classes such as neural networks (Ben-David et al. (2010b); Ben-David et al. (2010a)).Source: §1 · pp. 1–2
- Domain-adversarial networks align latent distributions while minimizing source error, but this can increase the third term of the cited target-error bound under label shift (Ganin et al. (2016); Ben-David et al. (2010a)).In the paper’s example, 100% source accuracy still permits at most 75% target accuracy after exact alignment.Source: §1 · p. 2
- The proposed relaxed distance avoids requiring exact latent-distribution matching and is minimized when target-to-source latent density ratios are bounded and target support is contained in source support.The paper says this relaxation need not produce a poor target classifier under label-distribution mismatch.Source: §1 · p. 2
- The paper provides theoretical analysis showing that relaxed alignment can guarantee target-domain performance under a stated set of data-distribution assumptions.Source: §1 · p. 3
- The authors propose multiple distances with the desired properties that can be optimized through adversarial training.Source: §1 · p. 3
- The relaxed matching losses improve target performance under label-distribution shift and perform comparably to standard approaches otherwise.Source: §1 · p. 3
2 Preliminaries
The preliminaries define unsupervised domain adaptation as classification from labeled source data and unlabeled target data, then describe why target examples outside source support complicate risk control. They frame domain-adversarial learning as jointly reducing source risk and latent-space distance.
2 Preliminaries
- The formal setup uses binary labels, source and target input distributions, a representation map into a latent space, and a predictor that classifies the mapped input.Source: §2 · p. 4
- The setting provides labeled source examples and unlabeled target examples; finite-sample approximations are outside this work’s focus on generalization across distributions with different supports.Source: §2 · p. 4
- For target inputs outside source support, their true labels are unavailable, making the second term of the target-risk decomposition inaccessible and not directly controllable.Source: §2 · p. 5
- Domain-adversarial learning jointly minimizes source classification risk and latent-distribution distance, using distances such as Jensen-Shannon divergence, Wasserstein distance, or Maximum Mean Discrepancy.The section motivates an alternative objective because driving the distributions to be identical can lead to undesirable performance.Source: §2 · p. 5
3 A Motivating Scenario
Exact representation matching can impose positive target error when source and target label proportions differ. The paper motivates relaxing that condition with bounded density ratios, while noting that optimization may still fail to learn a good target alignment.
3 A Motivating Scenario
- When source and target label proportions differ, zero source error and exact representation matching impose a positive lower bound on target error.The proof maps some target positive examples into latent regions corresponding to source negatives while preserving label consistency for the rest.Source: §3 · p. 5
- A bounded target-to-source latent density ratio can relax exact matching while allowing a perfect target classifier when target support is contained in source support.The relaxed condition does not suffer from the stated problem under shifting label distributions.Source: §3 · p. 6
- The proposed approach replaces exact distribution matching with a β-admissible distance that is zero when the density ratio is bounded by 1 + β.The distance is used in place of the standard distribution distance in the domain-adversarial objective.Source: §3 · p. 6
- Choosing β involves a trade-off: larger β raises the bound on the third error term, while smaller β can make a good target classifier impossible under label shift.The justification also considers imperfect source classifiers.Source: §3 · p. 6
- Minimizing the objective does not ensure the desired alignment is learned, because other alignments can satisfy it while predicting poorly on target data.The paper identifies how to guarantee learning the desired alignment as an open question.Source: §3 · pp. 6–7
- The theoretical analysis addresses imperfect source classifiers and relaxed alignment, and gives assumptions under which a good target classifier can be learned.Source: §3 · p. 7
4 Bounding the Target Domain Error
The section argues that existing bounds do not adequately motivate domain-adversarial objectives, then proposes a target-error bound that treats distribution mismatch asymmetrically and controls label consistency through representation regularity and data connectivity.
4 Bounding the Target Domain Error
- Prior bounds leave the unobservable label-consistency term model-dependent, so its supremum can be vacuous; unrestricted binary H-divergence also requires exact distribution matching.Source: §4 · p. 7
- The proposed bound treats source-target mismatch asymmetrically and bounds label consistency using representation Lipschitzness, distribution separation, and connectedness.Source: §4 · p. 7
- The analysis can be non-vacuous, guaranteeing correct classification for some target points even where source density is zero.Source: §4 · p. 7
4.1 A general bound
- Under an L-Lipschitz representation satisfying Construction 4.1 and Assumption 4.2, target error is bounded by (1 + β)ES(φ, h) + 3δ1 + 2(1 + β)δ2 + δ3.The bound requires both the construction properties and the connectedness assumption.Source: §4.1 · p. 8
- The bound is general: its construction constants can be adjusted, and δ3 can then be chosen to satisfy Assumption 4.2.Source: §4.1 · p. 8
4.2 Example of a perfect target domain classifier
- If each connected, label-consistent target cluster overlaps source support and latent source classes remain separated under bounded density ratios, zero source error implies zero target error.The example uses source small-to-medium cats and dogs and target medium-to-large cats and dogs, with overlap among medium examples.Source: §4.2 · p. 9
5 Asymmetrically-relaxed distances
Section 5 develops β-admissible distances for relaxed alignment: f-divergence, modified Wasserstein, and reweighting constructions that can be optimized with adversarial training.
5.1 f-divergence
- The f-divergence construction partially linearizes f so its relaxed divergence is zero exactly when the representation-space density ratio is bounded by 1 + β.The resulting distance is β-admissible.Source: §5.1 · p. 10
5.2 Wasserstein distance
- The modified Wasserstein distance models transport to a region with density ratio at most 1 + β and is β-admissible when no transport is needed within that region.Its dual can be optimized adversarially using a non-negative function with approximate Lipschitz continuity.Source: §5.2 · pp. 10–11
5.3 Reweighting distance
- A reweighted distribution is formed by weighting q, and the relaxed distance minimizes the base distance over allowed reweightings.The β-admissible set corresponds to distributions whose density ratio to q is bounded by 1 + β.Source: §5.3 · p. 11
- Implicit reweighting by sorting optimizes the reweighted distance without parameterizing w when the base distance is adversarially trainable.The minibatch procedure sorts samples by f2(g(z)) and assigns weight 1 to the highest-scoring 1/(1 + β) fraction; the authors note it lacks a principled justification for minibatch adversarial training.Source: §5.3 · pp. 11–12
6 Experiments
The experiments compare DANN variants with source-only training across synthetic and MNIST–USPS domain adaptation, with and without label distribution shift.
6 Experiments
- The synthetic task uses balanced source labels and a 10%/90% target-label split, with small source–target overlap for each label.The experiments compare target accuracy across source-only, vanilla DANN, Wasserstein, f-divergence, and reweighting variants.Source: §6 · pp. 12–13
- On synthetic data, adaptation raises target accuracy from 89% (source-only) to 99%; on MNIST–USPS tasks, proposed approaches fare better than DANN under label shift and comparably without it.The synthetic result excludes cases where β is too small to admit a good target classifier; β ≥ 0.8 is required in the reported setting.Source: §6 · p. 13, Table 1 · p. 13, Table 2 · p. 14
- For MNIST–USPS, target data is sampled from half the digits to create label shift, while source data remains class-balanced.Both transfer directions are tested with and without label shift.Source: §6 · p. 13
7 Related work
The paper distinguishes its theory and algorithms from prior domain-adaptation analyses and reweighting approaches.
7 Related work
- The authors provide a risk bound for domain-adversarial methods using model-independent assumptions on data distributions.They contrast this with prior theories without contained-support assumptions, which either do not address latent-space adversarial learning or use φ-dependent constants or assumptions [Ben-David et al., 2007, 2010a; Ben-David & Urner, 2014; Mansour et al., 2009; Cortes & Mohri, 2011].Source: §7 · p. 14
- Prior domain-adversarial work demonstrates empirical success, while Cao et al. (2018a,b) address label shift through reweighting.The paper notes that this reweighting presumes a good classifier, creating a cyclic dependency.Source: §7 · p. 14
- The paper positions its algorithmic contribution against approaches including Tzeng et al. (2014), Ganin et al. (2016), Bousmalis et al. (2016), Hoffman et al. (2017), and Shu et al. (2018).Among the listed works, Cao et al. (2018a,b) are specifically described as handling label distribution shift.Source: §7 · p. 14
8 Conclusions
The conclusion presents asymmetrically-relaxed distances as an alternative to exact latent-space distribution matching, with guarantees limited by assumptions on the data.
8 Conclusions
- The paper replaces standard exact-matching objectives with asymmetrically-relaxed distances to address limitations under label distribution mismatch.It also states a theoretical guarantee for target-domain performance under data-distribution assumptions.Source: §8 · p. 15
- The stated guarantees depend on assumptions about the data distributions.The conclusion identifies connectedness assumptions as potentially not covering completely disjoint domains.Source: §8 · p. 15
- Analysis under other types of assumptions is identified as a possible direction for future work.Source: §8 · p. 15
A Proofs
The proofs establish a target-risk bound by controlling representation mismatch under connectedness conditions. They also derive the relaxed discriminator objective and its GAN specialization.
- For Proposition 3.2, a piecewise mapping sends source and target data into [0,1], rescaling the target class intervals by the class proportions.Source: A Proofs · p. 18
- Construction A.1 requires each x in C_T to connect within C_T to a same-label point in C_T∩(C0∪C1) through steps shorter than Δ/L.Source: A Proofs · p. 19
- Under Constructions 4.1 and A.1, the target/source risk-difference term is bounded by 2δ1 + 2(1 + β)δ2 + δ3.Source: A Proofs · pp. 21–22
- Assumption 4.2 yields Construction A.1 by choosing B_S=C0∪C1 and B_T=φ^-1(φ(C0∪C1)).Source: A Proofs · p. 22
- With δ1=δ2=δ3=0 under the stated corollary conditions, the target error satisfies E_T(φ,h) ≤ (1 + β)E_S(φ,h).Source: A Proofs · p. 22
- The relaxed f-divergence objective is a discriminator supremum with T constrained by f′(1/(1+β)); the appendix gives its GAN specialization.Source: A Proofs · p. 23
B Experiment Details
The experiment details specify synthetic and image-dataset setups, optimization choices, and two unresolved practical issues: encoder Lipschitz control and selecting β.
- Synthetic experiments used class-conditional Gaussian data with source and target means and covariance diagonals swapped, plus three-layer label nets and two-layer critics.Source: B Experiment Details · p. 23
- MNIST and USPS experiments subsampled 2,000 and 1,800 examples, respectively, with subsampling conditioned on the label distribution.Source: B Experiment Details · p. 23
- Across experiments, λ=1 and ADAM used learning rate 0.0001 and β1=0.5; φ and h received L2 regularization with coefficient 0.001.Source: B Experiment Details · p. 23
- WDANN1-2 reached only 20% accuracy in 2 of 5 runs; the low-accuracy runs had higher encoder-weight L2 norms.Source: B Experiment Details · p. 23
- L2 regularization may be insufficient for Lipschitz control: the authors associate larger encoder norms with likely higher L and leave stronger enforcement open.Source: B Experiment Details · pp. 23–24
- The authors found no clear β-performance pattern once β accommodates label shift, leaving selection open and suggesting moderate values such as 2 or 4.Source: B Experiment Details · p. 24