arXiv reaDer
ロングテール画像分類のための対照学習ベースのハイブリッドネットワーク
Contrastive Learning based Hybrid Networks for Long-Tailed Image Classification
識別可能な画像表現を学習することは、不均衡な場合の分類器の学習を容易にすることができるため、ロングテール画像分類において重要な役割を果たします。有望なパフォーマンスの対比学習が最近表現学習で示されたので、この作業では、効果的な教師あり対比学習戦略を探索し、分類の精度を高めるために、不均衡なデータからより良い画像表現を学習するように調整します。具体的には、画像表現を学習するための教師あり対照損失と分類器を学習するためのクロスエントロピー損失で構成される新しいハイブリッドネットワーク構造を提案します。学習は特徴学習から分類器学習に段階的に移行し、より良い特徴という考えを具体化します。より良い分類器を作成します。表現学習の対照的損失の2つのバリエーションを調査します。これらは形式は異なりますが、正規化された埋め込みスペースで同じクラスのサンプルをまとめ、異なるクラスのサンプルを押し離すという共通の考え方を共有しています。それらの1つは、最近提案された教師あり対照(SC)損失です。これは、同じクラスの陽性サンプルを組み込むことにより、最先端の教師なし対照損失の上に設計されています。もう1つは、標準的なSC損失での集中的なメモリ消費に対処し、限られたメモリバジェットの下でより多くの可能性を示す、典型的な教師あり対照(PSC)学習戦略です。 3つのロングテール分類データセットに関する広範な実験は、ロングテール分類における提案された対照学習ベースのハイブリッドネットワークの利点を示しています。
Learning discriminative image representations plays a vital role in long-tailed image classification because it can ease the classifier learning in imbalanced cases. Given the promising performance contrastive learning has shown recently in representation learning, in this work, we explore effective supervised contrastive learning strategies and tailor them to learn better image representations from imbalanced data in order to boost the classification accuracy thereon. Specifically, we propose a novel hybrid network structure being composed of a supervised contrastive loss to learn image representations and a cross-entropy loss to learn classifiers, where the learning is progressively transited from feature learning to the classifier learning to embody the idea that better features make better classifiers. We explore two variants of contrastive loss for feature learning, which vary in the forms but share a common idea of pulling the samples from the same class together in the normalized embedding space and pushing the samples from different classes apart. One of them is the recently proposed supervised contrastive (SC) loss, which is designed on top of the state-of-the-art unsupervised contrastive loss by incorporating positive samples from the same class. The other is a prototypical supervised contrastive (PSC) learning strategy which addresses the intensive memory consumption in standard SC loss and thus shows more promise under limited memory budget. Extensive experiments on three long-tailed classification datasets demonstrate the advantage of the proposed contrastive learning based hybrid networks in long-tailed classification.
updated: Fri Mar 26 2021 05:22:36 GMT+0000 (UTC)
published: Fri Mar 26 2021 05:22:36 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アソシエイト