arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2503.02459cs.CV
收藏

探索视觉Transformer中令牌级增强用于半监督语义分割

Exploring Token-Level Augmentation in Vision Transformer for Semi-Supervised Semantic Segmentation

South China University of Technology(华南理工大学);Pengcheng Laboratory(鹏城实验室);China Tobacco Guangdong Industrial Corporation(广东中烟工业有限责任公司);Institute of Automation, Chinese Academy of Sciences(中国科学院自动化研究所)

Dengke Zhang, Quan Tang, Fagui Liu, Haiqing Mei, C. L. Philip Chen

AI总结:

本文提出TokenMix令牌级数据增强和双分支框架,解决Vision Transformer在半监督语义分割中的适应性问题,并在多个基准上取得领先精度。

摘要:

近年来,半监督语义分割取得了显著进展。然而,现有算法基于卷积神经网络,由于概念上的差异,直接将其应用于Vision Transformer存在一定局限性。为此,我们提出了TokenMix,这是一种专为基于Vision Transformer的半监督语义分割设计的数据增强技术。TokenMix通过在令牌级别混合图像,与全局注意力机制良好契合,增强了图像块之间上下文信息的学习能力。我们进一步结合图像增强和特征增强,以提升增强的多样性。此外,为增强一致性正则化,我们提出了一个双分支框架,其中每个分支对输入图像应用图像增强和特征增强。我们在多个基准数据集上进行了广泛实验,包括Pascal VOC 2012、Cityscapes和COCO。结果表明,所提方法优于最先进算法,尤其在有限精细标注下,准确率提升显著。

英文摘要:

Semi-supervised semantic segmentation has witnessed remarkable advancements in recent years. However, existing algorithms are based on convolutional neural networks and directly applying them to Vision Transformers poses certain limitations due to conceptual disparities. To this end, we propose TokenMix, a data augmentation technique specifically designed for semi-supervised semantic segmentation with Vision Transformers. TokenMix aligns well with the global attention mechanism by mixing images at the token level, enhancing learning capability for contextual information among image patches. We further incorporate image augmentation and feature augmentation to promote the diversity of augmentation. Moreover, to enhance consistency regularization, we propose a dual-branch framework where each branch applies image and feature augmentation to the input image. We conduct extensive experiments across multiple benchmark datasets, including Pascal VOC 2012, Cityscapes, and COCO. Results suggest that the proposed method outperforms state-of-the-art algorithms with notably observed accuracy improvement, especially under limited fine annotations.

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

↑