arXiv reaDer
ドメイン適応のための確率的対照学習
Probabilistic Contrastive Learning for Domain Adaptation
対照学習は、自己教師ありの方法で特徴の識別能力を大幅に向上させることができ、さまざまな視覚タスクで目覚ましい成功を収めています。しかし、望ましくないことに、標準的な対照パラダイム (特徴 + ℓ_2 正規化) はドメイン適応にほとんど役に立たないことが観察されています。この研究では、この現象を詳しく調査し、主な理由が、認識には不可欠であるにもかかわらず、最適化では無視されているクラスの重み (最終的な完全に接続された層の重み) によるものであることを発見しました。この問題に取り組むために、私たちは、シンプルかつ強力な確率的対照学習 (PCL) を提案します。これは、識別特徴の抽出を支援するだけでなく、それらをクラスの重みの周りに強制的にクラスター化します。具体的には、ℓ_2 正規化を削除し、特徴を確率に置き換えることによって、標準的な対照パラダイムを打ち破ります。このようにして、PCL はワンホット フォームに近似する確率を強制することができ、それによって特徴とクラスの重みの間の偏差を減らすことができます。 PCL はその簡潔さの利点を活かして、さまざまな設定にうまく一般化できます。この作業では、5 つのタスクについて広範な実験を実施し、教師なしドメイン適応 (UDA)、半教師ありドメイン適応 (SSDA)、半教師あり学習 (SSL)、UDA 検出、UDA セマンティック セグメンテーションという一貫したパフォーマンスの向上を観察しました。 。特に、SYNTHIA 上の UDA セマンティック セグメンテーションの場合、PCL は平均 IoU の点で洗練された CPSL-D を >\!2% 上回っており、トレーニング コストははるかに低くなります (PCL: 1*3090、5 日対 CPSL-D: 4*V100) 、11日間)。コードは https://github.com/ljjcoder/Probabilistic-Contrastive-Learning で入手できます。
Contrastive learning can largely enhance the feature discriminability in a self-supervised manner and has achieved remarkable success for various visual tasks. However, it is undesirably observed that the standard contrastive paradigm (features+ℓ_2 normalization) only brings little help for domain adaptation. In this work, we delve into this phenomenon and find that the main reason is due to the class weights (weights of the final fully connected layer) which are vital for the recognition yet ignored in the optimization. To tackle this issue, we propose a simple yet powerful Probabilistic Contrastive Learning (PCL), which does not only assist in extracting discriminative features but also enforces them to be clustered around the class weights. Specifically, we break the standard contrastive paradigm by removing ℓ_2 normalization and replacing the features with probabilities. In this way, PCL can enforce the probability to approximate the one-hot form, thereby reducing the deviation between the features and class weights. Benefiting from the conciseness, PCL can be well generalized to different settings. In this work, we conduct extensive experiments on five tasks and observe consistent performance gains, i.e., Unsupervised Domain Adaptation (UDA), Semi-Supervised Domain Adaptation (SSDA), Semi-Supervised Learning (SSL), UDA Detection, and UDA Semantic Segmentation. Notably, for UDA Semantic Segmentation on SYNTHIA, PCL surpasses the sophisticated CPSL-D by >\!2% in terms of mean IoU with a much smaller training cost (PCL: 1*3090, 5 days v.s. CPSL-D: 4*V100, 11 days). Code is available at https://github.com/ljjcoder/Probabilistic-Contrastive-Learning.
updated: Sun Jul 16 2023 14:07:07 GMT+0000 (UTC)
published: Thu Nov 11 2021 02:08:07 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アソシエイト