Domain Generalization via Entropy Regularization

Domain generalization aims to learn from multiple source domains a predictive model that can generalize to unseen target domains. One essential problem in domain generalization is to learn discriminative domain-invariant features. To arrive at this, some methods introduce a domain discriminator through adversarial learning to match the feature distributions in multiple source domains. However, adversarial training can only guarantee that the learned features have invariant marginal distributions, while the invariance of conditional distributions is more important for prediction in new domains. To ensure the conditional invariance of learned features, we propose an entropy regularization term that measures the dependency between the learned features and the class labels. Combined with the typical task-related loss, e. g. , cross-entropy loss for classification, and adversarial loss for domain discrimination, our overall objective is guaranteed to learn conditional-invariant features across all source domains and thus can learn classifiers with better generalization capabilities. We demonstrate the effectiveness of our method through comparison with state-of-the-art methods on both simulated and real-world datasets. Code is available at: https: //github. com/sshan-zhao/DG via ER.

Authors

• Shanshan Zhao Mathematics Discipline, Shandong University
• Mingming Gong University of Melbourne Mohamed bin Zayed University of Artificial Intelligence
• Tongliang Liu Sydney AI Centre, The University of Sydney, Sydney
• Huan Fu
• Dacheng Tao College of Computing and Data Science, Nanyang Technological University

Keywords

No keywords are indexed for this paper.