arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2209.02178cs.CV
收藏

Transformer-CNN Cohort:通过两个学生的优势实现半监督语义分割

Transformer-CNN Cohort: Semi-supervised Semantic Segmentation by the Best of Both Students

HKUST(GZ)(香港科技大学(广州));HKUST(香港科技大学);Brown University(布朗大学);Northeastern University(东北大学)

Xu Zheng, Yunhao Luo, Chong Fu, Kangcheng Liu, Lin Wang

AI总结:

提出融合 ViT 与 CNN 双学生的 TCC 半监督语义分割框架,通过 CFCD 和 CCD 实现伪标签、特征与预测一致性蒸馏,并在 Cityscapes、Pascal VOC 2012 上显著优于现有方法。

摘要:

流行的半监督语义分割方法大多采用基于卷积神经网络(CNN)的单一网络模型,并对模型在输入或模型扰动下的预测施加一致性。然而,这种学习范式存在两个关键局限:a)为无标签数据学习判别性特征;b)从整幅图像中同时学习全局和局部信息。本文提出一种名为 Transformer-CNN Cohort(TCC)的新型半监督学习(SSL)方法,它由两个学生组成,一个基于视觉 Transformer(ViT),另一个基于 CNN。我们的方法通过对无标签数据进行伪标签化,巧妙地在预测和异构特征空间上引入多级一致性正则化。首先,由于 ViT 学生的输入是图像块,所提取的特征图编码了关键的类别统计信息。为此,我们提出类别感知特征一致性蒸馏(CFCD),该方法首先利用每个学生的输出作为伪标签,并生成类别感知特征(CF)图,用于两个学生之间的知识迁移。其次,由于 ViT 学生在所有层上具有更统一的表示,我们提出一致性感知交叉蒸馏(CCD),以在该 cohort 的逐像素预测之间迁移知识。我们在 Cityscapes 和 Pascal VOC 2012 数据集上验证了 TCC 框架,结果表明其大幅优于现有 SSL 方法。

英文摘要:

The popular methods for semi-supervised semantic segmentation mostly adopt a unitary network model using convolutional neural networks (CNNs) and enforce consistency of the model's predictions over perturbations applied to the inputs or model. However, such a learning paradigm suffers from two critical limitations: a) learning the discriminative features for the unlabeled data; b) learning both global and local information from the whole image. In this paper, we propose a novel Semi-supervised Learning (SSL) approach, called Transformer-CNN Cohort (TCC), that consists of two students with one based on the vision transformer (ViT) and the other based on the CNN. Our method subtly incorporates the multi-level consistency regularization on the predictions and the heterogeneous feature spaces via pseudo-labeling for the unlabeled data. First, as the inputs of the ViT student are image patches, the feature maps extracted encode crucial class-wise statistics. To this end, we propose class-aware feature consistency distillation (CFCD) that first leverages the outputs of each student as the pseudo labels and generates class-aware feature (CF) maps for knowledge transfer between the two students. Second, as the ViT student has more uniform representations for all layers, we propose consistency-aware cross distillation (CCD) to transfer knowledge between the pixel-wise predictions from the cohort. We validate the TCC framework on Cityscapes and Pascal VOC 2012 datasets, which outperforms existing SSL methods by a large margin.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑