Efficient Vision-and-Language Pre-training with Text-Relevant Image Patch Selection

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Ye, Wei, Jiang, Chaoya, Xu, Haiyang, Ye, Chenhao, Li, Chenliang, Yan, Ming, Zhang, Shikun, Huang, Songhang, Huang, Fei
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913262990262272
author Ye, Wei
Jiang, Chaoya
Xu, Haiyang
Ye, Chenhao
Li, Chenliang
Yan, Ming
Zhang, Shikun
Huang, Songhang
Huang, Fei
author_facet Ye, Wei
Jiang, Chaoya
Xu, Haiyang
Ye, Chenhao
Li, Chenliang
Yan, Ming
Zhang, Shikun
Huang, Songhang
Huang, Fei
contents Vision Transformers (ViTs) have become increasingly popular in large-scale Vision and Language Pre-training (VLP) models. Although previous VLP research has demonstrated the efficacy of ViTs, these efforts still struggle with computational inefficiencies caused by lengthy visual sequences. To address this challenge, we introduce an efficient VLP approach called TRIPS, which stands for Text-Relevant Image Patch Selection. TRIPS progressively reduces the visual sequence using a text-guided patch-selection layer in the visual backbone, thereby accelerating both training and inference processes. This patch-selection layer dynamically computes text-dependent visual attention, enabling it to identify attentive image tokens with text guidance and fuse inattentive ones in an end-to-end fashion. Importantly, TRIPS does not add any extra parameters and generalizes to most ViT-based VLP models. We incorporate TRIPS into three representative VLP models covering single-stream, dual-stream, and generative paradigms, and conduct extensive experiments on five widely-used multi-modal benchmark datasets. Our experimental results reveal that TRIPS delivers a 40% speedup, while maintaining competitive or superior performance on downstream tasks.
format Preprint
id arxiv_https___arxiv_org_abs_2403_07883
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Efficient Vision-and-Language Pre-training with Text-Relevant Image Patch Selection
Ye, Wei
Jiang, Chaoya
Xu, Haiyang
Ye, Chenhao
Li, Chenliang
Yan, Ming
Zhang, Shikun
Huang, Songhang
Huang, Fei
Computer Vision and Pattern Recognition
Artificial Intelligence
Vision Transformers (ViTs) have become increasingly popular in large-scale Vision and Language Pre-training (VLP) models. Although previous VLP research has demonstrated the efficacy of ViTs, these efforts still struggle with computational inefficiencies caused by lengthy visual sequences. To address this challenge, we introduce an efficient VLP approach called TRIPS, which stands for Text-Relevant Image Patch Selection. TRIPS progressively reduces the visual sequence using a text-guided patch-selection layer in the visual backbone, thereby accelerating both training and inference processes. This patch-selection layer dynamically computes text-dependent visual attention, enabling it to identify attentive image tokens with text guidance and fuse inattentive ones in an end-to-end fashion. Importantly, TRIPS does not add any extra parameters and generalizes to most ViT-based VLP models. We incorporate TRIPS into three representative VLP models covering single-stream, dual-stream, and generative paradigms, and conduct extensive experiments on five widely-used multi-modal benchmark datasets. Our experimental results reveal that TRIPS delivers a 40% speedup, while maintaining competitive or superior performance on downstream tasks.
title Efficient Vision-and-Language Pre-training with Text-Relevant Image Patch Selection
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2403.07883