Deep Transfer Learning with Joint Adaptation Networks
Mingsheng LongHanhua ZhuJianmin WangMichael I. Jordan
Proposes Joint Adaptation Networks, an adversarial framework that aligns the joint distributions of multiple domain-specific layers via joint maximum mean discrepancy to achieve accurate deep domain adaptation with linear-time computation.
The article addresses the challenge of adapting deep learning models trained on one domain with abundant labels to a different target domain that lacks labels, where shifts in data distributions often degrade performance. This issue is increasingly relevant as organizations seek to deploy models across varied real-world settings without incurring high costs for new data labeling.
The article set out to develop and evaluate a method for aligning the joint distributions of features and labels across domains within deep networks to enable effective unsupervised domain adaptation.
The approach involved extending convolutional neural networks such as AlexNet and ResNet into Joint Adaptation Networks that minimize a Joint Maximum Mean Discrepancy criterion across multiple task-specific layers. Experiments used two standard benchmarks, Office-31 with six transfer tasks and ImageCLEF-DA with six tasks, comparing against prior transfer and deep learning baselines through classification accuracy on held-out target data.
The analysis shows that the proposed networks achieved higher average accuracy than previous methods, reaching 76.0 percent on Office-31 with AlexNet compared to 74.3 percent for the strongest baseline and 84.3 percent with ResNet compared to 82.2 percent. Gains were largest on difficult transfers where domains differed substantially, such as 58.3 percent versus 53.4 percent on one such task. The method also produced smaller measured distribution discrepancies than alternatives. An adversarial variant further improved results by maximizing distinguishability of the domains during training.
These outcomes indicate that explicitly matching joint distributions rather than marginal distributions alone yields more transferable features and classifiers, reducing the performance drop when moving models between domains. This can lower labeling expenses and accelerate deployment in new settings.
Further validation on additional datasets and tasks would strengthen confidence before broad adoption. The linear-time implementation supports scaling, yet performance remains sensitive to the weighting parameter between classification loss and distribution alignment.
- Paper: Learning Transferable Features with Deep Adaptation Networks, Mingsheng Long et al. (2015). This paper establishes the Deep Adaptation Network architecture using multi-kernel maximum mean discrepancy to align marginal feature distributions in deep neural networks, providing the immediate foundational framework that Joint Adaptation Networks extend to joint distributions.
- Paper: Deep Domain Confusion: Maximizing for Domain Invariance, Eric Tzeng et al. (2014). This work introduces deep domain confusion by optimizing an MMD loss directly within CNN representations, pioneering the discrepancy-based deep adaptation methodology built upon by JAN.
- Book: Domain-Adversarial Training of Neural Networks, Yaroslav Ganin et al. (2016). This work introduces domain-adversarial neural networks and the gradient reversal layer for domain adaptation, establishing the adversarial alignment concepts utilized in JAN's adversarial variant.
- Paper: How transferable are features in deep neural networks?, Jason Yosinski et al. (2014). This empirical study quantifies the transition from general to task-specific features across CNN layers, motivating the layer-specific adaptation strategy central to Joint Adaptation Networks.
- Paper: A theory of learning from different domains, Shai Ben-David et al. (2010). This foundational paper provides the learning-theoretic bounds relating target error to source error and domain divergence, supplying the core theoretical basis for deep discrepancy minimization.
- Paper: Adapting Visual Category Models to New Domains, Kate Saenko et al. (2010). This paper introduces the standard Office-31 benchmark and early metric-learning formulations for domain adaptation, establishing the canonical evaluation testbed used in JAN.
- Paper: Conditional Adversarial Domain Adaptation, Mingsheng Long et al. (2017). This paper extends joint feature-label alignment by conditioning adversarial discriminators on classifier prediction statistics via multilinear conditioning, advancing the joint adaptation paradigm of JAN.
- Paper: Adversarial Discriminative Domain Adaptation, Eric Tzeng et al. (2017). This work formulates an untied, adversarial discriminative domain adaptation framework that offers an alternative adversarial paradigm to the joint discrepancy minimization in JAN.
- Paper: Moment Matching for Multi-Source Domain Adaptation, Xingchao Peng et al. (2018). This study generalizes moment-matching and distribution alignment techniques from single-source settings to multi-source domain adaptation.
- Paper: Deep Hashing Network for Unsupervised Domain Adaptation, Hemanth Venkateswara et al. (2017). This work integrates multi-kernel maximum mean discrepancy domain alignment with deep hashing layers for scalable, fast cross-domain retrieval.
- Paper: Deep Visual Domain Adaptation: A Survey, Mei Wang et al. (2018). This survey provides a comprehensive synthesis of deep visual domain adaptation methods, framing discrepancy-based networks like JAN within the wider taxonomy of adversarial and reconstruction-based strategies.
- Paper: A Survey on Deep Transfer Learning, Chuanqi Tan et al. (2018). This survey provides a systematic taxonomy and evaluation of deep transfer learning paradigms, categorizing mapping-based architectures including JAN.
- Paper: A Comprehensive Survey on Transfer Learning, Fuzhen Zhuang et al. (2019). This comprehensive survey contextualizes joint distribution alignment methods alongside homogeneous and heterogeneous transfer learning approaches across multiple domains.
