HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Threshold Matters in WSSS: Manipulating the Activation for the Robust and Accurate Segmentation Model Against Thresholds

Minhyun Lee Dongseob Kim Hyunjung Shim

Threshold Matters in WSSS: Manipulating the Activation for the Robust and Accurate Segmentation Model Against Thresholds

Abstract

Weakly-supervised semantic segmentation (WSSS) has recently gained much attention for its promise to train segmentation models only with image-level labels. Existing WSSS methods commonly argue that the sparse coverage of CAM incurs the performance bottleneck of WSSS. This paper provides analytical and empirical evidence that the actual bottleneck may not be sparse coverage but a global thresholding scheme applied after CAM. Then, we show that this issue can be mitigated by satisfying two conditions; 1) reducing the imbalance in the foreground activation and 2) increasing the gap between the foreground and the background activation. Based on these findings, we propose a novel activation manipulation network with a per-pixel classification loss and a label conditioning module. Per-pixel classification naturally induces two-level activation in activation maps, which can penalize the most discriminative parts, promote the less discriminative parts, and deactivate the background regions. Label conditioning imposes that the output label of pseudo-masks should be any of true image-level labels; it penalizes the wrong activation assigned to non-target classes. Based on extensive analysis and evaluations, we demonstrate that each component helps produce accurate pseudo-masks, achieving the robustness against the choice of the global threshold. Finally, our model achieves state-of-the-art records on both PASCAL VOC 2012 and MS COCO 2014 datasets.

Code Repositories

gaviotas/amn
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
weakly-supervised-semantic-segmentation-onAMN (DeepLabV2-ResNet101)
Mean IoU: 69.5
weakly-supervised-semantic-segmentation-onAMN (DeepLabV2-ResNet101, MS-COCO-pretrained weights)
Mean IoU: 70.7
weakly-supervised-semantic-segmentation-on-1AMN (DeepLabV2-ResNet101, MS-COCO-pretrained weights)
Mean IoU: 70.6
weakly-supervised-semantic-segmentation-on-1AMN (DeepLabV2-ResNet101)
Mean IoU: 69.6
weakly-supervised-semantic-segmentation-on-4AMN (DeepLabV2-ResNet101)
mIoU: 44.7

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Threshold Matters in WSSS: Manipulating the Activation for the Robust and Accurate Segmentation Model Against Thresholds | Papers | HyperAI