arXiv reaDer
深い階層最適輸送による教師なしドメイン適応
Unsupervised Domain Adaptation via Deep Hierarchical Optimal Transport
教師なしドメイン適応は、ソースのラベル付きデータを利用して、ラベルのないターゲット ドメインの転送可能なモデルを推定することを目的とする挑戦的なタスクです。最適なトランスポート (OT) ベースの方法は、競争力のあるパフォーマンスにより、ドメイン適応の有望な方向性であることが最近証明されました。ただし、これらの方法のほとんどは、ソースとターゲットの分布を粗く調整しているため、ドメイン不変の表現を学習できるにもかかわらず、カテゴリ識別情報が混同される過剰調整の問題が発生します。この論文では、教師なしドメイン適応のための深層階層最適トランスポート法 (DeepHOT) を提案します。主なアイデアは、階層最適トランスポートを使用して、ドメイン データ間の豊富な構造相関をマイニングすることにより、ドメイン不変表現とカテゴリ識別表現の両方を学習することです。 DeepHOT フレームワークは、ドメイン レベルの OT と画像レベルの OT で構成され、後者は前者の地上距離メトリックとして使用されます。画像レベルの OT は、画像分類に有益なローカル画像領域の構造的関連性を捉えますが、ドメイン レベルの OT は、ドメインの基礎となるジオメトリを活用してドメイン不変表現を学習します。ただし、計算が非常に複雑なため、一部のシナリオでは最適なトランスポート ベースのモデルが制限されます。この目的のために、イメージレベルの OT でスライスされたワッサースタイン距離でオリジン OT を近似し、ドメインレベルの OT にミニバッチの不均衡な最適トランスポートを使用することにより、DeepHOT フレームワークの堅牢で効率的な実装を提案します。広範な実験により、DeepHOT が 4 つのベンチマーク データセットで最先端の方法を凌駕することが示されています。コードは GitHub で公開されます。
Unsupervised domain adaptation is a challenging task that aims to estimate a transferable model for unlabeled target domain by exploiting source labeled data. Optimal Transport (OT) based methods recently have been proven to be a promising direction for domain adaptation due to their competitive performance. However, most of these methods coarsely aligned source and target distributions, leading to the over-aligned problem where the category-discriminative information is mixed up although domain-invariant representations can be learned. In this paper, we propose a Deep Hierarchical Optimal Transport method (DeepHOT) for unsupervised domain adaptation. The main idea is to use hierarchical optimal transport to learn both domain-invariant and category-discriminative representations by mining the rich structural correlations among domain data. The DeepHOT framework consists of a domain-level OT and an image-level OT, where the latter is used as the ground distance metric for the former. The image-level OT captures structural associations of local image regions that are beneficial to image classification, while the domain-level OT learns domain-invariant representations by leveraging the underlying geometry of domains. However, due to the high computational complexity, the optimal transport based models are limited in some scenarios. To this end, we propose a robust and efficient implementation of the DeepHOT framework by approximating origin OT with sliced Wasserstein distance in image-level OT and using a mini-batch unbalanced optimal transport for domain-level OT. Extensive experiments show that DeepHOT surpasses the state-of-the-art methods in four benchmark datasets. Code will be released on GitHub.
updated: Mon Nov 21 2022 13:10:19 GMT+0000 (UTC)
published: Mon Nov 21 2022 13:10:19 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アソシエイト