Optimal Transport for Domain Adaptation
Nicolas CourtyRémi FlamaryDevis TuiaAlain Rakotomamonjy
Proposes a regularized optimal transport framework that aligns probability distributions between distinct domains while preserving class structure, providing a principled geometric solution for visual domain adaptation tasks.
Modern data analytics and machine learning applications frequently encounter dataset drift, where predictive models trained on data from one acquisition source perform poorly when deployed on target data gathered under different conditions, such as altered lighting, background noise, or distinct sensor properties. This issue is especially acute in unsupervised domain adaptation, where target datasets lack label annotations entirely. The article evaluates a framework designed to align source and target data distributions by formulating domain adaptation as an optimal transport problem, which mathematically determines a minimal-effort mapping to transform labeled source data into the target space.
The evaluated approach introduces a regularized discrete optimal transport model that calculates non-linear, sample-specific transformations between empirical probability distributions. To prevent unrelated source samples from collapsing onto identical target points, the authors incorporate domain-specific regularizers: a convex group-lasso penalty that groups samples by class and a graph Laplacian regularizer that preserves local neighborhood structures. The framework also adapts to semi-supervised settings when limited target labels are present and scales to large datasets through a generalized conditional gradient optimization algorithm that leverages fast matrix scaling operations. The methodology was tested across synthetic non-linear benchmarks and standard real-world computer vision datasets, including digit recognition (USPS and MNIST), facial recognition across poses (PIE), and multi-domain object recognition (Caltech-Office).
The experimental results demonstrate that regularized optimal transport consistently outperforms existing domain adaptation baselines. On synthetic non-linear data, optimal transport maintained lower classification error rates across escalating transformation severities, achieving nearly flawless adaptation below forty-degree rotations where standard methods deteriorated. In real-world visual benchmarks, the proposed group-lasso and Laplacian regularizers improved average accuracy compared to unadapted baselines and standard subspace alignment algorithms, achieving up to 63.90% mean accuracy on digits and 47.70% on standard object recognition features. Furthermore, when applied to deep learning representations, optimal transport boosted accuracy by over 20 percentage points on specific domain shifts, such as between webcam and digital SLR categories. In semi-supervised evaluations with three target labels per class, embedding labels directly into the transport plan increased average accuracy to 55.6%, outperforming post-transport labeling approaches.
These findings indicate that aligning datasets via optimal transport allows organizations to transfer existing labeled assets to new domains without retraining entire deep architectures or manually re-annotating target datasets. Unlike conventional linear subspace alignments, this approach accommodates complex, non-linear physical distortions while preserving class integrity, directly mitigating the risks, operational costs, and development timelines associated with data re-labeling. Practitioners seeking to deploy models across shifting environments should consider implementing optimal transport alignment as an intermediate calibration layer, selecting group-lasso regularization when class coherence is paramount and incorporating any available target labels directly into the transportation cost matrix.
While the method provides exact recovery guarantees for affine transformations and robust empirical performance across complex shifts, users must note key boundary conditions. Performance degrades when extreme transformations cause severe ambiguity or when severe label proportion imbalances exist between source and target sets. Additionally, selecting optimization hyperparameters without target validation labels remains challenging in fully unsupervised environments. Overall confidence in the approach is high for visual adaptation tasks, and future work should focus on developing physics-informed regularizers and extending the formulation to multi-domain adaptation pipelines.
- Paper: Analysis of Representations for Domain Adaptation, Shai Ben-David et al. (2006). This seminal work establishes the theoretical bounds on generalization error and distribution discrepancy across domains that underpin distribution-matching strategies in domain adaptation.
- Paper: A theory of learning from different domains, Shai Ben-David et al. (2010). It provides the foundational statistical learning theory and divergence measures bounding target error that justify learning aligned representations between source and target domains.
- Paper: Transfer Feature Learning with Joint Distribution Adaptation, Mingsheng Long et al. (2013). It demonstrates how aligning both marginal and conditional distributions via learned representations enables effective unsupervised visual domain adaptation.
- Paper: Adapting Visual Category Models to New Domains, Kate Saenko et al. (2010). It introduces the visual benchmark datasets and transformation-mapping formulation widely adopted to evaluate feature alignment under domain shifts.
- Paper: A Survey on Transfer Learning, Sinno Jialin Pan et al. (2010). This survey formalizes the taxonomy of transfer learning and feature-representation alignment across divergent source and target data distributions.
- Paper: Learning Transferable Features with Deep Adaptation Networks, Mingsheng Long et al. (2015). It extends representation-alignment strategies to deep neural network architectures using multi-kernel maximum mean discrepancy regularizers across task-specific layers.
- Paper: Unsupervised Domain Adaptation by Backpropagation, Yaroslav Ganin et al. (2015). It scales unsupervised domain alignment into an end-to-end deep learning framework via adversarial training and gradient reversal.
- Paper: Return of Frustratingly Easy Domain Adaptation, Baochen Sun et al. (2015). It provides a lightweight alternative for unsupervised domain alignment by directly matching second-order statistics between source and target distributions.
- Paper: Deep CORAL: Correlation Alignment for Deep Domain Adaptation, Baochen Sun et al. (2016). It builds on statistical alignment concepts by integrating covariance loss directly into deep convolutional neural network training.
- Paper: Conditional Adversarial Domain Adaptation, Mingsheng Long et al. (2017). It refines unsupervised adversarial distribution alignment by conditioning the domain discriminator on multi-class classifier predictions.
- Paper: Maximum Classifier Discrepancy for Unsupervised Domain Adaptation, Kuniaki Saito et al. (2017). It moves beyond standard global distribution matching by utilizing task-specific decision boundaries to align target representations.
- Paper: Deep Visual Domain Adaptation: A Survey, Mei Wang et al. (2018). This survey provides a comprehensive review of the modern deep visual domain adaptation paradigms that followed distribution-matching and optimal transport formulations.
