opencv网格划分:用于域适应语义分割的域无关先验
opencv网格划分:用于域适应语义分割的域无关先验图1 域适应过程混淆例子表1 两组相似类比的特征统计由于来自目标域的数据,没有标记,因此通过将映射到源域来学习语义对应,例如,通过基于GAN的图像级风格转移和通过伪标签模拟。这可以看做以一种弱监督的方式学习域适应函数。这导致近似的视觉表征,从而导致识别的不准确性。此外,我们对不同方法中的相似类别的特征进行了统计分析,表1列举了两组易混淆类别。可以看到在只有源域数据训练时,源域中相似类别虽然距离较近但是分布紧凑重叠较少,但是直接用这个模型统计目标域特征发现他们的重叠比较严重,一个经典的方法DACS[x],在引入ClassMix对两个域的图像进行混合操作后缓解了重叠的问题。
摘要:本文给大家分享一篇我们在CVPR 2022 上发表的论文:Domain-Agnostic Prior for Transfer Semantic Segmentation。文章提出了一种图像域无关的先验,可以有效地提升域适应语义分割的精度。
本文分享自华为云社区《EI盘古研究分享【CVPR2022】用于域适应语义分割的域无关先验-云社区-华为云》,作者: 198808xc。
1. UDA 分割中的相似类别混淆问题尽管self-training在UDA任务上取得不错的结果,但是在区分语义上相似的类时仍然存在困难,特别是当这些类在目标域中并不频繁出现时,例如摩托车只占总像素数的0.1%。
图1给出了摩托车类对和自行车类对容易混淆的例子,道路类对和人行道类对也容易混淆。对于上述现象,我们提出了一个假设。
由于来自目标域的数据,没有标记,因此通过将映射到源域来学习语义对应,例如,通过基于GAN的图像级风格转移和通过伪标签模拟。
这可以看做以一种弱监督的方式学习域适应函数。这导致近似的视觉表征,从而导致识别的不准确性。
此外,我们对不同方法中的相似类别的特征进行了统计分析,表1列举了两组易混淆类别。可以看到在只有源域数据训练时,源域中相似类别虽然距离较近但是分布紧凑重叠较少,但是直接用这个模型统计目标域特征发现他们的重叠比较严重,一个经典的方法DACS[x],在引入ClassMix对两个域的图像进行混合操作后缓解了重叠的问题。
表1 两组相似类比的特征统计
图1 域适应过程混淆例子
2. 方法2.1 基线方法DACS在mean-teacher训练框架下为目标域图像生成伪标签,然后用目标域和源域数据一起训练网络是一种常用的方法,在此基础上,DACS[1]提出用混合域M的训练数据替换来自目标域的训练数据。混合域M的训练数据生成过程:在每次训练迭代中,对带有(真或伪)标签的源图像和目标图像进行采样并裁剪成相同的分辨率。接下来,从源域标签中随机选择一个类子集,并得到一个与图像相同大小的二进制掩码,属于选中类别的位置为1其他为0,再根据此掩码将源域和目标域数据混合在一起。
2.2 为UDA分割任务提供域无关先验为了获得对目标域特征更准确的估计,我们参考贝叶斯理论,后验分布由先验和似然组成。在我们的设置中,似然来自目标数据集,没有足够的数据来保证准确的估计。解决方案是为每个类别引入信息先验,这些先验和域无关也不会受域差异的影响。我们找到两种先验,一种是one-hot编码,另外一种借助word2vec[2]来为每个类别生成编码。这样每个类别都有一个先验编码向量,然后根据每个图像的标签,按位置替换为先验编码,构成一个先验编码图,如图2。然后我们将其和网络提取的是视觉特征进行约束。为了让视觉特征更多保留和类别语义相关的信息减少域相关的信息,二者通过简单转换后用L2 loss拉近。这就是我们新引入的DAP loss。
该损失函数与DACS的分割交叉熵函数相加构成完整的训练过程。
图2 先验编码图构成过程
2.3 一些思考据我们所知,这是第一个将文本编码嵌入到UDA分割的工作,并产生了相当大的准确性增益,这证明了语言线索协助视觉识别的有效性。然而,这还只是一个初步的解决方案,还可以发现一些可能的方向
1)加强文本嵌入。目前使用的word2vec的特点是它不考虑同一个语义对应不同的单词。我们试图通过搜索语义相似的单词来增强先验,但获得的准确性几乎没有提高。这可能需要探索文本世界的复杂机制。
2)构建领域不可知论但视觉感知的先验。这就回答了一个问题:什么样的图像数据被认为可以提供域无关信息?答案可能存在于广义数据集,如ImageNet[3]或Conceptual Captioning [4],或甚至预先训练的图像文本模型,如CLIP[5]。这可能是解开领域相关的信息以避免过度拟合的一个主要挑战,值得我们继续探索下去。
3. 实验结果3.1 数据和设置我们在常用的街景分割场景中进行评估,将信息从合成域迁移到真实域。GTAv和SYNTHIA作为复合域数据集,Cityscapes作为真实域。为了与其他方法保持一致,我们使用Deeplabv2框架下RseNet101基本模型作为我们的图像编码器,ASPP分类器作为输出层。
3.2 GTAv迁移到Cityscapes我们首先在从GTAv到Cityscapes的域适应分割任务上评估DAP。DAP在19个级别上达到55.0%的平均IOU,这个结果超过基线2.9%,也优于其他单阶段和单模型训练的方法。我们将DAP的输出作为伪标签输入ProDA的第1阶段,然后保持ProDA[6]第2和第3阶段训练方法不变。我们将结果提升了2.3%,是目前最高的精度。
图4. GTAv迁移到Cityscapes结果比较。
3.3 SYNTHIA迁移到Cityscapes将SYNTHIA转移到Cityscapes的结果,如表所示,同样比DACS在13类和16类上均提升了1.3%,比ProDA提升了4.4%和2.3%。再次验证了我们方法的有效性。
图5. SYNTHIA迁移到Cityscapes结果比较。
3.4 DAP对于易混淆类别的辨别在GTAv到Cityscapes迁移实验中,自行车和摩托车的细分iou分别从42.6%、25.1%提高到53.1%、42.2%。从SYNTHIA到Cityscapes,这两个类别平均提升2.4%。我们在图6中可视化了一个分割的例子。除了定性观察自行车与摩托车、道路与人行道的区分改善之外,我们还注意到这一改善背后的原因是这些易混淆类别的特征分布得到改善。这与表1中显示的统计数据一致,表明DAP减少了自行车和摩托车的估计分布以及道路和人行道之间的IOU。
图6.GTA迁移到Cityscapes实验中自行车和摩托车分割结果对比,以及特征分布对比。
4. 结语到这里我们的文章就分享完了,论文中还有更详细的消融实验来证明我们方法的有效性,就不在这里详细列举,更多内容欢迎关注我们的原文。
论文:https://arxiv.org/abs/2204.02684
参考文献[1] 威廉·特兰海登、维克多·奥尔森、朱利亚诺·平托、伦纳特·斯文森。Dacs:通过跨域混合采样进行域适应。在IEEE / CVF计算机视觉应用冬季会议论文集,第1379-1389页,2021年。
[2] 托马斯·米科洛夫、伊利亚·苏茨克弗、陈凯、格雷格·科拉多、杰夫·迪恩。单词和短语的分布式表示及其组成性。在神经信息处理系统的进展,第3111-3119页,2013年。
[3] 邓佳、魏东、理查德·索彻、李丽佳、李凯、李飞飞。图像网:大规模的分层图像数据库。2009年IEEE计算机视觉和模式识别会议,第248-255页。艾维, 2009.
[4] 皮尤什·夏尔马、丁楠、塞巴斯蒂安·古德曼、拉杜·索里卡特。概念性标题:用于自动图像标题的已清理、超专名的图像替代文本数据集。收录于计算语言学协会第56届年会论文集(第1卷:长篇论文),第2556-2565页,2018年。
[5] 亚历克·拉德福德、金钟郁、克里斯·哈拉西、阿迪亚·拉梅什、加布里埃尔·吴、桑迪尼·阿加瓦尔、吉里什·萨斯特里、阿曼达·阿斯克尔、帕梅拉·米什金、杰克·克拉克等人。从自然语言监督中学习可转移的视觉模型。arXiv 预印本 arXiv:2103.00020, 2021.
[6] 张攀, 张波, 张婷, 陈东, 永旺, 方文.用于域自适应语义分割的典型伪标签去噪和目标结构学习。在IEEE / CVF计算机视觉和模式识别会议论文集,第12414-12424页,2021年。
点击下方,第一时间了解华为云新鲜技术~
华为云博客_大数据博客_AI博客_云计算博客_开发者中心-华为云