arXiv reaDer
双方向クロスアテンショントランスフォーマーを介したドメイン適応
Domain Adaptation via Bidirectional Cross-Attention Transformer
ドメインアダプテーション(DA)は、十分なラベル付きデータを持つソースドメインから学習した知識を、ラベルなしデータのみを持つターゲットドメインに活用することを目的としています。 DAに関する既存の研究のほとんどは、畳み込みベースのニューラルネットワークに基づいてドメインギャップを最小化することにより、両方のドメインのドメイン不変の特徴表現の学習に貢献しています。最近、ビジョントランスフォーマーは、複数のビジョンタスクのパフォーマンスを大幅に向上させました。ビジョントランスをベースに構築されたこの論文では、パフォーマンスの向上を目的として、DA用の双方向クロスアテンショントランス(BCAT)を提案します。提案されたBCATでは、アテンションメカニズムは、暗黙のソースとターゲットの混同機能表現を抽出して、ドメインの不一致を狭めることができます。具体的には、BCATでは、ドメイン不変の特徴表現を学習するために、双方向の相互注意メカニズムを備えた重み共有4分岐トランスを設計します。広範な実験により、提案されたBCATモデルは、畳み込みまたは変換器に基づく既存の最先端のDA手法よりも、4つのベンチマークデータセットで優れたパフォーマンスを達成することが実証されています。
Domain Adaptation (DA) aims to leverage the knowledge learned from a source domain with ample labeled data to a target domain with unlabeled data only. Most existing studies on DA contribute to learning domain-invariant feature representations for both domains by minimizing the domain gap based on convolution-based neural networks. Recently, vision transformers significantly improved performance in multiple vision tasks. Built on vision transformers, in this paper we propose a Bidirectional Cross-Attention Transformer (BCAT) for DA with the aim to improve the performance. In the proposed BCAT, the attention mechanism can extract implicit source and target mix-up feature representations to narrow the domain discrepancy. Specifically, in BCAT, we design a weight-sharing quadruple-branch transformer with a bidirectional cross-attention mechanism to learn domain-invariant feature representations. Extensive experiments demonstrate that the proposed BCAT model achieves superior performance on four benchmark datasets over existing state-of-the-art DA methods that are based on convolutions or transformers.
updated: Sat Jan 15 2022 16:49:56 GMT+0000 (UTC)
published: Sat Jan 15 2022 16:49:56 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アソシエイト