Multi-layered self-attention mechanism for weakly supervised semantic segmentation

被引:2
|
作者
Yaganapu, Avinash [1 ]
Kang, Mingon [1 ]
机构
[1] Univ Nevada Las Vegas, Dept Comp Sci, 4505 S Maryland Pkwy, Las Vegas, NV 89154 USA
基金
美国国家科学基金会;
关键词
Weakly supervised semantic segmentation; Segmentation; Self-attention; Image-level labels;
D O I
10.1016/j.cviu.2023.103886
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Weakly Supervised Semantic Segmentation (WSSS) provides efficient solutions for semantic image segmen-tation using image-level annotations. WSSS requires no pixel-level labeling that Fully Supervised Semantic Segmentation (FSSS) does, which is time-consuming and label-intensive. Most WSSS approaches have leveraged Class Activation Maps (CAM) or Self-Attention (SA) to generate pseudo pixel-level annotations to perform semantic segmentation tasks coupled with fully supervised approaches (e.g., Fully Convolutional Network). However, those approaches often provides incomplete supervision that mainly includes discriminative regions from the last convolutional layer. They may fail to capture regions of low-or intermediate-level features that may not be present in the last convolutional layer. To address the issue, we proposed a novel Multilayered Self-Attention (Multi-SA) method that applies a self-attention module to multiple convolutional layers, and then stack feature maps from the self-attention layers to generate pseudo pixel-level annotations. We demonstrated that integrated feature maps from multiple self-attention layers produce higher coverage in semantic segmentation than using only the last convolutional layer through intensive experiments using standard benchmark datasets.
引用
收藏
页数:9
相关论文
共 50 条
  • [1] Non-bias self-attention learning for weakly supervised semantic segmentation
    Sun, Wanchun
    Feng, Xin
    Liu, Jingyao
    [J]. COMPUTERS & ELECTRICAL ENGINEERING, 2023, 105
  • [2] Weakly supervised histopathology image segmentation with self-attention
    Li, Kailu
    Qian, Ziniu
    Han, Yingnan
    Chang, Eric I-Chao
    Wei, Bingzheng
    Lai, Maode
    Liao, Jing
    Fan, Yubo
    Xu, Yan
    [J]. MEDICAL IMAGE ANALYSIS, 2023, 86
  • [3] Self-Attention Prediction Correction with Channel Suppression for Weakly-Supervised Semantic Segmentation
    Sun, Guoying
    Yang, Meng
    [J]. 2023 IEEE INTERNATIONAL CONFERENCE ON MULTIMEDIA AND EXPO, ICME, 2023, : 846 - 851
  • [4] Saliency Guided Self-Attention Network for Weakly and Semi-Supervised Semantic Segmentation
    Yao, Qi
    Gong, Xiaojin
    [J]. IEEE ACCESS, 2020, 8 : 14413 - 14423
  • [5] Weakly supervised action segmentation with effective use of attention and self-attention
    Bin Ng, Yan
    Fernando, Basura
    [J]. COMPUTER VISION AND IMAGE UNDERSTANDING, 2021, 213
  • [6] Embedded Discriminative Attention Mechanism for Weakly Supervised Semantic Segmentation
    Wu, Tong
    Huang, Junshi
    Gao, Guangyu
    Wei, Xiaoming
    Wei, Xiaolin
    Luo, Xuan
    Liu, Chi Harold
    [J]. 2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, : 16760 - 16769
  • [7] Pyramid Self-attention for Semantic Segmentation
    Qi, Jiyang
    Wang, Xinggang
    Hu, Yao
    Tang, Xu
    Liu, Wenyu
    [J]. PATTERN RECOGNITION AND COMPUTER VISION, PT I, 2021, 13019 : 480 - 492
  • [8] Adaptive soft erasure with edge self-attention for weakly supervised semantic segmentation: Thyroid ultrasound image case study
    Yu, Mei
    Han, Ming
    Li, Xuewei
    Wei, Xi
    Jiang, Han
    Chen, Huiling
    Yu, Ruiguo
    [J]. COMPUTERS IN BIOLOGY AND MEDICINE, 2022, 144
  • [9] Weakly supervised semantic segmentation for point cloud based on view-based adversarial training and self-attention fusion
    Miao, Yongwei
    Ren, Guoxiang
    Wang, Jinrong
    Liu, Fuchang
    [J]. COMPUTERS & GRAPHICS-UK, 2023, 116 : 46 - 54
  • [10] Weakly Supervised Semantic Segmentation of Remote Sensing Images Based on Progressive Mining and Saliency-Enhanced Self-Attention
    Hao, Ting
    Bai, Shuya
    Wu, Tianyu
    Zhang, Libao
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2024, 21 : 1 - 5