近年、画像のキャプションが急速に進歩しています。ただし、大容量のメモリ ストレージと重い計算負荷が必要なため、これらのキャプション モデルをモバイル デバイスに展開することはできません。主な障害は、重い視覚的特徴抽出器 (つまり、オブジェクト検出器) と複雑なクロスモーダル フュージョン ネットワークにあります。この目的のために、リソースが限られたデバイス向けの軽量の画像キャプションである LightCap を提案します。コア デザインは、効率的な画像キャプションのために最近の CLIP モデルに基づいて構築されています。具体的には、一方では、時間のかかるオブジェクト検出器に頼ることなく、CLIP モデルを活用してコンパクト グリッドの特徴を抽出します。一方、CLIPの画像テキスト検索設計を画像キャプションシナリオに移すには、新しい視覚概念抽出器とクロスモーダル変調器を考案します。クロスモーダルフュージョンモデルと並列予測ヘッドを、シーケンシャルおよびアンサンブル蒸留によってさらに最適化します。慎重に設計されたアーキテクチャにより、モデルには 4,000 万のパラメーターしか含まれず、現在の最先端の方法と比較して、モデル サイズを 75% 以上、FLOP を 98% 以上節約できます。容量が少ないにもかかわらず、私たちのモデルは、COCO Karpathy テスト スプリットの 136.6 CIDEr など、一般的なデータセットで最先端のパフォーマンスを発揮します。提案された LightCap は、1 つの CPU のみを搭載したスマートフォンでテストしたところ、画像あたり 188 ミリ秒の高速な推論速度を示し、実用的なアプリケーションの準備が整っています。
Recent years have witnessed the rapid progress of image captioning. However, the demands for large memory storage and heavy computational burden prevent these captioning models from being deployed on mobile devices. The main obstacles lie in the heavyweight visual feature extractors (i.e., object detectors) and complicated cross-modal fusion networks. To this end, we propose LightCap, a lightweight image captioner for resource-limited devices. The core design is built on the recent CLIP model for efficient image captioning. To be specific, on the one hand, we leverage the CLIP model to extract the compact grid features without relying on the time-consuming object detectors. On the other hand, we transfer the image-text retrieval design of CLIP to image captioning scenarios by devising a novel visual concept extractor and a cross-modal modulator. We further optimize the cross-modal fusion model and parallel prediction heads via sequential and ensemble distillations. With the carefully designed architecture, our model merely contains 40M parameters, saving the model size by more than 75% and the FLOPs by more than 98% in comparison with the current state-of-the-art methods. In spite of the low capacity, our model still exhibits state-of-the-art performance on prevalent datasets, e.g., 136.6 CIDEr on COCO Karpathy test split. Testing on the smartphone with only a single CPU, the proposed LightCap exhibits a fast inference speed of 188ms per image, which is ready for practical applications.