arXiv reaDer
重心対照学習によるテールクラス表現の改善
Improving Tail-Class Representation with Centroid Contrastive Learning
視覚領域では、大規模な自然データセットは通常、頭と尾のクラスの間に大きなクラスの不均衡があるロングテール分布を示します。この分布は、テールクラスの適切な表現を学習するのに困難をもたらします。最近の開発では、トレーニングを表現学習と分類器のバランス調整に分離することで、優れたロングテールモデルを学習できることが示されています。しかし、これらの作品は、表現学習への長期的な影響について十分に考慮されていません。この作業では、ロングテール表現学習を改善するために、補間重心対照学習(ICCL)を提案します。 ICCLは、クラスに依存しないサンプラーとクラス認識のサンプラーから2つの画像を補間し、補間画像の表現を使用して両方のソースクラスの重心を取得できるようにモデルをトレーニングします。複数のロングテール画像分類ベンチマークでのアプローチの有効性を示します。私たちの結果は、実際のロングテール分布を使用したiNaturalist 2018データセットで2.8%の大幅な精度の向上を示しています。
In vision domain, large-scale natural datasets typically exhibit long-tailed distribution which has large class imbalance between head and tail classes. This distribution poses difficulty in learning good representations for tail classes. Recent developments have shown good long-tailed model can be learnt by decoupling the training into representation learning and classifier balancing. However, these works pay insufficient consideration on the long-tailed effect on representation learning. In this work, we propose interpolative centroid contrastive learning (ICCL) to improve long-tailed representation learning. ICCL interpolates two images from a class-agnostic sampler and a class-aware sampler, and trains the model such that the representation of the interpolative image can be used to retrieve the centroids for both source classes. We demonstrate the effectiveness of our approach on multiple long-tailed image classification benchmarks. Our result shows a significant accuracy gain of 2.8% on the iNaturalist 2018 dataset with a real-world long-tailed distribution.
updated: Tue Oct 19 2021 15:24:48 GMT+0000 (UTC)
published: Tue Oct 19 2021 15:24:48 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アソシエイト