[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
企业微信
悟空CRM  >   公司新闻  >  小红书提出 OvarNet 模型:开集预测的新SOTA,“万物识别”有了新玩法

小红书提出 OvarNet 模型:开集预测的新SOTA,“万物识别”有了新玩法

悟空软件 阅读次数:1026 次浏览

在 CVPR2023 上,小红书社区技术团队提出了一项全新任务 —— Open-vocabulary Object Attribute Recognition(面向开放词汇的目标检测与属性识别)。该任务旨在采用单一模型对图像中任何类别目标同时进行定位、分类和属性预测,从而推动目标识别能力向更高的深度和广度发展。本文提出了 OvarNet 这一全新模型,OvarNet 基于海量多模态数据预训练的视觉语言模型开发,利用可获得的目标检测和属性识别数据进行 Prompt Learning(提示式学习)。同时,为了进一步获得高度可用的 Zero-shot 识别能力,我们从大规模图文数据中提取细粒度的类别和属性表示,进行弱监督学习。此外,我们还采用知识蒸馏的方式简化模型计算复杂度,显著提升 OvarNet 的实用性和时效性。实验结果表明,OvarNet 在开集目标检测和属性识别任务上取得了新 SOTA,证明了识别语义类别和属性对于视觉场景理解的互补性。
1.背景

理解视觉场景中的目标一直以来都是计算机视觉发展的主要驱动力,例如,目标检测任务的目的是在图像中定位出目标并为其分配预定义的语义标签,如“汽车”、“人”或“杯子”。尽管目标检测算法取得了巨大的成功,但是这种任务定义在很大程度上简化了我们对视觉世界的理解,因为一个视觉目标除了语义类别外,还可以从许多其他方面进行描述,例如,一辆公共汽车可以是“黄色”或“黑色”,一件衬衫可以是“条纹”或“无图案”。因此,学习属性可以补充类别级别的识别,从而获得更全面更细粒度的视觉感知。

已有大量的研究表明,理解目标的属性可以极大地促进目标识别和检测,甚至是对在训练集中极少出现或根本没有的视觉目标样例。然而,这些开创性工作中所考虑的问题和今天的标准相去甚远,例如,属性分类通常在以目标为中心的图像上进行训练和评估,而且属性的类别是固定的,有时甚至将目标的类别作为已知的先验。

在论文中,我们考虑在开放词汇场景下同时检测目标并分类属性的任务,即模型仅在一组可见的目标类别和属性上进行训练,但在测试时需要推广到未见过的目标类别和属性,如下图所示。完成这些任务我们观察到三个主要挑战:(1) 首先,在现有的视觉语言模型中,例如 CLIP 和 ALIGN,从图文对中学习得到的表示往往偏向于目标类别,而不是属性,这导致在直接用于属性识别任务时存在特征不对齐的问题;(2) 其次,没有理想的训练数据集同时包含三种类型(目标框、类别和属性)的标注。据我们所知,只有 COCO Attributes 数据集提供这样程度的标注,但它的词汇量相对有限 ( 196 个属性,29 个类别);(3) 第三,在统一框架下训练这三项任务仍未被探索,具有一定到挑战性,即在开放词汇场景下同时定位(“在哪里”)和描述目标的语义类别和属性(“是什么”)。

为了解决上述问题,我们从一个简单的架构开始,称为 CLIP-Attr。该架构首先通过离线 RPN 从图像中提取目标候选框,然后通过比较属性词嵌入和候选目标的视觉嵌入之间的相似度来实现开放式词汇目标属性识别。为了更好地对齐属性词和候选目标之间的特征,我们在文本编码器端引入可学习的 prompt 向量,并在大量的图文对数据上微调原始 CLIP 模型。为了进一步提高模型效率,我们提出了 OvarNet,这是一个统一的框架,可以在一次前向传播中进行检测和属性识别。OvarNet 是通过在目标检测和属性预测两个数据集上训练,并通过蒸馏 CLIP-Attr 模型的知识来提高在 novel/unseen 属性的性能。我们提出的 OvarNet 是第一个可扩展的管道,在开放词汇场景下可以同时定位目标,并同时推断视觉属性和目标类别。实验结果表明,尽管只使用弱监督的图文对数据对 OvarNet 进行蒸馏,但它在 VAW、MSCOCO、LSA 和 OVAD 数据集上的表现优于先前的最新技术,展现了对新属性和类别的强大泛化能力。

2.方法

我们的方法主要分为 3 步:首先,利用目标检测数据和属性预测数据在开放词汇场景下训练一个简单的两阶段的方法;然后,通过利用大量的图文对数据进一步微调这个两阶段模型以提高在 novel/unseen 类别和属性上的性能;最后,为了保证泛化性和前传速度,我们基于知识蒸馏的范式设计了一个一阶段的算法。

2.1 两阶段算法

作为社交图谱数据,数据的一致性至关重要。我们需要严格保证数据的最终一致性以及一定场景下的强一致性。为此,我们采取了以下措施:

作为社交图谱数据,数据的一致性至关重要。我们需要严格保证数据的最终一致性以及一定场景下的强一致性。为此,我们采取了以下措施:

2.2 图文对数据微调

首先,我们对图文对数据进行预处理,从句子中解析出类别词、属性词、和名词短语;随后,我们使用预训练的 CLIP-Attr 对数据进行伪标签标注;最后为了避免噪声标签的影响,我们通过多实例 NCE 损失进行监督训练。

下图给出了利用海量图文对数据微调 CLIP-Attr 的网络结构图。

2.3 单阶段蒸馏算法

尽管通过预先计算的提议框和上述的 CLIP-Attr 模型可以实现开放词汇目标属性预测,但其推理过程非常耗时,因为每一个裁剪后的区域都会输入到视觉编码器中。在这一部分,我们的目标是解决缓慢的推理速度,端到端地训练一个用于目标检测和属性预测的 Faster-RCNN 类型的模型,称为 OvarNet。图像会被依次经过视觉编码器、类别无关的区域提议网络和开放词汇属性分类网络。

下图是其整体网络框架。为了提高模型处理未知类别和属性的能力,我们进一步加入了 CLIP-Attr 模型的额外知识进行蒸馏。

3.实验

我们考虑的开放词汇目标属性识别包括两个子任务:开放词汇目标检测和对所有检测到的目标进行属性分类。所采用的数据集有 MS-COCO、VAW、LSA 和 OVAD。为了评估模型的性能,在指标评价上我们同时考虑 box-given 和 box-free 的设定,使用 mAP 进行评价。

3.1 COCO 和 VAW 上的结果
3.2 OVAD 和 LSA 上的结果
3.3 一些可视化结果
4.结语

本文探讨了开放词汇目标检测和属性识别问题,即同时定位目标并推断它们的语义类别和视觉属性。我们从一个简单的两阶段框架( CLIP-Attr )开始,使用预训练的 CLIP 对目标提议进行分类。为了更好地对齐以目标为中心的视觉特征与属性概念,我们在文本编码器侧使用可学习提示向量。在训练方面,我们采用联合训练策略来利用目标检测和属性预测数据集,并探索一种弱监督训练方案,利用外部图像-文本对增加识别新属性的鲁棒性。最后,为了提高计算效率,我们将 CLIP-Attr 的知识蒸馏到一种 Faster-RCNN 类型的模型中(称为 OvarNet)。在 VAW、MS-COCO、LSA 和 OVAD 等四个不同的基准评估中,我们表明联合训练目标检测和属性预测有助于视觉场景理解,并大大优于现有方法独立处理两个任务的方法,展示了强大的泛化能力,可以处理新的属性和类别。

对内容的充分理解,是实现内容推荐和检索的有效工具。在小红书双列布局场景下,对笔记封面图篇的理解就显得格外重要。采用本文所示的开集目标属性识别能力,我们可以在更细粒度上对封面图进行结构化理解。比如除了可以检测到人物之外,还可以进一步标记人物衣着、姿态等属性,从而在进行推荐和检索时,实现更加智能和个性化的内容推荐。此外,在构建小红书的内容质量体系时,通过对图片的充分解析,我们能够更精确地描述一篇笔记的质量分级,进而实现内容基于质量的差异化分发。

  • 论文地址:arxiv.org/abs/2301.0950
5.作者介绍

陈科研

现博士就读于北京航空航天大学,曾在 Proceedings of the IEEE、TGRS、CVPR 等发表多篇论文。主要研究方向为遥感图像处理,多模态学习。

莱戈

小红书生态算法团队负责人。曾在 CVPR、ICCV、ECCV、TNNLS、TGRS 等会刊发表 10 余篇论文,曾获 YouTube-VOS 视频分割竞赛 Top-3 排名,曾获天池视觉分割大奖赛复赛第 1 名。主要研究方向:目标跟踪、视频分割、多模态分类/检索等。

汤神

小红书智能创作团队负责人。曾在 CVPR、ECCV、ICCV、TIFS、ACMMM 等会议与期刊发表近 20 篇论文。多次刷新 WiderFace 和 FDDB 国际榜单世界记录,ICCV Wider Challenge 人脸检测国际竞赛冠军,ICCV VOT 单目标跟踪冠军,CVPR UG2+亚军。

悟空CRM产品更多介绍:www.5kcrm.com

相关内容

CRM的价格是多少?咋定价的?

嗨,大家好!今天咱们来聊聊一个大家都关心的话题——CRM系统的定价问题。你知道吗,在选择适合自己的客户关系管理(CRM)系统时,价格往往是决定性因素之一。一、市场上的CRM产品大概价格首先得说一句:市场上CRM产品的比较多!通常分为开源CRM、SAAS云服务、私有化部署。开源CRM好的品牌比较少,通常时商用后开始收费。SAAS云服务通常几百元每人每年,而私有化部署的价格通常根据其需要,几万或者几十
相关文章

CRM系统如何轻松记录客户的过去与现在?

嘿!大家好呀,今天咱们来聊聊一个超厉害的东西——CRM系统。这玩意儿可真不是普通的工具哦,它就像你口袋里的私人助手一样贴心呢。不管是企业还是个人创业者,在和顾客打交道的过程中都离不开这个得力帮手。首先吧,什么是CRM呢?简单来说就是用来管理所有跟客户有关的信息、互动和服务的一种软件系统啦。通过使用CRM系统,你可以把每一位客户的历史记录、当前情况以及未来期望统统整理得井井有条,这样不仅提高了工作效
相关文章

企业营销中的视频推广工作原理与CRM客户管理

企业营销中的视频推广工作原理与CRM客户管理一、企业为啥要拍视频?目前很多客户已经从传统的搜索引擎流向了抖音视频等平台,进行视频营销推广对于企业的产品销售与营销具有重要的作用。通过视频,以传播CRM、销售知识等为介质,目的找到潜在客户,让客户去使用咱们的产品,然后购买咱们的产品。那么要求看到视频的潜在客户,可以通过看视频后,然后联系到咱们(使用CRM管理通过微信、私信、抖音下评论、电话来的客户线索
相关文章

什么是CRM软件?CRM软件的基本概念、功能、优点、优势、核心价值

什么是CRM软件?嘿,你知道吗?在这个信息爆炸的时代,企业想要在众多竞争对手中脱颖而出真的不容易。今天我想跟你聊聊一个特别的工具——CRM软件,它就像是企业的私人管家,专门负责打理与客户之间的关系。CRM是Customer Relationship Management(客户关系管理)的缩写,听起来可能有点复杂,但其实它的概念很简单。想象一下,你每天要面对那么多顾客,光是记住他们的名字就已经够难了
相关文章

CRM选型,如何选到合适的CRM系统

嘿!大家好,在这个数字时代,与客户的联系变得比以往任何时候都要重要。一个有效的CRM系统可以让我们更好地了解我们的客户,提高工作效率,并最终增加销售量。但是,面对市面上琳琅满目的CRM系统,你是不是感到有些迷茫呢?别担心,接下来我将为大家提供一些指导性建议,帮助各位挑选出最适合自己的那一款。一、明确自身需求在开始之前,请先认真思考一下自己真正需要什么功能,比如是否重视销售预测或客户服务?是更关注营
相关文章

CRM管理系统软件怎样优化企业的日常运营?

CRM管理系统软件怎样优化企业的日常运营?一、提升客户关系管理效率每个企业都希望与自己的顾客建立并保持长期稳定的关系。为了实现这个目标,我们需要了解每个客户的喜好和需求。然而,在没有合适的工具帮助的情况下,这项工作可能会变得繁琐且耗时。这时就需要一个强大的CRM管理系统软件来帮我们记录所有关于客户的详细信息,并根据这些数据进行分析,以便更好地理解客户需求,制定更有针对性的营销策略。二、加强团队合作
相关文章

CRM厂商众多,企业如何判断系统是否合适?

CRM厂商众多,企业如何判断系统是否合适?【序言】在数字化转型的大潮中,客户关系管理系统(Customer Relationship Management System,简称CRM)作为企业管理客户信息、提升销售效率的重要工具,受到了越来越多企业的青睐。然而面对市场上琳琅满目的CRM产品,很多企业在选择时往往会陷入迷茫:什么样的CRM才适合自己呢?接下来我们将探讨一些关键点帮助您找到答案。【明确自
相关文章

CRM客户关系管理系统都有哪些功能?这些功能咋用比较好?

《CRM客户关系管理系统都有哪些功能?这些功能咋用比较好?》嘿!大家好,在这个商业竞争日益激烈的时代里,我们都知道“客户就是上帝”。而为了更好地服务我们的“上帝”,一款好的CRM(Customer Relationship Management)客户关系管理系统可是帮了大忙啦!今天呢,我们就来聊聊它有哪些实用的功能以及如何更好地利用它们吧!一、了解你的客户CRM系统就像一个超级侦探,能帮你收集并整
相关文章

CRM客户关系管理软件有什么用?

CRM客户关系管理软件有什么用?嗨!大家好!今天我想和你们聊聊一个对于企业和组织来说非常重要的工具——CRM(Customer Relationship Management)客户关系管理软件。你可能会想,“这个听起来好像挺专业的啊!”没错,但它其实就在我们身边扮演着至关重要的角色。1. 帮助企业更好地理解客户需求首先,CRM系统就像是企业的“超级大脑”。它能够收集来自不同渠道的信息,比如网站访问
相关文章

CRM管理系统如何做好企业用户的精细化运营?

CRM管理系统如何做好企业用户的精细化运营?一、以人为本:构建360度客户画像“嘿,你有没有想过,为什么有的公司能和顾客建立深厚的情感纽带呢?” 要想让客户成为企业的铁杆粉丝,就需要从了解他们开始。运用CRM系统进行客户数据收集与分析是基础工作。在CRM的帮助下,我们可以全方位地描绘出每一位客户的喜好、行为模式甚至潜在需求,从而实现个性化服务。只有深入了解了客户的真实需要,才能制定出有效的营销策略
相关文章
最新文章

CRM的价格是多少?咋定价的?

嗨,大家好!今天咱们来聊聊一个大家都关心的话题——CRM系统的定价问题。你知道吗,在选择适合自己的客户关系管理(CRM)系统时,价格往往是决定性因素之一。一、市场上的CRM产品大概价格首先得说一句:市场上CRM产品的比较多!通常分为开源CRM、SAAS云服务、私有化部署。开源CRM好的品牌比较少,通常时商用后开始收费。SAAS云服务通常几百元每人每年,而私有化部署的价格通常根据其需要,几万或者几十
最新文章

CRM系统如何轻松记录客户的过去与现在?

嘿!大家好呀,今天咱们来聊聊一个超厉害的东西——CRM系统。这玩意儿可真不是普通的工具哦,它就像你口袋里的私人助手一样贴心呢。不管是企业还是个人创业者,在和顾客打交道的过程中都离不开这个得力帮手。首先吧,什么是CRM呢?简单来说就是用来管理所有跟客户有关的信息、互动和服务的一种软件系统啦。通过使用CRM系统,你可以把每一位客户的历史记录、当前情况以及未来期望统统整理得井井有条,这样不仅提高了工作效
最新文章

企业营销中的视频推广工作原理与CRM客户管理

企业营销中的视频推广工作原理与CRM客户管理一、企业为啥要拍视频?目前很多客户已经从传统的搜索引擎流向了抖音视频等平台,进行视频营销推广对于企业的产品销售与营销具有重要的作用。通过视频,以传播CRM、销售知识等为介质,目的找到潜在客户,让客户去使用咱们的产品,然后购买咱们的产品。那么要求看到视频的潜在客户,可以通过看视频后,然后联系到咱们(使用CRM管理通过微信、私信、抖音下评论、电话来的客户线索
最新文章

快速注册,立即使用

仅需一分钟,立即注册悟空云。悟空CRM免费试用。
客户端下载