[论文][半监督语义分割]Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision

CVPR2021 原文

半监督语义分割方法的总结:

主要思想: Consistency regularization :希望不同扰动之下网络的输出结果一致,扰动的加入的位置:(1)在输入图片上加扰动(2)在某一层的输出特征上添加扰动

创新点:

  1. 鼓励两个初始化不同(不同扰动)的网络的一致性
  2. 利用半监督的方式相当于扩充了数据集

网络结构

两个网络的结构相同,但使用不同的初始化(作为不同的扰动),具体而言,2个网络backbone部分使用同样的ImageNet预训练权重,Segmentation Head部分使用不同的随机初始化权重。

[论文][半监督语义分割]Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision
[论文][半监督语义分割]Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision
[论文][半监督语义分割]Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision
上图中的Y1和Y2分别表示2个网络输出的one hot标签,⟶ 表示前向计算,− − → 表示监督信息的传播,//表示不进行梯度计算。
对于无标签数据:
将上分支网络one-hot结果作为label监督下分支,下分支网络one-hot结果作为label监督上分支,计算一致性损失:
[论文][半监督语义分割]Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision
对于有标签数据:
将带标签的数据输入2个网络,使用ground truth分别监督这2个网络进行训练。用交叉熵函数计算损失
[论文][半监督语义分割]Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision
同时,也计算一致性损失(方式和无标签数据相同)
总网络的一致性损失分为有标签和无标签两种:
[论文][半监督语义分割]Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision
总网络损失:
[论文][半监督语义分割]Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision

多种网络结构对比

[论文][半监督语义分割]Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision
a) 本论文网络结构图CPS,叙述如上
b) GCT网络结构图,和本方法相同,唯一的区别在于GCT使用confidence map作为监督信号,而CPS使用one-hot label
c)MeanTeacher 使用学生网络和老师网络,两个网络结构相同,参数初始化不同,学生网络用老师网络得到的confidence map作为监督信号,老师网络随着学生网络的权重变化按照指数平均不断变化
d)一张原图X分别经过弱数据增强和强数据增强放入同一个网络中,弱数据增强所得one hot结果作为强数据增强结果的真值,用于监督弱数据增强的结果

实验

a,b图分别使用ResNet50,ResNet101作为backbone
蓝:baseline(不加入cutmix数据增强):只使用有标签的数据
红:使用两个网络,半监督学习,不加入cutmix数据增强
绿:使用两个网络,半监督学习,加入cutmix数据增强

[论文][半监督语义分割]Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision
Table1:在Pascal VOC上使用不同backbone和不同有标签数据比例得到结果和其他SOTA方法对比
[论文][半监督语义分割]Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision
Table1:在Cityscapes上使用不同backbone和不同有标签数据比例得到结果和其他SOTA方法对比
[论文][半监督语义分割]Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision
[论文][半监督语义分割]Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision

[论文][半监督语义分割]Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision
在两个数据集上分别进行的消融实验,损失函数的各部分对于结果的影响
[论文][半监督语义分割]Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision
使用极少的有标签数据实验结果
[论文][半监督语义分割]Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision
[论文][半监督语义分割]Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision
[论文][半监督语义分割]Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision

Original: https://www.cnblogs.com/qiqiblogs/p/15913083.html
Author: 柒七同学
Title: [论文][半监督语义分割]Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/575662/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球