arXiv reaDer
SmallCap: 軽量の画像キャプションは、検索の増強によって促されます
SmallCap: Lightweight Image Captioning Prompted with Retrieval Augmentation
画像キャプションの最近の進歩は、データとモデル サイズのスケーリングに重点を置いており、事前トレーニングと微調整のコストが大幅に増加しています。大規模なモデルの代替として、データストアから取得した入力画像と関連するキャプションに基づいてキャプションを生成する SmallCap を紹介します。私たちのモデルは軽量でトレーニングが高速です。学習済みのパラメーターは、事前トレーニング済みの CLIP エンコーダーと GPT-2 デコーダーの間に新しく導入されたクロスアテンション レイヤーにあるだけだからです。 SmallCap は、追加の微調整なしで新しいドメインに移行でき、データストアのコンテンツを簡単に置き換えることができるため、トレーニングなしで大規模なデータを活用できます。私たちの実験では、COCO のみでトレーニングされた SmallCap は、このベンチマークで競争力のあるパフォーマンスを発揮し、ターゲット ドメイン データからの取得のみによって、再トレーニングなしで他のドメインに転送されることが示されています。人によってラベル付けされたさまざまな Web データをトレーニングなしで活用することで、さらなる改善が達成されます。これは、目に見えない視覚的概念への一般化をテストするために設計された nocaps ベンチマークを含む、さまざまなドメインで効果的であることが証明されています。
Recent advances in image captioning have focused on scaling the data and model size, substantially increasing the cost of pre-training and finetuning. As an alternative to large models, we present SmallCap, which generates a caption conditioned on an input image and related captions retrieved from a datastore. Our model is lightweight and fast to train, as the only learned parameters are in newly introduced cross-attention layers between a pre-trained CLIP encoder and GPT-2 decoder. SmallCap can transfer to new domains without additional finetuning and can exploit large-scale data in a training-free fashion since the contents of the datastore can be readily replaced. Our experiments show that SmallCap, trained only on COCO, has competitive performance on this benchmark, and also transfers to other domains without retraining, solely through retrieval from target-domain data. Further improvement is achieved through the training-free exploitation of diverse human-labeled and web data, which proves to be effective for a range of domains, including the nocaps benchmark, designed to test generalization to unseen visual concepts.
updated: Tue Mar 28 2023 12:04:49 GMT+0000 (UTC)
published: Fri Sep 30 2022 09:03:22 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アソシエイト