用用户专属的键值缓存提升AI助手回答准确性
Jarvis: Towards Personalized AI Assistant via Personal KV-Cache Retrieval
- 将用户信息存入文本与视觉的键值缓存中
- 在问答时检索个人缓存,提升局部细节响应精度
- 适合需要精准个性化服务的场景
视觉语言模型(VLMs)的发展推动了开放域感知与推理。近期研究尝试将通用VLM转化为个性化助手,如ChatGPT已支持通过用户特定信息进行模型个性化。然而,现有方法或学习一组概念标记,或训练VLM利用用户信息,均难以生成准确回答。我们提出Jarvis,一种通过个人键值缓存(KV-Cache)实现个性化AI助手的新框架,将用户特定信息存储于文本与视觉标记的键值缓存中。文本标记通过总结用户信息生成元数据,视觉标记则从用户图像中提取独特图像块。回答问题时,Jarvis首先从个人存储中检索相关键值缓存,并用于确保回答准确性。我们还引入一个细粒度基准测试,使用相同的图像块挖掘流程,强调基于细粒度用户信息的精确问答。Jarvis在多个数据集上的视觉问答和纯文本任务中均达到当前最优表现,表明其为构建个性化AI助手提供了可行路径。代码与数据集将公开。
原文摘要 · Abstract (English)
The rapid development of Vision-language models (VLMs) enables open-ended perception and reasoning. Recent works have started to investigate how to adapt general-purpose VLMs into personalized assistants. Even commercial models such as ChatGPT now support model personalization by incorporating user-specific information. However, existing methods either learn a set of concept tokens or train a VLM to utilize user-specific information. However, both pipelines struggle to generate accurate answers as personalized assistants. We introduce Jarvis, an innovative framework for a personalized AI assistant through personal KV-Cache retrieval, which stores user-specific information in the KV-Caches of both textual and visual tokens. The textual tokens are created by summarizing user information into metadata, while the visual tokens are produced by extracting distinct image patches from the user's images. When answering a question, Jarvis first retrieves related KV-Caches from personal storage and uses them to ensure accuracy in responses. We also introduce a fine-grained benchmark built with the same distinct image patch mining pipeline, emphasizing accurate question answering based on fine-grained user-specific information. Jarvis is capable of providing more accurate responses, particularly when they depend on specific local details. Jarvis achieves state-of-the-art results in both visual question answering and text-only tasks across multiple datasets, indicating a practical path toward personalized AI assistants. The code and dataset will be released.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。