港中文贾佳亚团队提出 LISA 大模型:解锁多模态大模型“推理分割”能力

访客1年前AI生活162

当前的视觉识别系统都依赖人类用户明确指代目标物体或预先设定识别类别,进而进行识别。它们仍然只能处理简单明确的指令(如“橙子”),而无法解析相对隐式和复杂的指令(如在下图中指出 “维生素C含量高的食物”)。

港中文贾佳亚团队提出 LISA 大模型:解锁多模态大模型“推理分割”能力

而真正的智能感知系统应该根据用户指令推理其真实意图。例如,在指示机器人时,人们往往倾向于直接给一个指令“我想要看电视“,而不是分几个步骤”走去茶几旁边,帮我找到遥控器,然后按下按钮打开电视“。这些场景都要求感知系统具有复杂推理和联系世界知识的能力。

最近,香港中文大学贾佳亚团队发布一项新研究,提出一项新任务——推理分割(Reasoning Segmentation),该任务要求模型能够处理复杂的自然语言指令,并给出精细的分割结果。

港中文贾佳亚团队提出 LISA 大模型:解锁多模态大模型“推理分割”能力

如上图所示,推理分割任务具有很大的挑战性,可能需要借鉴世界知识(例如,左图需要了解“短镜头更适合拍摄近物体”),或进行复杂图文推理(如右图需要分析图像和文本语义,才能理解图中“栅栏保护婴儿”的含义),才能获得最终理想的分割结果。

尽管当前多模态大模型(例如Flamingo [1], BLIP-2 [2], LLaVA [3], miniGPT-4 [4], Otter [5])使得AI能够根据图像内容推理用户的复杂问题,并给出相应的文本分析和回答,但仍无法像视觉感知系统那样在图像上精确定位指令对应的目标区域。

因此,此项研究工作提出LISA(Large Language Instructed Segmentation Assistant)多模态大模型。LISA通过引入一个标记来扩展初始大型模型的词汇表,并采用Embedding-as-Mask的方式赋予现有多模态大型模型分割功能,最终展现出强大的零样本泛化能力。

同时,该工作还创建了ReasonSeg数据集,其中包含上千张高质量图像及相应的推理指令和分割标注。

实验证明,在训练过程中仅使用不包含复杂推理的分割数据(通过将现有的语义分割数据如ADE20K [6],COCO-Stuff [7]以及现有指代分割数据refCOCO系列 [8]中的每条数据转换成“图像-指令-分割Mask”三元组) ,LISA能在推理分割任务上展现出优异的零样本泛化能力。此外,进一步使用239个推理分割数据进行微调训练还能显著提升LISA在推理分割任务上的性能。而且LISA还表现出高效的训练特性,只需在8张具有24GB显存的3090显卡上进行10,000次训练迭代,即可完成7B模型的训练。

技术方案概述

港中文贾佳亚团队提出 LISA 大模型:解锁多模态大模型“推理分割”能力

首先将图像和文本送到多模态-大语言模型(在实验中即LLaVA),得到输出的文本结果,如果此时文本结果包含标记,则表示需要通过输出分割预测来解决当前问题。反之,若不包含标记,则无分割结果输出。

如果存在标记,则将标记在多模态大模型最后一层对应的embedding经过一个MLP层得到,并将其与分割视觉特征一起传递给解码器(其中分割视觉特征由输入编码器对图像进行编码得到)。最终,根据生成最终的分割结果。

LISA在训练过程中使用了自回归交叉熵损失函数,以及对分割结果监督的BCE和DICE损失函数。

模型效果

最终,LISA不仅在传统的语言-图像分割指标(refCOCO、refCOCO+和refCOCOg)上展现出优异性能,还能处理以下分割任务情景:1)复杂推理,2)联系世界知识,3)解释分割结果以及4)多轮对话。

在有复杂情景的ReasonSeg数据集上,LISA显著领先于其他相关工作(如Table 1),进一步证明其出色的推理分割能力。

港中文贾佳亚团队提出 LISA 大模型:解锁多模态大模型“推理分割”能力

港中文贾佳亚团队提出 LISA 大模型:解锁多模态大模型“推理分割”能力

效果展示

港中文贾佳亚团队提出 LISA 大模型:解锁多模态大模型“推理分割”能力港中文贾佳亚团队提出 LISA 大模型:解锁多模态大模型“推理分割”能力港中文贾佳亚团队提出 LISA 大模型:解锁多模态大模型“推理分割”能力

港中文贾佳亚团队提出 LISA 大模型:解锁多模态大模型“推理分割”能力

本站部分文章来自互联网,文章版权归原作者所有。如有疑问请联系QQ:362580117!

相关文章

苹果造车天团

苹果造车天团

小编消息,7月16日,据外媒Business Insider报道,苹果技术副总裁Kevin Lynch将加入苹果造车战队。 面对互联网造车热...

福布斯2023全球区块链50强发布 亚洲公司占比26%

福布斯2023全球区块链50强发布 亚洲公司占比26%

2月8日,福布斯发布2023年全球区块链50强榜单,蚂蚁、百度、建行等6家中国公司系数上榜,其中也不乏Alphabet、Meta、NBA、甲骨...

国家发行的数字货币和比特币的区别与不同!

国家发行的数字货币和比特币的区别与不同!

“数字货币”这个词或许挺陌生,但近几年被炒得挺热的“比特币”,应该有不少人听说过。哦,原来是它呀!不过,是它,也不是它。 同样...

IEEE Fellow黄建伟:「网络经济、群体智能」跨界融合,智慧城市的新风暴

IEEE Fellow黄建伟:「网络经济、群体智能」跨界融合,智慧城市的新风暴

自然界中,「蚂蚁觅食」是一种寻常但奇特的现象。 成群结队的蚂蚁,总是能够在食物与蚁巢之间寻找到最佳路径,快速地将食物搬运至蚁巢中。这引起了生物...

CES 2023 LG推出新款可变色MoodUp冰箱,时刻焕新生活尽享美好时光

CES 2023 LG推出新款可变色MoodUp冰箱,时刻焕新生活尽享美好时光

       近日,LG电子在2023年CES国际消费电子产品展览会上,推出了一系列富有黑科技的创新产品。其中LG MoodUP冰箱让在场观...

平台战略,生态模式,15 万开发者打造 AI 领域的 App Store!

平台战略,生态模式,15 万开发者打造 AI 领域的 App Store!

如今AI圈的角逐,高手如云,战事胶着。 其中一家叫极视角的企业,异军突起,出其不意地进入了大家的视线。 在一众企业鏖战A...

发表评论    

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。