arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2305.17455cs.CVcs.CL

CrossGET:用于加速视觉-语言 Transformer 的跨引导 Token 集成

CrossGET: Cross-Guided Ensemble of Tokens for Accelerating Vision-Language Transformers

Tsinghua University(清华大学);Shanghai AI Laboratory(上海人工智能实验室);The University of Hong Kong(香港大学);Stanford University(斯坦福大学)

Dachuan Shi, Chaofan Tao, Anyi Rao, Zhendong Yang, Chun Yuan, Jiaqi Wang

AI总结:

本文提出通用视觉-语言 Transformer 加速框架 CrossGET,通过跨模态引导 token 匹配集成与完全图软匹配,在推理时自适应合并 token,兼顾效率、性能与广泛架构适用性。

摘要:

近期视觉-语言模型取得了巨大进展。然而,其计算成本也在急剧上升,这使得模型加速变得极其关键。为了追求更高效的视觉-语言 Transformer,本文提出 Cross-Guided Ensemble of Tokens(CrossGET),这是一个面向视觉-语言 Transformer 的通用加速框架。该框架在推理过程中实时、自适应地合并 token,在保持高性能的同时显著降低计算成本。CrossGET 具有两项主要创新:1)跨引导匹配与集成。CrossGET 利用跨模态引导的 token 匹配和集成,以有效利用跨模态信息,从而在模态无关模型(例如 CLIP)和模态依赖模型(例如 BLIP2)上都实现更广泛的适用性。2)完全图软匹配。CrossGET 为 token 匹配机制引入了一种算法,可确保可靠的匹配结果,同时提升可并行性和高效率。研究在图像-文本检索、视觉推理、图像描述和视觉问答等多种视觉-语言任务上进行了大量实验。在经典多模态架构和新兴多模态 LLM 上的表现证明了该框架的有效性和多功能性。代码可在 https://github.com/sdc17/CrossGET 获取。

英文摘要:

Recent vision-language models have achieved tremendous advances. However, their computational costs are also escalating dramatically, making model acceleration exceedingly critical. To pursue more efficient vision-language Transformers, this paper introduces Cross-Guided Ensemble of Tokens (CrossGET), a general acceleration framework for vision-language Transformers. This framework adaptively combines tokens in real-time during inference, significantly reducing computational costs while maintaining high performance. CrossGET features two primary innovations: 1) Cross-Guided Matching and Ensemble. CrossGET leverages cross-modal guided token matching and ensemble to effectively utilize cross-modal information, achieving wider applicability across both modality-independent models, e.g., CLIP, and modality-dependent ones, e.g., BLIP2. 2) Complete-Graph Soft Matching. CrossGET introduces an algorithm for the token-matching mechanism, ensuring reliable matching results while facilitating parallelizability and high efficiency. Extensive experiments have been conducted on various vision-language tasks, such as image-text retrieval, visual reasoning, image captioning, and visual question answering. The performance on both classic multimodal architectures and emerging multimodal LLMs demonstrates the framework's effectiveness and versatility. The code is available at https://github.com/sdc17/CrossGET.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑