Leaner and Faster: Two-Stage Model Compression for Lightweight Text-Image Retrieval
Not on the map. Why?
Abstract (arXiv)
Current text-image approaches (e.g., CLIP) typically adopt dual-encoder architecture using pre-trained vision-language representation. However, these models still pose non-trivial memory requirements and substantial incremental indexing time, which makes them less practical on mobile devices. In this paper, we present an effective two-stage framework to compress large pre-trained dual-encoder for lightweight text-image retrieval. The resulting model is smaller (39% of the original), faster (1.6x/2.9x for processing image/text respectively), yet performs on par with or better than the original full model on Flickr30K and MSCOCO benchmarks. We also open-source an accompanying realistic mobile image search application.
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Builds on 19 of its 20 references that are in the map. First posted on arXiv 29 Apr 2022.
Builds on
19 of 20 references.
References · 19
- 2017Attention Is All You Needcited by 59,932
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2014Microsoft COCO: Common Objects in Contextcited by 17,498
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2015Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networkscited by 9,617
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
- 2021Scaling Up Visual and Vision-Language Representation Learning With Noisy Text Supervisioncited by 3,848
- 2015VQA: Visual Question Answeringcited by 2,489
- 2017Billion-scale similarity search with GPUscited by 2,477
- 2021SimCSE: Simple Contrastive Learning of Sentence Embeddingscited by 2,168
- 2017Decoupled Weight Decay Regularizationcited by 1,722
7 more
- 2019LXMERT: Learning Cross-Modality Encoder Representations from Transformerscited by 1,548
- 2015Flickr30k Entities: Collecting Region-to-Phrase Correspondences for Richer Image-to-Sentence Modelscited by 1,352
- 2020Oscar: Object-Semantics Aligned Pre-training for Vision-Language Taskscited by 1,213
- 2019TinyBERT: Distilling BERT for Natural Language Understandingcited by 1,153
- 2021ImageNet-21K Pretraining for the Massescited by 607
- 2018A Corpus for Reasoning About Natural Language Grounded in Photographscited by 457
- 2019UNITER: UNiversal Image-TExt Representation Learningcited by 225
Built on by
Citing papers per year: 2023: 3; 2024: 1.
4 papers.
- 2023ConaCLIP: Exploring Distillation of Fully-Connected Knowledge Interaction Graph for Lightweight Text-Image Retrievalcited by 3
- 2023MOSAIC: Multi-Object Segmented Arbitrary Stylization Using CLIPcited by 2
- 2024The Use of Multimodal Large Language Models to Detect Objects from Thermal Images: Transportation Applicationscited by 2
- 2023MCAD: Multi-teacher Cross-modal Alignment Distillation for efficient image-text retrievalcited by 1
- 2024The Use of Multimodal Large Language Models to Detect Objects from Thermal Images: Transportation Applications20 Jun 2024
- 2023MCAD: Multi-teacher Cross-modal Alignment Distillation for efficient image-text retrieval30 Oct 2023
- 2023MOSAIC: Multi-Object Segmented Arbitrary Stylization Using CLIP24 Sep 2023
- 2023ConaCLIP: Exploring Distillation of Fully-Connected Knowledge Interaction Graph for Lightweight Text-Image Retrieval28 May 2023