arXiv reaDer
一般化パラメトリック対照学習
Generalized Parametric Contrastive Learning
この論文では、不均衡データと均衡データの両方でうまく機能する一般化パラメトリック対照学習 (GPaCo/PaCo) を提案します。理論的分析に基づいて、教師付きコントラスト損失は高頻度クラスにバイアスをかける傾向があるため、不均衡な学習の難しさが増すことがわかります。最適化の観点からバランスを再調整するために、一連のパラメトリックなクラスごとの学習可能なセンターを導入します。さらに、バランスの取れた設定で GPaCo/PaCo 損失を分析します。私たちの分析は、より多くのサンプルが対応するセンターと一緒に引き出され、ハードサンプル学習に役立つため、GPaCo / PaCoは同じクラスのサンプルを近づける強度を適応的に強化できることを示しています。ロングテール ベンチマークの実験では、ロングテール認識の新しい最先端技術が明らかになりました。完全な ImageNet では、CNN から GPaCo 損失でトレーニングされたビジョン トランスフォーマーまでのモデルは、MAE モデルと比較して、より優れた一般化パフォーマンスとより強力なロバスト性を示します。さらに、GPaCo はセマンティック セグメンテーション タスクに適用でき、4 つの最も一般的なベンチマークで明らかな改善が見られます。コードは https://github.com/dvlab-research/Parametric-Contrastive-Learning で入手できます。
In this paper, we propose the Generalized Parametric Contrastive Learning (GPaCo/PaCo) which works well on both imbalanced and balanced data. Based on theoretical analysis, we observe that supervised contrastive loss tends to bias high-frequency classes and thus increases the difficulty of imbalanced learning. We introduce a set of parametric class-wise learnable centers to rebalance from an optimization perspective. Further, we analyze our GPaCo/PaCo loss under a balanced setting. Our analysis demonstrates that GPaCo/PaCo can adaptively enhance the intensity of pushing samples of the same class close as more samples are pulled together with their corresponding centers and benefit hard example learning. Experiments on long-tailed benchmarks manifest the new state-of-the-art for long-tailed recognition. On full ImageNet, models from CNNs to vision transformers trained with GPaCo loss show better generalization performance and stronger robustness compared with MAE models. Moreover, GPaCo can be applied to the semantic segmentation task and obvious improvements are observed on the 4 most popular benchmarks. Our code is available at https://github.com/dvlab-research/Parametric-Contrastive-Learning.
updated: Sat May 27 2023 06:40:55 GMT+0000 (UTC)
published: Mon Sep 26 2022 03:49:28 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アソシエイト