HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

ISIM: Iterative Self-Improved Model for Weakly Supervised Segmentation

Cenk Bircanoglu Nafiz Arica

ISIM: Iterative Self-Improved Model for Weakly Supervised Segmentation

Abstract

Weakly Supervised Semantic Segmentation (WSSS) is a challenging task aiming to learn the segmentation labels from class-level labels. In the literature, exploiting the information obtained from Class Activation Maps (CAMs) is widely used for WSSS studies. However, as CAMs are obtained from a classification network, they are interested in the most discriminative parts of the objects, producing non-complete prior information for segmentation tasks. In this study, to obtain more coherent CAMs with segmentation labels, we propose a framework that employs an iterative approach in a modified encoder-decoder-based segmentation model, which simultaneously supports classification and segmentation tasks. As no ground-truth segmentation labels are given, the same model also generates the pseudo-segmentation labels with the help of dense Conditional Random Fields (dCRF). As a result, the proposed framework becomes an iterative self-improved model. The experiments performed with DeepLabv3 and UNet models show a significant gain on the Pascal VOC12 dataset, and the DeepLabv3 application increases the current state-of-the-art metric by %2.5. The implementation associated with the experiments can be found: https://github.com/cenkbircanoglu/isim.

Code Repositories

cenkbircanoglu/isim
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
weakly-supervised-semantic-segmentation-onISIM (ResNet-101)
Mean IoU: 70.51
weakly-supervised-semantic-segmentation-onISIM (ResNeSt-200)
Mean IoU: 74.90
weakly-supervised-semantic-segmentation-on-1ISIM (ResNeSt-200)
Mean IoU: 74.98
weakly-supervised-semantic-segmentation-on-1ISIM (ResNet-101)
Mean IoU: 71.45

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
ISIM: Iterative Self-Improved Model for Weakly Supervised Segmentation | Papers | HyperAI