私たちの目的は、大規模な画像およびビデオデータセットの言語ベースの検索です。このタスクでは、テキストとビジョンをジョイント埋め込みスペース(別名デュアルエンコーダー)に個別にマッピングするアプローチは、検索スケールとして魅力的であり、近似最近傍探索を使用して数十億の画像に対して効率的です。クロスアテンション付きのビジョンテキストトランスフォーマーを使用する別のアプローチでは、ジョイント埋め込みよりも精度が大幅に向上しますが、テスト時に各サンプルに必要なクロスアテンションメカニズムのコストを考えると、大規模な検索には実際には適用できないことがよくあります。 。この作品は、両方の長所を兼ね備えています。以下の3つの貢献をします。まず、トランスベースのモデルに新しいきめ細かいクロスアテンションアーキテクチャを装備し、スケーラビリティを維持しながら検索精度を大幅に向上させます。次に、高速デュアルエンコーダモデルと低速であるが正確なトランスベースモデルを蒸留と再ランク付けによって組み合わせるための一般的なアプローチを紹介します。最後に、Flickr30K画像データセットでのアプローチを検証します。ここでは、最先端の結果と競合する結果を得ながら、推論速度が数桁増加することを示しています。また、メソッドをビデオドメインに拡張し、VATEXデータセットの最新技術を改善します。
Our objective is language-based search of large-scale image and video datasets. For this task, the approach that consists of independently mapping text and vision to a joint embedding space, a.k.a. dual encoders, is attractive as retrieval scales and is efficient for billions of images using approximate nearest neighbour search. An alternative approach of using vision-text transformers with cross-attention gives considerable improvements in accuracy over the joint embeddings, but is often inapplicable in practice for large-scale retrieval given the cost of the cross-attention mechanisms required for each sample at test time. This work combines the best of both worlds. We make the following three contributions. First, we equip transformer-based models with a new fine-grained cross-attention architecture, providing significant improvements in retrieval accuracy whilst preserving scalability. Second, we introduce a generic approach for combining a Fast dual encoder model with our Slow but accurate transformer-based model via distillation and re-ranking. Finally, we validate our approach on the Flickr30K image dataset where we show an increase in inference speed by several orders of magnitude while having results competitive to the state of the art. We also extend our method to the video domain, improving the state of the art on the VATEX dataset.