中国科学院自动化研究所机构知识库

Machine learning-based diagnosis methods have achieved remarkable success under the assumption that the training and test data are identically distributed. However, a critical requirement of these methods is the generalization capability to unseen domains when deployed to actual diagnosis scenarios. We introduce the challenging problem of domain generalization, i.e., learning from multiple source domains to produce a model that can directly generalize to unseen domains without target information. We adopt a model-agnostic learning produce that maximizes the dot product of gradients between the source domains. Such a gradient alignment objective encourages finding a common optimization path for all source domains, which helps to focus on invariant representations. Furthermore, we propose two feature regularizations that explicitly regularize the feature space. Global feature regularization aligns class relationships between different domains to preserve the domain-invariant knowledge. Local feature regularization encourages the model to learn domain-agnostic class-specific representations with intraclass compactness and interclass separability. The effectiveness of the proposed method is demonstrated with generalization experiments on two benchmarks.