この作業の目的は、事前トレーニング済みのビジュアル基盤モデルをセマンティック セグメンテーションのさまざまなダウンストリーム タスクに効果的かつ効率的に適応させる方法を探ることです。以前の方法では通常、特定のデータセットごとにネットワーク全体を微調整していましたが、これらのネットワークの大量のパラメーターを保存するには負担がかかります。最近のいくつかの研究では、いくつかの追加のトレーニング可能なパラメーターを凍結されたネットワークに挿入して、パラメーターを効率的に調整するための視覚的なプロンプトを学習しようとしました。ただし、これらの作品はトランスフォーマー専用に設計されているため、一般性に乏しいものでした。さらに、これらのスキームで限られた情報を使用すると、有益なプロンプトを学習する能力が低下しました。これらの問題を軽減するために、一般的で効果的なビジュアル プロンプト チューニングのための新しいステージごとのプロンプト マッチ フレームワークを提案します。具体的には、一般性を確保するために、凍結されたパラメーターを使用して事前トレーニングされたバックボーンを複数の段階に分割し、異なる段階間で迅速な学習を実行します。これにより、提案されたスキームが CNN および Transformer のさまざまなアーキテクチャに適用可能になります。効果的なチューニングのために、軽量のセマンティック認識 Prompt Matcher (SPM) は、中間セマンティック マップの豊富な情報に導かれて、反復メカニズムを使用して合理的なプロンプトを段階的に学習するように設計されています。提案された SPM は、表現学習のディープ マッチド フィルターとして機能し、前の段階の出力を次の段階の望ましい入力にうまく変換できるため、事前にトレーニングされた知識のより良いマッチング/刺激を実現できます。 4 つのベンチマークでの広範な実験により、提案されたスキームがパラメーター効率とパフォーマンス効果の間の有望なトレードオフを達成できることが実証されています。私たちのコードとモデルがリリースされます。
The objective of this work is to explore how to effectively and efficiently adapt pre-trained visual foundation models to various downstream tasks of semantic segmentation. Previous methods usually fine-tuned the entire networks for each specific dataset, which will be burdensome to store massive parameters of these networks. A few recent works attempted to insert some extra trainable parameters into the frozen networks to learn visual prompts for parameter-efficient tuning. However, these works showed poor generality as they were designed specifically for Transformers. Moreover, using limited information in these schemes, they exhibited a poor capacity to learn beneficial prompts. To alleviate these issues, we propose a novel Stage-wise Prompt-Matched Framework for generic and effective visual prompt tuning. Specifically, to ensure generality, we divide the pre-trained backbone with frozen parameters into multiple stages and perform prompt learning between different stages, which makes the proposed scheme applicable to various architectures of CNN and Transformer. For effective tuning, a lightweight Semantic-aware Prompt Matcher (SPM) is designed to progressively learn reasonable prompts with a recurrent mechanism, guided by the rich information of interim semantic maps. Working as deep matched filter of representation learning, the proposed SPM can well transform the output of the previous stage into a desirable input for the next stage, thus achieving the better matching/stimulating for the pre-trained knowledge. Extensive experiments on four benchmarks demonstrate that the proposed scheme can achieve a promising trade-off between parameter efficiency and performance effectiveness. Our code and models will be released.