Learning to Detour: Shortcut Mitigating Augmentation for Weakly Supervised Semantic Segmentation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Kwon, JuneHyoung, Lee, Eunju, Cho, Yunsung, Kim, YoungBin
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909212243656704
author Kwon, JuneHyoung
Lee, Eunju
Cho, Yunsung
Kim, YoungBin
author_facet Kwon, JuneHyoung
Lee, Eunju
Cho, Yunsung
Kim, YoungBin
contents Weakly supervised semantic segmentation (WSSS) employing weak forms of labels has been actively studied to alleviate the annotation cost of acquiring pixel-level labels. However, classifiers trained on biased datasets tend to exploit shortcut features and make predictions based on spurious correlations between certain backgrounds and objects, leading to a poor generalization performance. In this paper, we propose shortcut mitigating augmentation (SMA) for WSSS, which generates synthetic representations of object-background combinations not seen in the training data to reduce the use of shortcut features. Our approach disentangles the object-relevant and background features. We then shuffle and combine the disentangled representations to create synthetic features of diverse object-background combinations. SMA-trained classifier depends less on contexts and focuses more on the target object when making predictions. In addition, we analyzed the behavior of the classifier on shortcut usage after applying our augmentation using an attribution method-based metric. The proposed method achieved the improved performance of semantic segmentation result on PASCAL VOC 2012 and MS COCO 2014 datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2405_18148
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Learning to Detour: Shortcut Mitigating Augmentation for Weakly Supervised Semantic Segmentation
Kwon, JuneHyoung
Lee, Eunju
Cho, Yunsung
Kim, YoungBin
Computer Vision and Pattern Recognition
Artificial Intelligence
Weakly supervised semantic segmentation (WSSS) employing weak forms of labels has been actively studied to alleviate the annotation cost of acquiring pixel-level labels. However, classifiers trained on biased datasets tend to exploit shortcut features and make predictions based on spurious correlations between certain backgrounds and objects, leading to a poor generalization performance. In this paper, we propose shortcut mitigating augmentation (SMA) for WSSS, which generates synthetic representations of object-background combinations not seen in the training data to reduce the use of shortcut features. Our approach disentangles the object-relevant and background features. We then shuffle and combine the disentangled representations to create synthetic features of diverse object-background combinations. SMA-trained classifier depends less on contexts and focuses more on the target object when making predictions. In addition, we analyzed the behavior of the classifier on shortcut usage after applying our augmentation using an attribution method-based metric. The proposed method achieved the improved performance of semantic segmentation result on PASCAL VOC 2012 and MS COCO 2014 datasets.
title Learning to Detour: Shortcut Mitigating Augmentation for Weakly Supervised Semantic Segmentation
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2405.18148