対照的学習による不変表現の学習は、ドメイン一般化 (DG) で最先端のパフォーマンスを見てきました。このような成功にもかかわらず、この論文では、その中核となる学習戦略 (機能の調整) がモデルの一般化を大きく妨げる可能性があることを発見しました。ニューロンの解釈可能性の最近の進歩に触発されて、ニューロン活性化の観点からこの問題を特徴付けます。具体的には、特徴要素をニューロン活性化状態として扱うことにより、従来のアラインメント方法は、すべてのニューロン活性化の違いを無差別に最小化するため、学習した不変特徴の多様性を低下させる傾向があることを示します。これは代わりに、ニューロン間の豊富な関係を無視します。それらの多くは、異なる方法で出現するにもかかわらず、同じ視覚的概念を識別することがよくあります。この発見により、ニューロンにエンコードされた高レベルの概念を対比することにより、要素ごとの機能の配置を緩和する、シンプルで効果的なアプローチ、Concept Contrast (CoCo) を提示します。このアプローチは非常に柔軟で、DG の対照的な方法に統合できます。広範な実験を通じて、CoCo が機能表現の多様性を促進し、DomainBed ベンチマークよりもモデルの一般化機能を一貫して向上させることをさらに実証しました。
Learning invariant representations via contrastive learning has seen state-of-the-art performance in domain generalization (DG). Despite such success, in this paper, we find that its core learning strategy -- feature alignment -- could heavily hinder the model generalization. Inspired by the recent progress in neuron interpretability, we characterize this problem from a neuron activation view. Specifically, by treating feature elements as neuron activation states, we show that conventional alignment methods tend to deteriorate the diversity of learned invariant features, as they indiscriminately minimize all neuron activation differences. This instead ignores rich relations among neurons -- many of them often identify the same visual concepts though they emerge differently. With this finding, we present a simple yet effective approach, Concept Contrast (CoCo), which relaxes element-wise feature alignments by contrasting high-level concepts encoded in neurons. This approach is highly flexible and can be integrated into any contrastive method in DG. Through extensive experiments, we further demonstrate that our CoCo promotes the diversity of feature representations, and consistently improves model generalization capability over the DomainBed benchmark.