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

访客1年前AI生活160

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

相关文章

2024 全球开发者先锋大会即将强势开启,科技狂欢盛典再燃上海

2024 全球开发者先锋大会即将强势开启,科技狂欢盛典再燃上海

备受瞩目的2024全球开发者先锋大会(2024 GDC)将于2024年3月23日至24日在上海徐汇滨江隆重举行。 全球开发者先锋大会(GDC...

自动驾驶下半场,低成本高性价比或成趋势?|2023上海车展

自动驾驶下半场,低成本高性价比或成趋势?|2023上海车展

智能驾驶随着硬件和软件的提升渐渐走入下半场,智能化的下半场玩家们关注的点也逐渐现实,以往生动的故事再也不是打动客户的依仗,可量产落地的产品才...

大中华区首家云管理服务提供商成立!平安云和GitHub达成战略合作!

大中华区首家云管理服务提供商成立!平安云和GitHub达成战略合作!

2018年10月17日,平安云发布会在深圳举行。会上正式宣布平安云与全球知名代码托管及开源社区平台GitHub达成战略合作,未来,将通过平安...

华为双11什么值得买?华为Mate60 Pro乐臻版可以无脑冲!

华为双11什么值得买?华为Mate60 Pro乐臻版可以无脑冲!

一年一度的双11购物节已经正式打响,各类平台商家都给出了各种优惠与折扣,一场属于消费者们的狂欢已经到来。此前Mate60 Pro悄然上市引爆全...

五道口大模型简史

五道口大模型简史

「预见未来最好的方式就是亲手创造未来。」 01 「范式」革命 故事的开头发生在 2018 年的秋天,北京海淀区。 那一天,10 月 11 日,...

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

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

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

发表评论    

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