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

访客1年前AI生活94

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

相关文章

AI芯片行业领导者推出最新一代“开工开物”,用于产业智慧升级!

AI芯片行业领导者推出最新一代“开工开物”,用于产业智慧升级!

数字经济时代,人工智能将成为像水电煤一样的基础设施。但如何让更多企业具备 AI 能力,以生态协同撬动 AI 普及,是产业参与者共同面临的一大...

骁龙888带着CPU、GPU和AI性能的显著提升,和小米11一起拉开了小米新十年的序幕!

骁龙888带着CPU、GPU和AI性能的显著提升,和小米11一起拉开了小米新十年的序幕!

新SoC的发布节点,直接左右旗舰手机的发布周期。刚在12月发布的骁龙888,不但打破了历史规律,还带着CPU、GPU和AI性能的显著提升,和...

百度智能云2022成绩单:云智一体深入产业,多领域蝉联市场第一

百度智能云2022成绩单:云智一体深入产业,多领域蝉联市场第一

随着疫情走向终结,2023年伊始,工厂加快步伐复工复产、城市烟火气涌现、消费活力加速释放,企业对未来发展呈现乐观预期。有外媒称,“中国经济将实...

金山办公「规模之道」:超1500亿个文档上云,办公软件迈入云原生时代

金山办公「规模之道」:超1500亿个文档上云,办公软件迈入云原生时代

8月26日,金山办公在珠海举办以「规模之道」为主题的 2022 年技术开放日活动。 活动一开始,金山办公首先介绍三个重要的数字: 第一个数字...

梅赛德斯-奔驰与腾讯在高级别自动驾驶领域达成战略合作

梅赛德斯-奔驰与腾讯在高级别自动驾驶领域达成战略合作

2022年7月11日,梅赛德斯-奔驰集团股份公司旗下公司——戴姆勒大中华区投资有限公司与腾讯云计算(北京)有限责任公司签署合作备忘录,围绕高级...

国内首家! 阿里云全方位支持Llama 3 训练推理,并提供免费算力

国内首家! 阿里云全方位支持Llama 3 训练推理,并提供免费算力

4月18日,Meta开源最新Llama 3系列,阿里云魔搭社区第一时间上架全部4款模型。今天,阿里云百炼大模型服务平台宣布在国内首家推出针对L...

发表评论    

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