BUS:Efficient and Effective Vision-language Pre-training with Bottom-Up Patch Summarization

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Jiang, Chaoya, Xu, Haiyang, Ye, Wei, Ye, Qinghao, Li, Chenliang, Yan, Ming, Bi, Bin, Zhang, Shikun, Huang, Fei, Huang, Songfang
Natura: Preprint
Pubblicazione: 2023
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866917596434006016
author Jiang, Chaoya
Xu, Haiyang
Ye, Wei
Ye, Qinghao
Li, Chenliang
Yan, Ming
Bi, Bin
Zhang, Shikun
Huang, Fei
Huang, Songfang
author_facet Jiang, Chaoya
Xu, Haiyang
Ye, Wei
Ye, Qinghao
Li, Chenliang
Yan, Ming
Bi, Bin
Zhang, Shikun
Huang, Fei
Huang, Songfang
contents Vision Transformer (ViT) based Vision-Language Pre-training (VLP) models have demonstrated impressive performance in various tasks. However, the lengthy visual token sequences fed into ViT can lead to training inefficiency and ineffectiveness. Existing efforts address the challenge by either bottom-level patch extraction in the ViT backbone or top-level patch abstraction outside, not balancing training efficiency and effectiveness well. Inspired by text summarization in natural language processing, we propose a Bottom-Up Patch Summarization approach named BUS, coordinating bottom-level extraction and top-level abstraction to learn a concise summary of lengthy visual token sequences efficiently. Specifically, We incorporate a Text-Semantics-Aware Patch Selector (TSPS) into the ViT backbone to perform a coarse-grained visual token extraction and then attach a flexible Transformer-based Patch Abstraction Decoder (PAD) upon the backbone for top-level visual abstraction. This bottom-up collaboration enables our BUS to yield high training efficiency while maintaining or even improving effectiveness. We evaluate our approach on various visual-language understanding and generation tasks and show competitive downstream task performance while boosting the training efficiency by 50\%. Additionally, our model achieves state-of-the-art performance on many downstream tasks by increasing input image resolution without increasing computational costs over baselines.
format Preprint
id arxiv_https___arxiv_org_abs_2307_08504
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle BUS:Efficient and Effective Vision-language Pre-training with Bottom-Up Patch Summarization
Jiang, Chaoya
Xu, Haiyang
Ye, Wei
Ye, Qinghao
Li, Chenliang
Yan, Ming
Bi, Bin
Zhang, Shikun
Huang, Fei
Huang, Songfang
Computer Vision and Pattern Recognition
Vision Transformer (ViT) based Vision-Language Pre-training (VLP) models have demonstrated impressive performance in various tasks. However, the lengthy visual token sequences fed into ViT can lead to training inefficiency and ineffectiveness. Existing efforts address the challenge by either bottom-level patch extraction in the ViT backbone or top-level patch abstraction outside, not balancing training efficiency and effectiveness well. Inspired by text summarization in natural language processing, we propose a Bottom-Up Patch Summarization approach named BUS, coordinating bottom-level extraction and top-level abstraction to learn a concise summary of lengthy visual token sequences efficiently. Specifically, We incorporate a Text-Semantics-Aware Patch Selector (TSPS) into the ViT backbone to perform a coarse-grained visual token extraction and then attach a flexible Transformer-based Patch Abstraction Decoder (PAD) upon the backbone for top-level visual abstraction. This bottom-up collaboration enables our BUS to yield high training efficiency while maintaining or even improving effectiveness. We evaluate our approach on various visual-language understanding and generation tasks and show competitive downstream task performance while boosting the training efficiency by 50\%. Additionally, our model achieves state-of-the-art performance on many downstream tasks by increasing input image resolution without increasing computational costs over baselines.
title BUS:Efficient and Effective Vision-language Pre-training with Bottom-Up Patch Summarization
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2307.08504