arXiv reaDer
自己管理表現のより良い理解に向けて
Towards Better Understanding of Self-Supervised Representations
自己教師あり学習法は、下流の分類タスクで印象的な結果を示しています。ただし、学習した表現を理解して解釈する作業は限られています。この論文では、SimCLR、SwaV、MoCo V2、BYOLなどのいくつかの最先端の自己監視モデルの表現空間を研究します。クラスラベル情報を使用せずに、最初に、サンプルのさまざまなサブセットに対して非常にアクティブであり、画像内の固有の物理的属性に対応する識別機能を発見します。このような識別機能を使用すると、下流の線形分類に大きな影響を与えることなく、自己監視モデルの表現空間を最大50%圧縮できることを示します。次に、ラベル情報にアクセスせずに計算できるサンプルごとの自己監視表現品質スコア(またはQスコア)を提案します。 Q-Scoreは、識別機能を利用して、特定のサンプルがダウンストリーム分類タスクで誤って分類される可能性があるかどうかを確実に予測し、SimCLRで0.91のAUPRCを達成し、ImageNet-100でトレーニングされたBYOLを実現します。 Q-Scoreは、ImageNet-100、CIFAR-10、CIFAR-100、およびSTL-10の4つの自己監視ベースラインすべてで、精度が最大8%向上する低品質の表現を修正するための正則化用語としても使用できます。さらに、ヒートマップ分析を通じて、Qスコアの正則化が識別機能を強化し、機能ノイズを削減して、モデルの解釈可能性を向上させることを示します。
Self-supervised learning methods have shown impressive results in downstream classification tasks. However, there is limited work in understanding and interpreting their learned representations. In this paper, we study the representation space of several state-of-the-art self-supervised models including SimCLR, SwaV, MoCo V2 and BYOL. Without the use of class label information, we first discover discriminative features that are highly active for various subsets of samples and correspond to unique physical attributes in images. We show that, using such discriminative features, one can compress the representation space of self-supervised models up to 50% without affecting downstream linear classification significantly. Next, we propose a sample-wise Self-Supervised Representation Quality Score (or, Q-Score) that can be computed without access to any label information. Q-Score, utilizes discriminative features to reliably predict if a given sample is likely to be mis-classified in the downstream classification task achieving AUPRC of 0.91 on SimCLR and BYOL trained on ImageNet-100. Q-Score can also be used as a regularization term to remedy low-quality representations leading up to 8% relative improvement in accuracy on all 4 self-supervised baselines on ImageNet-100, CIFAR-10, CIFAR-100 and STL-10. Moreover, through heatmap analysis, we show that Q-Score regularization enhances discriminative features and reduces feature noise, thus improving model interpretability.
updated: Wed Jul 06 2022 18:10:44 GMT+0000 (UTC)
published: Thu Mar 03 2022 17:48:23 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アソシエイト