Connecting Domains and Contrasting Samples: A Ladder for Domain Generalization
By: Tianxin Wei , Yifan Chen , Xinrui He and more
Potential Business Impact:
Helps computers learn from different data types.
Distribution shifts between training and testing samples frequently occur in practice and impede model generalization performance. This crucial challenge thereby motivates studies on domain generalization (DG), which aim to predict the label on unseen target domain data by solely using data from source domains. It is intuitive to conceive the class-separated representations learned in contrastive learning (CL) are able to improve DG, while the reality is quite the opposite: users observe directly applying CL deteriorates the performance. We analyze the phenomenon with the insights from CL theory and discover lack of intra-class connectivity in the DG setting causes the deficiency. We thus propose a new paradigm, domain-connecting contrastive learning (DCCL), to enhance the conceptual connectivity across domains and obtain generalizable representations for DG. On the data side, more aggressive data augmentation and cross-domain positive samples are introduced to improve intra-class connectivity. On the model side, to better embed the unseen test domains, we propose model anchoring to exploit the intra-class connectivity in pre-trained representations and complement the anchoring with generative transformation loss. Extensive experiments on five standard DG benchmarks are performed. The results verify that DCCL outperforms state-of-the-art baselines even without domain supervision. The detailed model implementation and the code are provided through https://github.com/weitianxin/DCCL
Similar Papers
Domain Generalization in-the-Wild: Disentangling Classification from Domain-Aware Representations
CV and Pattern Recognition
Helps AI understand new things without seeing them.
Revisiting Theory of Contrastive Learning for Domain Generalization
Machine Learning (Stat)
Helps computers learn from new, unseen data.
CLIP-Powered Domain Generalization and Domain Adaptation: A Comprehensive Survey
CV and Pattern Recognition
Helps AI understand new things without retraining.