本文にスキップ
AI News HubLIVE
サイト内リライト2 分で読了

視覚基盤モデルにおける微分可能検索による層別プロンプト融合の発見

記事の要約

本論文は、視覚プロンプトチューニングにおいて画像トークンとプロンプトトークンをどのように融合するかを研究する。融合方式の選択を二レベル最適化問題として定式化し、微分可能アーキテクチャ検索で解く。連結と加算に加えてアフィン変換とクロスアテンションを導入し、34データセットで一貫した改善を達成。ハイブリッド融合がViTの層意味論をより有効に活用することを示した。

ソースarXiv Computer Vision著者: Xi Xiao, Xingjian Li, Yunbei Zhang, Cheng Han, Tianming Liu, Tianyang Wang, Runmin Jiang, Jihun Hamm, Xiao Wang, Min Xu
視覚基盤モデルにおける微分可能検索による層別プロンプト融合の発見
誤りを報告

訂正窓口はまだ利用できません。記事情報をコピーして保存できます。

訂正案内
本文へ

視覚プロンプトチューニング(Visual Prompt Tuning)は、大規模な視覚Transformer(ViT)を下流タスクに適応させるためのパラメータ効率の良い微調整手法として広く用いられている。この手法では、入力空間または特徴空間に少数の学習可能なプロンプトトークンを導入し、画像トークンと共にTransformerの各層のアテンション計算に参加させる。しかし、画像トークンとプロンプトトークンをどのように融合すべきかという基本的かつ重要な問題は、これまで十分に検討されてこなかった。既存の手法の多くは連結(concatenation)や加算(addition)をそのまま採用しているが、単一の融合方式が常に最適であるか、また混合融合が追加の利得をもたらすかは明らかではない。

この問題に対処するため、Xi XiaoらによるECCV 2026採択論文「Layer-Specific Prompt Fusion Discovery via Differentiable Search in Vision Foundation Models」は、微分可能アーキテクチャ検索に基づく解決策を提案する。著者らは融合方式の選択とプロンプトパラメータの学習を二レベル最適化問題として統一的に定式化する。外側の最適化は各層でどの融合操作を採用するかを探索し、内側の最適化は学習可能なプロンプトを更新する。微分可能アーキテクチャ検索により、両方の変数を同時に最適化し、層ごとの融合戦略を自動的に発見することが可能となる。

探索空間を拡大するため、研究チームは従来の連結と加算に加えて、アフィン変換(affine transformation)とクロスアテンション(cross-attention)という2つの融合方式を新たに導入した。アフィン変換はプロンプトトークルをチャネルごとにスケーリングおよびシフトすることで、その特徴をより柔軟に調整できる。クロスアテンションは画像トークンとプロンプトトークンの間で双方向の情報交換を可能にし、単純なスタッキングや加算以上の相互作用を実現する。これら4つの候補操作が豊富なアーキテクチャ探索空間を構成し、探索プロセスが各層の意味特性に応じて最適な融合方式を選択できるようにする。

実験はVTAB-1k、FGVC、HTAの3つのベンチマークを含む34のデータセットで実施された。ViTバックボーンを凍結した条件下で、本手法は精度、レイテンシ、パラメータ数の間でVPT-Deepや最近の変種よりも有利なトレードオフを実現する。一貫した性能向上は、融合方式の選択が視覚プロンプトチューニングの効果に大きな影響を与えることを示しており、単一の融合方式では全ての層に適応することが難しいことを明らかにしている。

さらに論文は、ハイブリッド融合戦略がViTの異なる層の意味情報をより有効に活用できることを指摘する。浅い層では画像トークンとプロンプトトークンの独立性を保つ連結が適している一方、深い層ではクロスアテンションによる深い相互作用がより適している可能性がある。この発見は視覚プロンプトチューニング研究に新たな視点を提供する。つまり、プロンプト融合は固定されたハイパーパラメータとしてではなく、学習可能なアーキテクチャ決定として扱うべきである。本研究は、パラメータ効率の良い微調整分野における微分可能アーキテクチャ検索の応用可能性も示しており、今後の研究の方向性を切り開くものである。

要点と分析を開く

記事インテリジェンス

エンジニア上級

要点

  • プロンプト融合の選択を二レベル最適化として定式化し、微分可能アーキテクチャ検索で求解
  • 連結と加算に加え、アフィン変換とクロスアテンションを融合探索空間に追加
  • VTAB-1k、FGVC、HTAの34データセットでプロンプトチューニングベースラインを一貫して上回る
  • ハイブリッド融合戦略がViTの層別意味論をより有効に活用することを発見

要点と分析は自動生成され、誤りを含む場合があります。原典をご確認ください。