压缩外部知识提升视觉问答效率,兼顾准确与速度。
Learning to Compress Contexts for Efficient Knowledge-based Visual Question Answering
- 用可学习的压缩机制将检索知识转为紧凑的键值缓存
- 在OK-VQA上达63.92%准确率,推理延迟降低22%-59.7%
- 兼容多种模型与知识源,适合实际部署场景
多模态大语言模型在视觉问答任务中表现优异。但在知识增强型视觉问答(KB-VQA)中,模型可能缺乏特定领域知识,需从外部知识源检索信息。以往方法如RAVQA-v2倾向于使用尽可能多的输入信息(如图像描述和检索知识)以提升性能,但随着输入令牌数增加,推理效率显著下降,难以满足实际应用需求。为此,本文提出检索增强型多模态大模型压缩上下文(RACC)。RACC学习对给定图像-问题对的检索知识进行压缩与聚合,生成紧凑的键值(KV)缓存,以调制下游冻结的MLLM,实现高效推理。RACC在OK-VQA上达到63.92%的SOTA准确率,相比RAVQA-v2推理延迟降低22.0%-59.7%。大量实验表明其具有广泛适用性,兼容多种现成的MLLMs,并可处理文本与多模态文档等多种知识源。
原文摘要 · Abstract (English)
Multimodal large language models (MLLMs) have demonstrated great performance on visual question answering (VQA). When it comes to knowledge-based Visual Question Answering (KB-VQA), MLLMs may lack the specialized domain knowledge needed to answer questions, necessitating the retrieval of necessary information from external knowledge sources. Previous works like Retrival-Augmented VQA-v2 (RAVQA-v2) focus on utilizing as much input information, such as image-based textual descriptions and retrieved knowledge, as possible to improve performance, but they all overlook the issue that with the number of input tokens increasing, inference efficiency significantly decreases, which contradicts the demands of practical applications. To address this issue, we propose \textbf{R}etrieval-\textbf{A}ugmented MLLMs with Compressed Contexts (RACC). RACC learns to compress and aggregate retrieved knowledge for a given image-question pair, generating a compact modulation in the form of Key-Value (KV) cache to adapt the downstream frozen MLLM, thereby achieving effective and efficient inference. RACC achieves a state-of-the-art (SOTA) performance of 63.92\% on OK-VQA. Moreover, it significantly reduces inference latency by 22.0\%-59.7\% compared to the prominent RAVQA-v2. Abundant experiments show RACC's broad applicability. It is compatible with various off-the-shelf MLLMs and can also handle different knowledge sources including textual and multimodal documents.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。