Cited by 1,127Multimodal Machine Learning ApplicationsDomain Adaptation and Few-Shot LearningAdvanced Neural Network Applications

Sigmoid Loss for Language Image Pre-Training

Open access
Zhai, Xiaohua, Mustafa, Basil, Kolesnikov, Alexander, Beyer, Lucas·2023·2023 IEEE/CVF International Conference on Computer Vision (ICCV)

Abstract

We propose a simple pairwise sigmoid loss for imagetext pre-training. Unlike standard contrastive learning with softmax normalization, the sigmoid loss operates solely on image-text pairs and does not require a global view of the pairwise similarities for normalization. The sigmoid loss simultaneously allows further scaling up the batch size, while also performing better at smaller batch sizes. With only four TPUv4 chips, we can train a Base CLIP model at 4k batch size and a Large LiT model at 20k batch size, the latter achieves 84.5% ImageNet zero-shot accuracy in two days. This disentanglement of the batch size from the loss further allows us to study the impact of examples vs pairs and negative to positive ratio. Finally, we push the batch size to the extreme, up to one million, and find that the benefits of growing batch size quickly diminish, with a more reasonable batch size of 32k being sufficient. We hope our research motivates further explorations in improving the quality and efficiency of language-image pre-training.

Integrity signals

Registry

Crossref · 10.1109/iccv51070.2023.01100

Cites

5 of 5 references verified0 citation errors · 0 not checkable

Statistics

Recomputing p-values needs the paper’s full text.

Claims the paper makes

We couldn’t retrieve this paper’s full text, so its claims and self-check weren’t run. The registry and citation checks still apply.

Claims other papers make about this one

Each is checked against this paper’s own text: does it say what the citing paper says it does?

  • Existing foundation models for vision and language such as CLIP, SigLIP, and Llama 2 are capable of generalization stemming from Internet-scale pretraining.

    “existing foundation models for vision and language such as CLIP [8], SigLIP [9], and Llama 2 [10] are capable of these types of generalization and more, stemming from the priors captured by their Internet-scale pretraining datasets.”

    From OpenVLA: An Open-Source Vision-Language-Action Model (2024) · cites this paper as [9]

  • The OpenVLA model architecture consists of a fused vision backbone that combines the SigLIP and DinoV2 encoders.

    “The OpenVLA model architecture consists of a fused vision backbone that combines the SigLIP [9] and DinoV2 [25] encoders.”

    From OpenVLA: An Open-Source Vision-Language-Action Model (2024) · cites this paper as [9]

Cites

Flagged first, then by citations

Cited by

Sorted by citations

Authors
Zhai, X., Mustafa, B., Kolesnikov, A., & Beyer, L.
Affiliations
Google (Switzerland) (CH)
Venue
2023 IEEE/CVF International Conference on Computer Vision (ICCV)
Publisher
IEEE
Published
2023-10-01
Pages
11941-11952
OpenAlex
W4390873312
Cited by
1,127
References
5
Citations by year
2026: 575 · 2025: 471 · 2024: 79 · 2023: 2
FWCI
120.53
Type
conference paper
Language
en
Open access
Locations
  • IEEE/CVF International Conference on Computer Vision (ICCV) (publishedVersion)
  • arXiv (Cornell University) (submittedVersion, cc-by)
  • arXiv (Cornell University) (cc-by)
Funders
DeepMind
Topics
  • Physical Sciences › Computer Science › Computer Vision and Pattern Recognition › Multimodal Machine Learning Applications
  • Physical Sciences › Computer Science › Artificial Intelligence › Domain Adaptation and Few-Shot Learning
  • Physical Sciences › Computer Science › Computer Vision and Pattern Recognition › Advanced Neural Network Applications
Keywords
contrastive learning, pairwise loss, batch size scaling, CLIP, liver tumor segmentation, image-text pairs, zero-shot image classification

About these checks

Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.

Statistics and arithmetic are recomputed, not judged.

Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.

What counts as a claim ›How we measure confidence ›How we check this ›

Are you an author, or do you see a problem with this page? Request a review