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

访客10个月前AI生活76

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

相关文章

巴西在G20会议上提升数字包容性和技术标准

巴西在G20会议上提升数字包容性和技术标准

该倡议的目标是弥合数字鸿沟,在全国范围内促进包容性数字采用。 巴西在马塞奥举行的G20数字经济会议上加强了对数字包容和互联互通的...

波场TRON亮相币安区块链周 获彭博社、美联社等外媒高度关注

波场TRON亮相币安区块链周 获彭博社、美联社等外媒高度关注

近日,波场TRON作为钻石赞助商亮相伊斯坦布尔币安区块链周,引发全球知名媒体高度关注。国际权威财经媒体金融时报、彭博社、法国第一大报费加罗报...

开源框架TVM支持各种类型AI芯片,还能让芯片有2-10倍的提升!

开源框架TVM支持各种类型AI芯片,还能让芯片有2-10倍的提升!

新推出的AI芯片因架构的独特性和软件的易用性增加了客户尝试和迁移的成本,因此,软件成为了能否快速、低成本迁移的关键。现在普遍的做法是在Ten...

Grayscale与SEC合作推出比特币现货ETF!

Grayscale与SEC合作推出比特币现货ETF!

Grayscale重新推动了长达数月的努力,以说服美国证券交易委员会批准将这家加密投资公司的 400 亿美元比特币信托基金转换为交易...

美国政府和苹果公司联合开发用于对抗新病毒的网站!

美国政府和苹果公司联合开发用于对抗新病毒的网站!

3月28日消息,据国外媒体报道,当地时间周五苹果公司表示,其正与特朗普政府和联邦机构合作发布一款应用程序和一个网站,引导用户从美国疾...

抖音电商开放日:过去一年投入超10亿元保障用户体验

抖音电商开放日:过去一年投入超10亿元保障用户体验

5月13日,抖音电商在北京举办“用户体验开放日”,邀请抖音电商业务相关负责人及媒体到场,分享过去一年平台在售后体验、客服体验以及平台治理等三方...

发表评论    

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