arXiv reaDer
ドメイン一般化のための因果関係に基づく二重対比学習フレームワーク
Causality-based Dual-Contrastive Learning Framework for Domain Generalization
Domain Generalization (DG) は本質的に、複数のソース ドメインからモデルをトレーニングし、目に見えないターゲット ドメインに一般化する、分散外の一般化のサブブランチです。最近、いくつかのドメイン汎化アルゴリズムが登場しましたが、それらのほとんどは、転送不可能な複雑なアーキテクチャで設計されました。さらに、対照学習は、DG のシンプルさと効率性を実現する有望なソリューションになりました。しかし、既存の対照学習は、深刻なモデルの混乱を引き起こすドメイン シフトを無視していました。この論文では、機能とプロトタイプのコントラストに関するデュアルコントラスト学習 (DCL) モジュールを提案します。さらに、プロトタイプを達成するために単一の画像の多様なビューを融合するための新しい因果融合注意(CFA)モジュールを設計します。さらに、多様性シフトに関する情報を活用するために、Similarity-based Hard-pair Mining (SHM) 戦略を導入します。広範な実験により、3 つの DG データセットで、この方法が最先端のアルゴリズムよりも優れていることが示されています。提案されたアルゴリズムは、ドメイン ラベルを使用せずにプラグ アンド プレイ モジュールとしても機能します。
Domain Generalization (DG) is essentially a sub-branch of out-of-distribution generalization, which trains models from multiple source domains and generalizes to unseen target domains. Recently, some domain generalization algorithms have emerged, but most of them were designed with non-transferable complex architecture. Additionally, contrastive learning has become a promising solution for simplicity and efficiency in DG. However, existing contrastive learning neglected domain shifts that caused severe model confusions. In this paper, we propose a Dual-Contrastive Learning (DCL) module on feature and prototype contrast. Moreover, we design a novel Causal Fusion Attention (CFA) module to fuse diverse views of a single image to attain prototype. Furthermore, we introduce a Similarity-based Hard-pair Mining (SHM) strategy to leverage information on diversity shift. Extensive experiments show that our method outperforms state-of-the-art algorithms on three DG datasets. The proposed algorithm can also serve as a plug-and-play module without usage of domain labels.
updated: Sun Jan 22 2023 13:07:24 GMT+0000 (UTC)
published: Sun Jan 22 2023 13:07:24 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト