arXiv reaDer
ゼロショット画像分類のためのマスクされた教師なし自己トレーニング
Masked Unsupervised Self-training for Zero-shot Image Classification
最先端のコンピュータービジョンモデルは、ほとんどの場合、人間がラベル付けした画像を使用した教師あり学習でトレーニングされます。これは、高価な注釈コストのためにスケーラビリティを制限します。自己監視表現学習は目覚ましい進歩を遂げましたが、それでもラベル付けされたデータの微調整の第2段階が必要です。一方、大規模なテキスト画像監視(CLIPなど)で事前トレーニングされたモデルは、ダウンストリーム画像分類タスクへのゼロショット転送を可能にしました。ただし、CLIPのようなモデルのゼロショットのパフォーマンスは、実際の採用には不十分なことがよくあります。この論文では、豊富なラベルなしデータを活用して、ダウンストリームタスクで事前にトレーニングされたゼロショット分類器のパフォーマンスを向上させることを目指しています。マスクされた教師なし自己トレーニング(MUST)を提案します。これは、疑似ラベルと生の画像という2つの異なる補完的な監視ソースを活用する新しいアプローチです。クラスレベルのグローバル機能とピクセルレベルのローカル機能の両方を学習するために、3つの目的を共同で最適化し、2つの間の正則化を実施する必要があります。さまざまなドメインにわたる8つのダウンストリームタスクでのMUSTの有効性を示します。ここでは、CLIPが大幅に改善され、教師なし分類と教師あり分類の間のパフォーマンスのギャップが狭まります。たとえば、MUSTは、ViT-Bを使用してImageNetで77.7%のゼロショットトップ1精度を達成します。これは、CLIPよりも+ 9.4%高くなります。私たちのコードはhttps://github.com/salesforce/MUSTで入手できます。
State-of-the-art computer vision models are mostly trained with supervised learning using human-labeled images, which limits their scalability due to the expensive annotation cost. While self-supervised representation learning has achieved impressive progress, it still requires a second stage of finetuning on labeled data. On the other hand, models pre-trained with large-scale text-image supervision (e.g., CLIP) have enabled zero-shot transfer to downstream image classification tasks. However, the zero-shot performance of CLIP-like models are often insufficient for real-world adoption. In this paper, we aim to leverage the abundant unlabeled data to improve the performance of a pre-trained zero-shot classifier on downstream tasks. We propose Masked Unsupervised Self-Training (MUST), a new approach which leverages two different and complimentary sources of supervision: pseudo-labels and raw images. MUST jointly optimizes three objectives to learn both class-level global feature and pixel-level local feature and enforces a regularization between the two. We demonstrate the efficacy of MUST on 8 downstream tasks across a variety of domains, where it improves upon CLIP by a large margin and narrows the performance gap between unsupervised and supervised classification. For instance, MUST achieves a zero-shot top-1 accuracy of 77.7% on ImageNet using ViT-B, +9.4% higher than CLIP. Our code is available at https://github.com/salesforce/MUST.
updated: Tue Jun 07 2022 02:03:06 GMT+0000 (UTC)
published: Tue Jun 07 2022 02:03:06 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アソシエイト