基于类别级记忆库的像素级对比学习的半监督语义分割

本文提出了一种新颖的半监督语义分割方法。该方法的核心在于我们设计的对比学习模块，该模块强制分割网络在全数据集范围内，对同类别样本产生相似的像素级特征表示。为实现这一目标，我们构建了一个持续更新的内存库，其中存储来自标注数据的高质量、相关性高的特征向量。在端到端的训练过程中，来自标注数据和未标注数据的特征均被优化，使其与内存库中同类别样本的特征保持一致。实验结果表明，该方法在多个知名公开基准上均超越了当前最先进的半监督语义分割与半监督域适应性能，尤其在标注数据稀缺这一更具挑战性的场景下，取得了更为显著的提升。项目代码已开源：https://github.com/Shathe/SemiSeg-Contrastive