arXiv reaDer
ラベルのない画像分類のためのマスクされた教師なし自己トレーニング
Masked Unsupervised Self-training for Label-free Image Classification
最先端のコンピューター ビジョン モデルは、ほとんどの場合、人間がラベル付けした画像を使用した教師あり学習でトレーニングされます。これは、高価な注釈コストのためにスケーラビリティを制限します。自己教師あり表現学習は目覚ましい進歩を遂げましたが、ラベル付けされたデータを微調整する第 2 段階がまだ必要です。一方、大規模なテキスト画像監視 (CLIP など) で事前にトレーニングされたモデルは、下流の画像分類タスクへのゼロショット転送を可能にしました。ただし、CLIP のようなモデルのゼロ ショット パフォーマンスは、多くの場合、実際の採用には不十分です。このホワイトペーパーでは、事前トレーニング済みモデルの教師なし微調整により、ターゲットドメインからの豊富なラベルなしデータを活用して、事前トレーニング済みゼロショット分類器のパフォーマンスを向上させることを目指しています。 Masked Unsupervised Self-Training (MUST) を提案します。これは、トレーニング信号の 2 つの異なる補完的なソースである疑似ラベルと生画像を活用する新しい教師なし適応方法です。クラスレベルのグローバル機能とピクセルレベルのローカル機能の両方を学習するために 3 つの目的を共同で最適化し、2 つの間の正則化を強制する必要があります。さまざまなダウンストリーム タスクでの MUST の有効性を実証し、CLIP を大幅に改善します。また、教師ありの少数ショット適応法よりも優れている必要があります。 ViT-B を使用した ImageNet で 77.7% のトップ 1 精度を達成し、CLIP より +9.4% 高く、16 ショット CLIP 適応より +6.2% 高い。コードは 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 from a target domain to improve the performance of a pre-trained zero-shot classifier, by unsupervised finetuning of the pre-trained model. We propose Masked Unsupervised Self-Training (MUST), a new unsupervised adaptation method which leverages two different and complementary sources of training signals: 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 a variety of downstream tasks, where it improves upon CLIP by a large margin. MUST also outperforms supervised few-shot adaptation methods. It achieves a top-1 accuracy of 77.7% on ImageNet using ViT-B, +9.4% higher than CLIP, and +6.2% higher than 16-shot CLIP adaptation. Our code is available at https://github.com/salesforce/MUST.
updated: Fri Mar 10 2023 01:15:56 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アソシエイト