arXiv reaDer
一般化された監視された対照的な学習フレームワーク
A Generalized Supervised Contrastive Learning Framework
自己監視表現学習における対照学習の最近の注目すべき成果に基づいて、監視対照学習(SupCon)は、バッチ対照アプローチを監視コンテキストに拡張し、ResNet上のさまざまなデータセットでクロスエントロピーを上回りました。この作業では、GenSCLを紹介します。これは、一般化された監視対象の対比損失によって、最新の画像ベースの正則化(Mixup-Cutmixなど)と知識蒸留(KD)をSupConにシームレスに適応させる一般化された監視対象の対比学習フレームワークです。一般化された監視対象の対比損失は、ラベルの類似性と潜在的な特徴の類似性の間のクロスエントロピーを測定する監視対象の対比損失をさらに拡張したものです。次に、モデルは、潜在空間のアンカーにコントラストをどの程度近づける必要があるかを学習できます。 GenSCLは、ラベル情報を明示的かつ完全に活用することで、従来のポジティブとネガティブの境界を打ち破り、事前にトレーニングされたあらゆる種類の教師分類子を利用できます。 Mixup-CutmixとKDを使用してGenSCLでトレーニングされたResNet-50は、外部データなしでCIFAR10とCIFAR100で97.6%と84.7%の最先端の精度を達成し、元のSupConで報告された結果を大幅に改善します(1.6%と8.2 %、 それぞれ)。 Pytorchの実装はhttps://t.ly/yuUOで入手できます。
Based on recent remarkable achievements of contrastive learning in self-supervised representation learning, supervised contrastive learning (SupCon) has successfully extended the batch contrastive approaches to the supervised context and outperformed cross-entropy on various datasets on ResNet. In this work, we present GenSCL: a generalized supervised contrastive learning framework that seamlessly adapts modern image-based regularizations (such as Mixup-Cutmix) and knowledge distillation (KD) to SupCon by our generalized supervised contrastive loss. Generalized supervised contrastive loss is a further extension of supervised contrastive loss measuring cross-entropy between the similarity of labels and that of latent features. Then a model can learn to what extent contrastives should be pulled closer to an anchor in the latent space. By explicitly and fully leveraging label information, GenSCL breaks the boundary between conventional positives and negatives, and any kind of pre-trained teacher classifier can be utilized. ResNet-50 trained in GenSCL with Mixup-Cutmix and KD achieves state-of-the-art accuracies of 97.6% and 84.7% on CIFAR10 and CIFAR100 without external data, which significantly improves the results reported in the original SupCon (1.6% and 8.2%, respectively). Pytorch implementation is available at https://t.ly/yuUO.
updated: Wed Jun 01 2022 10:38:21 GMT+0000 (UTC)
published: Wed Jun 01 2022 10:38:21 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アソシエイト