AmorLIP: Efficient Language-Image Pretraining via Amortization

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Sun, Haotian, Li, Yitong, Zhuang, Yuchen, He, Niao, Dai, Hanjun, Dai, Bo
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866908605164290048
author Sun, Haotian
Li, Yitong
Zhuang, Yuchen
He, Niao
Dai, Hanjun
Dai, Bo
author_facet Sun, Haotian
Li, Yitong
Zhuang, Yuchen
He, Niao
Dai, Hanjun
Dai, Bo
contents Contrastive Language-Image Pretraining (CLIP) has demonstrated strong zero-shot performance across diverse downstream text-image tasks. Existing CLIP methods typically optimize a contrastive objective using negative samples drawn from each minibatch. To achieve robust representation learning, these methods require extremely large batch sizes and escalate computational demands to hundreds or even thousands of GPUs. Prior approaches to mitigate this issue often compromise downstream performance, prolong training duration, or face scalability challenges with very large datasets. To overcome these limitations, we propose AmorLIP, an efficient CLIP pretraining framework that amortizes expensive computations involved in contrastive learning through lightweight neural networks, which substantially improves training efficiency and performance. Leveraging insights from a spectral factorization of energy-based models, we introduce novel amortization objectives along with practical techniques to improve training stability. Extensive experiments across 38 downstream tasks demonstrate the superior zero-shot classification and retrieval capabilities of AmorLIP, consistently outperforming standard CLIP baselines with substantial relative improvements of up to 12.24%.
format Preprint
id arxiv_https___arxiv_org_abs_2505_18983
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle AmorLIP: Efficient Language-Image Pretraining via Amortization
Sun, Haotian
Li, Yitong
Zhuang, Yuchen
He, Niao
Dai, Hanjun
Dai, Bo
Machine Learning
Computer Vision and Pattern Recognition
Contrastive Language-Image Pretraining (CLIP) has demonstrated strong zero-shot performance across diverse downstream text-image tasks. Existing CLIP methods typically optimize a contrastive objective using negative samples drawn from each minibatch. To achieve robust representation learning, these methods require extremely large batch sizes and escalate computational demands to hundreds or even thousands of GPUs. Prior approaches to mitigate this issue often compromise downstream performance, prolong training duration, or face scalability challenges with very large datasets. To overcome these limitations, we propose AmorLIP, an efficient CLIP pretraining framework that amortizes expensive computations involved in contrastive learning through lightweight neural networks, which substantially improves training efficiency and performance. Leveraging insights from a spectral factorization of energy-based models, we introduce novel amortization objectives along with practical techniques to improve training stability. Extensive experiments across 38 downstream tasks demonstrate the superior zero-shot classification and retrieval capabilities of AmorLIP, consistently outperforming standard CLIP baselines with substantial relative improvements of up to 12.24%.
title AmorLIP: Efficient Language-Image Pretraining via Amortization
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2505.18983