arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2406.05821cs.CV

F-LMM:为冻结的大型多模态模型赋予视觉定位能力

F-LMM: Grounding Frozen Large Multimodal Models

Nanyang Technological University(南洋理工大学);SenseTime Research(商汤科技研究院);Shanghai AI Laboratory(上海人工智能实验室);The Chinese University of Hong Kong(香港中文大学)

Size Wu, Sheng Jin, Wenwei Zhang, Lumin Xu, Wentao Liu, Wei Li, Chen Change Loy

AI总结:

F-LMM通过冻结现成LMM、仅训练少量CNN层将词-像素注意力权重转化为掩码,在不损害对话能力的前提下实现视觉定位,并在多个基准上取得有竞争力的性能。

摘要:

赋予大型多模态模型(LMMs)视觉定位能力可以显著增强AI对视觉世界的理解及其与人类的交互。然而,现有方法通常微调LMMs的参数以学习额外的分割标记,并在定位和分割数据集上过拟合。这种设计不可避免地会导致通用AI助手不可或缺的对话能力发生灾难性下降。在本文中,我们在多个多模态问答基准上全面评估了最先进的定位LMMs,观察到性能急剧下降,表明通用知识理解和指令遵循能力减弱。为解决这一问题,我们提出了F-LMM——在人类-AI对话中为冻结的现成LMMs赋予定位能力——这是一种简单而有效的设计,其依据是:在训练良好的LMMs的注意力机制中,天然存在有利于视觉定位的词-像素对应关系。仅使用少量可训练的CNN层,我们就能将词-像素注意力权重转换为掩码logits,再由基于SAM的掩码细化器进一步优化。我们的F-LMM既不学习特殊的分割标记,也不使用高质量的定位指令微调数据,但在指代表达分割和全景叙事定位基准上取得了有竞争力的性能,同时完全保留了LMMs原有的对话能力。此外,由于保留了指令遵循能力并获得了定位能力,F-LMM可以直接应用于复杂任务,如推理分割、定位对话生成和视觉思维链推理。我们的代码可在https://github.com/wusize/F-LMM找到。

英文摘要:

Endowing Large Multimodal Models (LMMs) with visual grounding capability can significantly enhance AIs' understanding of the visual world and their interaction with humans. However, existing methods typically fine-tune the parameters of LMMs to learn additional segmentation tokens and overfit grounding and segmentation datasets. Such a design would inevitably cause a catastrophic diminution in the indispensable conversational capability of general AI assistants. In this paper, we comprehensively evaluate state-of-the-art grounding LMMs across a suite of multimodal question-answering benchmarks, observing drastic performance drops that indicate vanishing general knowledge comprehension and weakened instruction following ability. To address this issue, we present F-LMM -- grounding frozen off-the-shelf LMMs in human-AI conversations -- a straightforward yet effective design based on the fact that word-pixel correspondences conducive to visual grounding inherently exist in the attention mechanism of well-trained LMMs. Using only a few trainable CNN layers, we can translate word-pixel attention weights to mask logits, which a SAM-based mask refiner can further optimise. Our F-LMM neither learns special segmentation tokens nor utilises high-quality grounded instruction-tuning data, but achieves competitive performance on referring expression segmentation and panoptic narrative grounding benchmarks while completely preserving LMMs' original conversational ability. Additionally, with instruction-following ability preserved and grounding ability obtained, F-LMM can be directly applied to complex tasks like reasoning segmentation, grounded conversation generation and visual chain-of-thought reasoning. Our code can be found at https://github.com/wusize/F-LMM.

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

↑