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

访客2年前AI生活214

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

相关文章

人民日报说得很对!国内芯片实现14nm量产后,各国一反常态!

人民日报说得很对!国内芯片实现14nm量产后,各国一反常态!

中国芯片市场空间巨大,吸引了包括美国、荷兰、日本和韩国在内的半导体巨头纷纷进入。这些国家的企业在中国建立研发中心和生产基地,与中国展开深入合...

正齐半导体年产6万颗高阶功率模块研发生产项目落地萧山经济技术开发区

正齐半导体年产6万颗高阶功率模块研发生产项目落地萧山经济技术开发区

距离投资百亿元的合盛新能源材料和器件模组制造项目开工仅3个多月,近日,又一重量级半导体项目——正齐半导体年产6万颗高阶功率模块研发生产项目落...

工信部官宣:加快5G部署,推广5G+VR/AR、赛事直播等消费!

工信部官宣:加快5G部署,推广5G+VR/AR、赛事直播等消费!

3月24日,南都记者从工信部官网获悉,工信部发布《工业和信息化部关于推动5G加快发展的通知》(以下简称《通知》)。《通知》内称要加快...

网传中芯国际12英寸圆晶软件项目暂停,技术承包方否认

8月8日消息,据新浪科技报道,中芯国际与上扬软件合作,计划在北京新建的12英寸晶圆厂CIM国产化项目于近期宣布暂停。该报道称有知情人士透露,之...

腾讯云PaaS份额稳居市场第二

国际权威行业研究机构IDC最新发布的《中国公有云市场份额报告,2023H1》显示,2023上半年中国公有云服务市场整体规模(IaaS/PaaS...

以太经典/ETC

项目信息 项目名称 Ethereum Classic 项目市值(亿) $ 43.00 初始估值(百万) $ 71.14 初始交易...

发表评论    

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