Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Frozen in Time: A Joint Video and Image Encoder for End-to-End Retrieval

About

Our objective in this work is video-text retrieval - in particular a joint embedding that enables efficient text-to-video retrieval. The challenges in this area include the design of the visual architecture and the nature of the training data, in that the available large scale video-text training datasets, such as HowTo100M, are noisy and hence competitive performance is achieved only at scale through large amounts of compute. We address both these challenges in this paper. We propose an end-to-end trainable model that is designed to take advantage of both large-scale image and video captioning datasets. Our model is an adaptation and extension of the recent ViT and Timesformer architectures, and consists of attention in both space and time. The model is flexible and can be trained on both image and video text datasets, either independently or in conjunction. It is trained with a curriculum learning schedule that begins by treating images as 'frozen' snapshots of video, and then gradually learns to attend to increasing temporal context when trained on video datasets. We also provide a new video-text pretraining dataset WebVid-2M, comprised of over two million videos with weak captions scraped from the internet. Despite training on datasets that are an order of magnitude smaller, we show that this approach yields state-of-the-art results on standard downstream video-retrieval benchmarks including MSR-VTT, MSVD, DiDeMo and LSMDC.

Max Bain, Arsha Nagrani, G\"ul Varol, Andrew Zisserman• 2021

Related benchmarks

TaskDatasetResultRank
Image-to-Text RetrievalFlickr30K 1K (test)
R@161
439
Text-to-Image RetrievalFlickr30k (test)
Recall@161
423
Text-to-Video RetrievalDiDeMo (test)
R@134.6
376
Text-to-Image RetrievalFlickr30K 1K (test)
R@161
375
Action RecognitionUCF101--
365
Text-to-Video RetrievalDiDeMo
R@131
360
Action RecognitionSomething-Something v2
Top-1 Accuracy61.6
341
Action RecognitionSomething-Something v2 (test)
Top-1 Acc55.1
333
Text-to-Video RetrievalMSR-VTT
Recall@131
313
Action RecognitionHMDB51 (test)--
249
Showing 10 of 92 rows
...

Other info

Code

Follow for update