arXiv reaDer
ロングテール認識のための対象を絞った教師あり対照学習
Targeted Supervised Contrastive Learning for Long-Tailed Recognition
実世界のデータは、多くの場合、クラスの不均衡が大きいロングテール分布を示します。この場合、多数派のクラスがトレーニングプロセスを支配し、少数派のクラスの決定境界を変更する可能性があります。最近、研究者は、ロングテール認識のための教師あり対照学習の可能性を調査し、それが強力なパフォーマンスの向上を提供することを実証しました。このホワイトペーパーでは、監視された対照学習がパフォーマンスの向上に役立つ一方で、過去のベースラインは、不均衡なデータ分布によってもたらされる均一性の低下に悩まされていることを示しています。この不十分な均一性は、特徴空間での分離性が低い少数派クラスのサンプルに現れます。この問題に対処するために、超球上の特徴分布の均一性を改善する、ターゲットを絞った教師あり対照学習(TSC)を提案します。 TSCは最初に、超球上に均一に分散されたターゲットのセットを生成します。次に、トレーニング中に、さまざまなクラスの機能をこれらの明確で均一に分散されたターゲットに収束させます。これにより、マイノリティクラスを含むすべてのクラスが、フィーチャスペースで均一な分布を維持し、クラス境界を改善し、ロングテールデータが存在する場合でもより良い一般化を提供します。複数のデータセットでの実験は、TSCがロングテール認識タスクで最先端のパフォーマンスを達成することを示しています。
Real-world data often exhibits long tail distributions with heavy class imbalance, where the majority classes can dominate the training process and alter the decision boundaries of the minority classes. Recently, researchers have investigated the potential of supervised contrastive learning for long-tailed recognition, and demonstrated that it provides a strong performance gain. In this paper, we show that while supervised contrastive learning can help improve performance, past baselines suffer from poor uniformity brought in by imbalanced data distribution. This poor uniformity manifests in samples from the minority class having poor separability in the feature space. To address this problem, we propose targeted supervised contrastive learning (TSC), which improves the uniformity of the feature distribution on the hypersphere. TSC first generates a set of targets uniformly distributed on a hypersphere. It then makes the features of different classes converge to these distinct and uniformly distributed targets during training. This forces all classes, including minority classes, to maintain a uniform distribution in the feature space, improves class boundaries, and provides better generalization even in the presence of long-tail data. Experiments on multiple datasets show that TSC achieves state-of-the-art performance on long-tailed recognition tasks.
updated: Sat Nov 27 2021 22:40:10 GMT+0000 (UTC)
published: Sat Nov 27 2021 22:40:10 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アソシエイト