arXiv reaDer
視覚言語モデルを促す学習
Learning to Prompt for Vision-Language Models
CLIPのような事前に訓練された大規模な視覚言語モデルは、広範囲のダウンストリームタスク間で転送可能な表現を学習する上で大きな可能性を示しています。主に離散化されたラベルに基づく従来の表現学習とは異なり、視覚言語の事前トレーニングでは、画像とテキストを共通の特徴空間に配置します。これにより、プロンプトを介してダウンストリームタスクにゼロショット転送できます。つまり、分類の重みは次のように合成されます。関心のあるクラスを説明する自然言語。この作業では、このようなモデルを実際に展開するための主要な課題は、ドメインの専門知識を必要とし、非常に時間がかかる迅速なエンジニアリングであることを示します-言葉遣いのわずかな変更のため、言葉の調整にかなりの時間を費やす必要がありますパフォーマンスに大きな影響を与える可能性があります。自然言語処理(NLP)の迅速な学習研究の最近の進歩に触発されて、コンテキスト最適化(CoOp)を提案します。これは、CLIPのような視覚言語モデルをダウンストリーム画像認識に適応させるためのシンプルなアプローチです。具体的には、CoOpは、事前にトレーニングされたパラメーター全体を固定したまま、学習可能なベクトルを使用してプロンプトのコンテキストワードをモデル化します。さまざまな画像認識タスクを処理するために、CoOpの2つの実装を提供します。統合コンテキストとクラス固有のコンテキストです。 11個のデータセットでの広範な実験を通じて、CoOpが手作りのプロンプトを適切なマージンで打ち負かすのに必要なショットはわずか1〜2回であり、より多くのショットを使用すると大幅な改善が得られることを示しています。たとえば、16ショットの場合、平均ゲインは約15%(最高値は45%以上に達します)。学習ベースのアプローチであるにもかかわらず、CoOpは、手作りのプロンプトを使用するゼロショットモデルと比較して、優れたドメイン一般化パフォーマンスを実現します。
Large pre-trained vision-language models like CLIP have shown great potential in learning representations that are transferable across a wide range of downstream tasks. Different from the traditional representation learning that is based mostly on discretized labels, vision-language pre-training aligns images and texts in a common feature space, which allows zero-shot transfer to any downstream task via prompting, i.e., classification weights are synthesized from natural language describing classes of interest. In this work, we show that a major challenge for deploying such models in practice is prompt engineering, which requires domain expertise and is extremely time-consuming -- one needs to spend a significant amount of time on words tuning since a slight change in wording could have a huge impact on performance. Inspired by recent advances in prompt learning research in natural language processing (NLP), we propose Context Optimization (CoOp), a simple approach specifically for adapting CLIP-like vision-language models for downstream image recognition. Concretely, CoOp models a prompt's context words with learnable vectors while the entire pre-trained parameters are kept fixed. To handle different image recognition tasks, we provide two implementations of CoOp: unified context and class-specific context. Through extensive experiments on 11 datasets, we demonstrate that CoOp requires as few as one or two shots to beat hand-crafted prompts with a decent margin and is able to gain significant improvements when using more shots, e.g., with 16 shots the average gain is around 15% (with the highest reaching over 45%). Despite being a learning-based approach, CoOp achieves superb domain generalization performance compared with the zero-shot model using hand-crafted prompts.
updated: Sun Feb 06 2022 12:10:40 GMT+0000 (UTC)
published: Thu Sep 02 2021 17:57:31 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アソシエイト