Token导航 LogoToken导航

RGB-D语义分割受限于NYUv2旧数据集 新数据集能否突破

更新时间 2026-10-20来源 科技行者正文 4905字阅读约 16分钟

你可能没想到过这样一件事:一个让机器人"看懂"室内环境的技术领域,用的还是十几年前的老数据。

2012年发布的NYUv2数据集,只有1449张图片,40个类别。这个数据集撑起了整整十几年的RGB-D语义分割研究。想象一下,如果一个孩子学认物体,翻来覆去只看过1449张照片,还只认识40种东西,那他长大以后见到一个没见过的"晾衣架"或者"落地灯",大概率是认不出来的。这不是危言耸听,这就是目前主流RGB-D分割模型面对的真实处境。

RGB-D语义分割

:一种计算机视觉任务,模型同时接收彩色图像(RGB)和深度图像(记录每个像素离摄像头多远),输出图像中每个像素属于什么物体类别,比如"这里是椅子""那里是墙"。加了深度信息之后,模型能更好地分辨前后遮挡关系,理论上应该比纯RGB分割更准。

问题是,理论归理论,数据跟不上,模型再厉害也白搭。这就是这篇论文要讲的故事:一群来自北德克萨斯大学的研究者,做了一件看起来"笨重"但其实极其关键的事,重新造了一个更大、更全、更干净的RGB-D数据集,叫RGBD20K,同时顺手提出了一个新的融合算法,刷新了各大榜单的成绩。

核心问题:小数据集卡住了整个领域的脖子

先说数据规模。NYUv1有2347张图,NYUv2缩水到1449张,后来的SUN RGB-D把规模拉到10335张,已经算是当时的"大手笔"了。但放到今天,这个体量在深度学习面前几乎不够看。

视觉Transformer

:一类基于注意力机制处理图像的神经网络架构,相比传统卷积网络,通常需要更多的训练数据才能发挥出应有的性能,数据量不够时反而容易表现不佳。

现在主流的分割模型基本都在用Transformer架构,而Transformer这类模型出了名地"吃数据"。用一万张图训练一个动辄几亿参数的模型,就像想用一小碗饭喂饱一个正在长身体的运动员,理论上能撑一阵,但很难真正发挥出模型的潜力。

再说语义覆盖面。NYUv1只有13个类别,NYUv2是40个,SUN RGB-D是37个。这意味着什么?意味着模型一辈子只学过"墙、地板、床、桌子"这几十种东西,真实世界里办公室、教室、餐厅、图书馆、健身房里成百上千种物品,它压根没见过。这就好比一个只在城市里长大的人突然被扔进热带雨林,认得出柏油马路认不出藤蔓植物,不是因为他笨,是因为他的"训练数据"里从来没出现过这些东西。

最后一个问题,也是最容易被忽视但影响最大的:标注质量。论文里提到一个很扎心的实验,他们把SUN RGB-D测试集里的标注重新精细校对了一遍,什么都没改,只是把原来标错、标漏、边界画得毛糙的地方修干净了,结果同一个模型(DPLNet)在原始测试集上是52.8的mIoU,换成精修后的测试集直接跳到55.0。

mIoU

:全称mean Intersection over Union,衡量图像分割准确度的核心指标,数值越高说明模型预测的物体轮廓和真实轮廓重合度越高,是这个领域最常用的评分标准。

这2.2个百分点的差距是从哪来的?不是模型突然变聪明了,模型一步没动,是"考卷"本身之前出错了。这说明过去十几年里,很多论文比拼的分数里,一部分差异可能根本不是算法优劣带来的,而是标注噪声造成的运气成分。这个发现挺让人后背发凉的,等于说这个领域长期以来的评估基准本身就不太靠谱。

RGBD20K:把数据集这件事做扎实

面对这些问题,研究团队没有从零采集20000张新照片,而是走了一条更聪明的路:把散落在各处、原本服务于不同任务(比如物体追踪、显著性检测、预训练)的RGB-D图像资源整合起来,统一用一套160类的分类标准重新标注。

具体来说,他们从SUN RGB-D拿了10335张,从RGB-D镜面数据集拿了824张,还从视频显著目标检测、深度跟踪、ARKitTrack等好几个原本跟语义分割毫不相干的数据集里各取了一部分,又从DIML RGB-D数据集拉来8365张原本用于预训练的图片,凑够了20000张。

这个思路有点像做菜时不是非要从种地开始,而是把菜市场里不同摊位买来的食材,用统一的刀工和调味标准重新处理一遍,做出一桌风格一致的菜。如果不这么整合,而是从头拍20000张照片,成本和周期都会高到不现实,团队规模也支撑不起这种工程量。整合已有资源加统一精细标注,是在有限资源下把数据规模和质量都拉上去的一条务实路径。

整合完之后是硬骨头:重新标注。团队采用了一套分层标注体系,从粗到细,比如先标出"家具"这个大类,再往下细分成"椅子的具体类型""桌子的具体类型"。同时引入了深度感知的遮挡处理机制,简单说就是利用深度图信息判断哪个物体在前哪个在后,避免出现"桌子标注盖住了桌子后面的椅子轮廓"这种常见错误。

最终,RGBD20K覆盖160个细粒度类别,是NYUv2的4倍,是SUN RGB-D的4倍多;场景类型达到75种,包括办公室、教室、卧室、客厅、厨房、礼堂、音乐室、健身房、实验室、图书馆、餐厅、电脑室、零售店、饭店、教堂、仓库等等,几乎覆盖了常见室内环境的方方面面。整体规模20000对图像,训练测试按18000比2000划分,标注质量经过多轮人工核校,总标注实例数达到368312个,平均每张图18.4个标注对象。

这里有一组数字值得拿出来对比看看:NYUv1只有7种场景,NYUv2有26种,SUN RGB-D有47种,RGBD20K做到了75种。场景越多样,意味着模型见过的"世界"越丰富,泛化能力才有可能真正建立起来。

分数净化融合模型:先"净化"再"关注"

光有数据不够,论文同时提出了一个新的融合算法,叫SPF,全称score-purified fusion(分数净化融合)。

跨模态注意力融合

:一种让RGB图像特征和深度图像特征互相"交流参考"的机制,通常借助注意力机制,让模型在处理RGB信息时也能参考对应的深度信息,反之亦然,从而提升识别准确度。

这类方法过去存在一个普遍问题,论文里称之为"注意力稀释"。简单说,深度传感器本身噪声就不小,边界经常对不齐RGB图像,如果直接把这种带噪声的深度特征和RGB特征放在一起做注意力计算,注意力机制既要处理噪声干扰,又要提取有用的上下文信息,两件事情搅在一起做,效果自然打折扣。

这就好比开会讨论问题时,如果发言人一边要澄清谣言一边要陈述正事,听众的注意力势必被分散,最后正事没讲透,谣言也没完全澄清。SPF的做法是先把"谣言"(噪声、不可靠的跨模态信息)过滤掉,再让注意力机制专心处理"正事"(真正有价值的上下文信息)。

具体的实现方式是这样的:模型会给RGB特征和深度特征分别打一个"可信度分数",这个分数不是随便拍脑袋定的,而是通过一个叫"关系仲裁器"的小型神经网络模块,同时参考自身模态信息和对方模态信息计算出来的。有了这个分数之后,模型在构建用于注意力计算的Key和Value(可以理解为"检索用的索引"和"实际提取的内容")时,会按照可信度分数对RGB特征和深度特征做加权融合,可信度低的成分自然被压低权重。

Key和Value

:Transformer注意力机制里的两个关键概念,Key相当于给每份信息贴的"标签",用于判断相关性;Value是信息本身的内容。模型通过比对Query(查询)和Key的相似度,决定从各个Value里提取多少内容。

这套设计还加入了一个细节:学习出来的噪声参数。每一层网络会单独学习一组噪声偏置,加到Query特征上再参与融合计算。消融实验显示,这种"可加性学习噪声"比简单相乘的方式效果更好,比随机高斯噪声效果好出整整1个百分点(46.3 vs 45.3)。这说明噪声不是越随机越好,而是要让模型自己学出"哪种程度的扰动对当前层最有帮助",这跟随手往菜里撒盐不一样,得根据这道菜本身的味道调整用量。

消融实验还测试了不同的关系仲裁器结构,2层MLP加Sigmoid激活函数的组合效果最好,达到46.3的mIoU,比用Softmax或者卷积网络替代都要好,差距在0.5到0.9个百分点之间。这类细节实验很枯燥,但正是这些看似微小的调整堆叠起来,才把一个"能用"的模型打磨成一个"好用"的模型。

三个基准,全面登顶

论文在NYUv2、SUN RGB-D和自家的RGBD20K上都做了对比实验,参照的对手包括PGDENet、TokenFusion、GeminiFusion、MultiMAE、CMX、CMNeXt、DFormer v1/v2、DPLNet等一票近几年的代表性方法。

结果如下(mIoU数值,越高越好):

在NYUv2上,SPF拿到60.5,此前最强的GeminiFusion是60.2。

在SUN RGB-D原始标注上,SPF是55.0,GeminiFusion是54.6。

在重新精修标注的SUN RGB-D+上,SPF达到62.2,明显高于GeminiFusion的61.5。

在RGBD20K这个新基准上,SPF拿到46.3,第二名GeminiFusion是45.4,DFormer v2是44.1,DPLNet反而只有28.0,说明面对160类的细粒度分割任务,不同模型架构的鲁棒性差异被显著放大了。

值得一提的是,几乎所有对比方法在精修后的SUN RGB-D+上表现都明显优于原始SUN RGB-D,这再次印证了前面提到的结论:标注质量本身对模型评估结果的影响,一点不比算法设计的影响小。

消融实验里还有一个挺有意思的发现:如果去掉RGB和深度特征之间专门的对齐投影层,性能会从46.3掉到45.7;如果完全不用分数机制而是简单相加,性能掉到45.5;如果只对Key用分数机制、Value用简单求和,掉到46.0;反过来只对Value用分数机制,掉到45.9。这说明Key和Value都需要经过"净化",缺一不可,就像炒菜时油温和火候都得控制好,只调好一个另一个乱来,菜照样炒不好。

写在后面

读完这篇论文,一个细节让我印象很深:他们没有选择重新拍摄20000张全新照片,而是从七八个原本服务于不同任务(追踪、显著性检测、预训练)的数据集里"拼凑"素材,再统一标注体系精修一遍。这在学术圈里其实是个挺务实但容易被忽视的做法,很多团队更愿意从头做一个"纯血"数据集来彰显工作量,但这篇论文选择了效率更高的整合路线,把有限的精力砸在标注质量的把控上,而不是数据采集这个相对边际效益更低的环节上。

另一个让我反复琢磨的地方是那个52.8到55.0的标注质量实验。它其实在提醒整个领域一件容易被忽略的事:很多论文比拼的0.5、1个百分点的提升,有可能根本不是算法的功劳,而是测试集本身标注的运气成分。这种反思其实挺重要的,一个领域如果长期在有噪声的基准上刷分,那些细微的排名变化可能根本没有统计意义。

这篇论文还留了一个没有明说但值得追问的问题:160个类别、75种场景,这已经是相当细的粒度了,但真实世界的物体种类远不止160种。下一步是不是应该往"开放词汇"方向走,让模型能识别训练时完全没见过的新类别?这可能是这个数据集之后自然会被问到的问题。

Q&A

Q1:RGBD20K数据集相比NYUv2和SUN RGB-D有哪些主要优势?

A:RGBD20K拥有20000张RGB-D图像对,覆盖160个细粒度类别和75种场景类型,远超NYUv2的1449张图片40个类别和SUN RGB-D的10335张图片37个类别,同时经过多轮人工核校,标注质量更高更干净。

Q2:分数净化融合模型SPF的核心创新点是什么?

A:SPF采用"先净化再关注"的设计思路,在做跨模态注意力计算之前,先根据可信度分数过滤掉RGB和深度特征中不可靠的成分,避免注意力机制同时处理噪声和有效信息,从而在NYUv2、SUN RGB-D和RGBD20K三个基准上都取得了最佳效果。

Q3:为什么说SUN RGB-D数据集的标注质量会影响模型评估结果?

A:研究团队将SUN RGB-D测试集重新精细标注后,同一个预训练模型DPLNet的mIoU从52.8提升到55.0,说明原始标注中存在的类别混淆、边界模糊等问题会直接导致模型性能被低估,标注质量对评估结果的影响不容忽视。

文章标签学术研究
资讯来源:由AI资讯编辑整理自互联网公开内容,版权归原作者所有,未经许可,不得转载。

继续浏览更多资讯

返回资讯目录

相关资讯

更多