A Fistful of Words: Learning Transferable Visual Models from Bag-of-Words Supervision

dc.contributor.authorTejankar, Ajinkya
dc.contributor.authorSanjabi, Maziar
dc.contributor.authorWu, Bichen
dc.contributor.authorXie, Saining
dc.contributor.authorKhabsa, Madian
dc.contributor.authorPirsiavash, Hamed
dc.contributor.authorFirooz, Hamed
dc.date.accessioned2022-11-14T15:48:04Z
dc.date.available2022-11-14T15:48:04Z
dc.date.issued2022-01-06
dc.description.abstractUsing natural language as a supervision for training visual recognition models holds great promise. Recent works have shown that if such supervision is used in the form of alignment between images and captions in large training datasets, then the resulting aligned models perform well on zero-shot classification as downstream tasks2 . In this paper, we focus on teasing out what parts of the language supervision are essential for training zero-shot image classification models. Through extensive and careful experiments, we show that: 1) A simple Bag-of-Words (BoW) caption could be used as a replacement for most of the image captions in the dataset. Surprisingly, we observe that this approach improves the zero-shot classification performance when combined with word balancing. 2) Using a BoW pretrained model, we can obtain more training data by generating pseudo-BoW captions on images that do not have a caption. Models trained on images with real and pseudo-BoW captions achieve stronger zero-shot performance. On ImageNet-1k zero-shot evaluation, our best model, that uses only 3M image-caption pairs, performs on-par with a CLIP model trained on 15M image-caption pairs (31.5% vs 31.3%)en_US
dc.description.urihttps://arxiv.org/abs/2112.13884en_US
dc.format.extent16 pagesen_US
dc.genrejournal articlesen_US
dc.genrepreprintsen_US
dc.identifierdoi:10.13016/m2tcx0-pser
dc.identifier.urihttps://doi.org/10.48550/arXiv.2112.13884
dc.identifier.urihttp://hdl.handle.net/11603/26319
dc.language.isoen_USen_US
dc.relation.isAvailableAtThe University of Maryland, Baltimore County (UMBC)
dc.relation.ispartofUMBC Computer Science and Electrical Engineering Department Collection
dc.relation.ispartofUMBC Faculty Collection
dc.relation.ispartofUMBC Student Collection
dc.rightsThis item is likely protected under Title 17 of the U.S. Copyright Law. Unless on a Creative Commons license, for uses protected by Copyright Law, contact the copyright holder or the author.en_US
dc.titleA Fistful of Words: Learning Transferable Visual Models from Bag-of-Words Supervisionen_US
dc.typeTexten_US

Files

Original bundle
Now showing 1 - 1 of 1
Loading...
Thumbnail Image
Name:
2112.13884.pdf
Size:
3.53 MB
Format:
Adobe Portable Document Format
Description:
License bundle
Now showing 1 - 1 of 1
No Thumbnail Available
Name:
license.txt
Size:
2.56 KB
Format:
Item-specific license agreed upon to submission
Description: