arXiv reaDer
ユニバーサルに制御可能な画像キャプションのための組み合わせプロンプトの学習
Learning Combinatorial Prompts for Universal Controllable Image Captioning
Controllable Image Captioning (CIC) は、与えられた制御信号の指導の下で画像に関する自然言語記述を生成するもので、次世代キャプション システムに向けた最も有望な方向性の 1 つです。これまでに、CICの制御信号は、内容に関する制御から構造に関する制御まで、さまざまな種類が提案されている。しかし、さまざまな制御信号のフォーマットとターゲットのギャップにより、既存のすべての CIC 作品 (またはアーキテクチャ) は 1 つの特定の制御信号のみに焦点を当てており、人間のような組み合わせ能力を見落としています。 「組み合わせ」とは、人間が記述を生成するときに複数のニーズ (または制約) を同時に簡単に満たすことができることを意味します。この目的のために、ComPro と呼ばれる Combinatorial Prompts を学習することで、CIC のための新しいプロンプトベースのフレームワークを提案します。事前学習済み言語モデル GPT-2 を言語モデルとして直接利用し、異なる信号固有の CIC アーキテクチャ間のギャップを埋めるのに役立ちます。次に、CIC をプロンプトガイド文生成問題として再定式化し、新しい軽量の言語モデルを提案します。プロンプト生成ネットワークを使用して、さまざまな種類の制御信号の組み合わせプロンプトを生成します。さまざまな制御信号に対して、プロンプトベースの CIC を実現するための新しいマスク アテンション メカニズムをさらに設計します。そのシンプルさにより、ComPro はさらに多くの種類に拡張できます2 つの一般的な CIC ベンチマークに関する広範な実験により、単一制御信号と結合制御信号の両方に対する ComPro の有効性と効率が検証されました。
Controllable Image Captioning (CIC) -- generating natural language descriptions about images under the guidance of given control signals -- is one of the most promising directions towards next-generation captioning systems. Till now, various kinds of control signals for CIC have been proposed, ranging from content-related control to structure-related control. However, due to the format and target gaps of different control signals, all existing CIC works (or architectures) only focus on one certain control signal, and overlook the human-like combinatorial ability. By ``combinatorial", we mean that our humans can easily meet multiple needs (or constraints) simultaneously when generating descriptions. To this end, we propose a novel prompt-based framework for CIC by learning Combinatorial Prompts, dubbed as ComPro. Specifically, we directly utilize a pretrained language model GPT-2 as our language model, which can help to bridge the gap between different signal-specific CIC architectures. Then, we reformulate the CIC as a prompt-guide sentence generation problem, and propose a new lightweight prompt generation network to generate the combinatorial prompts for different kinds of control signals. For different control signals, we further design a new mask attention mechanism to realize the prompt-based CIC. Due to its simplicity, our ComPro can be further extended to more kinds of combined control signals by concatenating these prompts. Extensive experiments on two prevalent CIC benchmarks have verified the effectiveness and efficiency of our ComPro on both single and combined control signals.
updated: Wed Aug 02 2023 12:56:53 GMT+0000 (UTC)
published: Sat Mar 11 2023 07:53:15 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アソシエイト