Exploring Cross-Image Pixel Contrast for Semantic Segmentation
创始人
2024-06-02 06:17:12
0

ICCV2021

先将图片经过FCN encoder(Resnet和hrnet)得到feature I,其中一条支路会经过fSEG得到Y,|C|是真实类别的数量,

是GT label,将其onehot后和softmax后的y进行交叉熵损失: 

fPROJ将每个嵌入i∈I的高维像素映射为256的l2归一化特征向量,用于计算对比损失Lnce。fPROJ通过ReLU实现为两个1×1卷积层。

下式Pi和Ni分别表示从memory bank得到的像素i的正样本和负样本的pixel embedding集合。正/负样本和锚点i并不局限于来自同一幅图像。可以通过拉近相同的类像素样本和推开不同的类样本来学习嵌入空间。λ为1,为0.1.

 

总体损失为: 

分割网络利用一元交叉熵损失和两两度量损失的优势,可以产生更多的鉴别特征。

Pixel-to-Region Contrast(memory bank的设计):

1、为了减小memory bank的大小,不会存下所有的像素,会选择为每个类别维护一个像素队列。对于每个类别,从最新的batch中的每张图像中只随机抽取少量像素V,并拉入队列,队列大小为T >>V。V = 10,T = 10N(N为图像数量)

2、region bank大小为|C|×N×D,其中D为像素嵌入的维数,N为图像数量。

Region bank中的第(, n)个元素是对第n幅图像中标为类的所有像素的嵌入进行平均池化得到的d维特征向量。有两个好处:1)存储更多具有代表性的“pixel”sample,占用内存少;和2)可以让对比损失进一步探索像素-区域关系。 

Memory bank总体大小:

Hard Example Sampling

Semi-Hard Example Sampling:一些研究建议使用困难样本,因为用最难的样本优化度量学习可能会导致糟糕的局部极小值。因此,用了Semi-Hard Example Sampling:对于每个锚嵌入,我们首先收集前10%的最接近的负样本(从memory bank M中抽取前10%的最远正例),然后随机抽取K个负样本。

Segmentation-Aware Hard Anchor Sampling:开发了一个锚点采样策略,而不是挖掘信息丰富的正负样本,具体是:具有错误预测的像素,即,被视为hard anchor。对于对比损失计算(参见Eq.(3)),一半的锚是随机抽样的,另一半是hard anchor。这种锚点采样策略使我们的对比学习能够更多地关注难以分类的像素。

对于Semi-Hard Example Sampling和Segmentation-Aware Hard Anchor Sampling:正负样本采样实例数量上,将采样实例的数量(即K)分别设置为1024和2048。对于每个小批次,每个类别采样50个锚(一半是随机采样,另一半是难以分割的)。

结果:

使用不同损失函数的可视化结果图:

 

表1是只在单张图像内部(intra)选择正负样本和在所有图像(inter)选择正负样本作比较,表2是比较了没有memory bank,和memory bank只有pixel memory和region memory各自的部分等的比较,表3是比较困难负样本挖掘方法。

 

问题:

1、fPROJ和memory bank在训练期间和inference阶段的使用?

projection head和memory bank只在training期间应用,在inference时删除。因此,它对网络不引入任何改变或额外的计算成本。

  1. Specifically, some methods [69] directly store the embeddings of all the training samples in the memory, however, easily suffering from asynchronous update这一句话中异步更新是指什么?

同步更新类似于刷新,更新一个内容要把所有的内容都刷新一遍。而异步则只刷新更新或修改的部分,不重新将所有内容全部刷新一遍。

相关内容

热门资讯

你的晚安是我的早安是什么歌曲 你的晚安是我的早安是什么歌曲苏天伦《你的早安是我的晚安》“你的晚安是我的早安”是单小源的歌曲《东京遇...
积极进取的小故事 积极进取的小故事现代的普通人,不要名人的。不能与别人重复,尽快回答   啄木鸟的故事       啄...
熊出没之探险日记3什么时候播出... 熊出没之探险日记3什么时候播出?熊出没之探险日记3,春节前播放。熊出没只是探险日记三2020年5月4...
谁知道所有有关“七”的歌?拜托... 谁知道所有有关“七”的歌?拜托了各位 谢谢就是歌曲名里有“七”这个字的!谢谢七月七迅谈日晴 看我七十...
求一本小说 女主穿越了三次 每... 求一本小说 女主穿越了三次 每次都在福临身边 后来怀孕了孩子被打掉了那个 女主叫什么雯?那个女主就是...
如果记忆不说话,流年也会开出花... 如果记忆不说话,流年也会开出花的基本信息书 名:《如果记忆不弯饥好说话,流年也会开出花》埋铅 作 者...
你好,旧时光漫画版在哪里可以看... 你好,旧时光漫画版在哪里可以看?暂时在绘心上连载
一首英文歌,男的组合唱的,MV... 一首英文歌,男的组合唱的,MV是一个婚礼的过程。求歌名。是不是darin的can'tstoplove...
为什么很多人喜欢用胶片相机? 为什么很多人喜欢用胶片相机?有一种情怀叫做“怀旧“吧,现在数码相机越来越普遍了,已经到了”全民摄影“...
女主先爱上男主,男主却不喜欢女... 女主先爱上男主,男主却不喜欢女主或者是另有所爱,最后女主男主还是在一起的穿越小说。有木有再生缘:我的...
爱情失恋伤感句子 爱情失恋伤感句子越是美好的从前,越幸福的曾经,现在只能带来锥心的疼痛,痛到撕心裂肺,肝肠寸断,终于痛...
24岁穿这个会不会显老 24岁穿这个会不会显老有点显老,这个颜色款式,颜色有点暗,没有活力,属于那种气质佳,长得高雅的女人,...
哈尔的移动城堡英语版 哈尔的移动城堡英语版可以发给我吗度盘~请查收~
秦时明月之万里长城什么时候播 秦时明月之万里长城什么时候播据说是今年暑假开播别急,官网什么的信他你就输了,12年之前底应该会出,杭...
孩子会得抽动症吗? 孩子会得抽动症吗?我天生的气性比较大,有时跟别人斗嘴时候就会手脚哆嗦,麻木,我问一下这是不是抽动症就...
亨德尔一生为音乐献出了怎样的贡... 亨德尔一生为音乐献出了怎样的贡献?亨德尔一生写了歌剧41部,清唱剧21部,以及大量的管乐器与弦乐器的...
礼仪起源和发展的经典故事? 礼仪起源和发展的经典故事?一、礼仪的起源;1、天神生礼仪;2、礼为天地人的统一体;3、礼产生于人的自...
描写桂林山水的句子有哪些? 描写桂林山水的句子有哪些?天下风光数桂林有杨万里的“梅花五岭八桂林,青罗带绕碧玉簪”;有邹应龙的“无...
避免与强敌正面对决的成语 避免与强敌正面对决的成语避免与强敌正面对决的成语避实就虚 【近义】避重就轻、避难就易、声东击西【反义...
多愁善感类的成语 多愁善感类的成语心细如发【解释】:极言小心谨慎,考虑周密。亦作“心细于发”。【出自】:吴梅《题天香石...