View-coherent correlation consistency for semi-supervised semantic segmentation

被引:4
|
作者
Hou, Yunzhong [1 ]
Gould, Stephen [1 ]
Zheng, Liang [1 ]
机构
[1] Australian Natl Univ, Canberra, ACT 2601, Australia
关键词
Semi-supervised learning; Semantic segmentation; Contrastive learning; Data augmentation;
D O I
10.1016/j.patcog.2023.110089
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Semi-supervised semantic segmentation needs rich and robust supervision for unlabeled data. However, promoting or punishing feature similarities with vanilla contrastive learning can be unreliable for semi supervised semantic segmentation: pixel pairs are assigned as either positive or negative based on noisy pseudo labels, and both reliable and wrongly-assigned pairs receive uniform penalties. To address this issue, we propose correlation consistency learning, which leverages rich pairwise relationships in self-correlation matrices and matches them to the similarities between soft pseudo labels to provide robust supervision. Unlike vanilla contrastive learning, our approach prioritizes pairs with highly confident pseudo labels and applies weaker penalties for pairs that are less confident. We also introduce a strong semi-supervised learning pipeline that applies data augmentation in a view-coherent manner: even under complex augmentation strategies, for each pixel, a match can be found in different augmentation views. The novelties of the proposed method are the correlation consistency loss and the view-coherent data augmentation, and their combination gives us the view-coherent correlation consistency (VC3) system, which achieves state-of-the-art results in several semi-supervised settings on two datasets.
引用
收藏
页数:10
相关论文
共 50 条
  • [1] Revisiting Consistency for Semi-Supervised Semantic Segmentation
    Grubisic, Ivan
    Orsic, Marin
    Segvic, Sinisa
    SENSORS, 2023, 23 (02)
  • [2] SEMI-SUPERVISED SEMANTIC SEGMENTATION CONSTRAINED BY CONSISTENCY REGULARIZATION
    Li, Xiaoqiang
    He, Qin
    Dai, Songmin
    Wu, Pin
    Tong, Weiqin
    2020 IEEE INTERNATIONAL CONFERENCE ON MULTIMEDIA AND EXPO (ICME), 2020,
  • [3] Conflict-Based Cross-View Consistency for Semi-Supervised Semantic Segmentation
    Wang, Zicheng
    Zhao, Zhen
    Xing, Xiaoxia
    Xu, Dong
    Kong, Xiangyu
    Zhou, Luping
    2023 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2023, : 19585 - 19595
  • [4] Semi-supervised Semantic Segmentation with Directional Context-aware Consistency
    Lai, Xin
    Tian, Zhuotao
    Jiang, Li
    Liu, Shu
    Zhao, Hengshuang
    Wang, Liwei
    Jia, Jiaya
    2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, : 1205 - 1214
  • [5] Semi-supervised Semantic Segmentation with Prototype-based Consistency Regularization
    Xu, Hai-Ming
    Liu, Lingqiao
    Bian, Qiuchen
    Yang, Zhen
    ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 35 (NEURIPS 2022), 2022,
  • [6] Perturbation consistency and mutual information regularization for semi-supervised semantic segmentation
    Wu, Yulin
    Liu, Chang
    Chen, Lei
    Zhao, Dong
    Zheng, Qinghe
    Zhou, Hongchao
    MULTIMEDIA SYSTEMS, 2023, 29 (02) : 511 - 523
  • [7] Perturbation consistency and mutual information regularization for semi-supervised semantic segmentation
    Yulin Wu
    Chang Liu
    Lei Chen
    Dong Zhao
    Qinghe Zheng
    Hongchao Zhou
    Multimedia Systems, 2023, 29 : 511 - 523
  • [8] Noise-robust consistency regularization for semi-supervised semantic segmentation
    Zhang, Haikuan
    Li, Haitao
    Zhang, Xiufeng
    Yang, Guanyu
    Li, Atao
    Du, Weisheng
    Xue, Shanshan
    Liu, Chi
    NEURAL NETWORKS, 2025, 184
  • [9] Revisiting Weak-to-Strong Consistency in Semi-Supervised Semantic Segmentation
    Yang, Lihe
    Qi, Lei
    Feng, Litong
    Zhang, Wayne
    Shi, Yinghuan
    2023 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR, 2023, : 7236 - 7246
  • [10] Transferable Semi-Supervised Semantic Segmentation
    Xiao, Huaxin
    Wei, Yunchao
    Liu, Yu
    Zhang, Maojun
    Feng, Jiashi
    THIRTY-SECOND AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE / THIRTIETH INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE CONFERENCE / EIGHTH AAAI SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2018, : 7420 - 7427