[RE] Background-Aware Pooling and Noise-Aware Loss for Weakly-Supervised Semantic Segmentation

Aryan Mehta, Karan Uppal, Kaushal Jadhav, Monish Natarajan, Mradul Agrawal, Debashish Chakravarty

Keywords: Semantic Segmentation, Weakly-Supervised Semantic Segmentation

TL;DR: The following submission is a reproducibility report for "Background-Aware Pooling and Noise-Aware Loss for Weakly-Supervised Semantic Segmentation" published in the CVPR 2021 as part of the ML Reproducibility Challenge 2021.

Abstract:

The following submission is a reproducibility report for "Background-Aware Pooling and Noise-Aware Loss for Weakly-Supervised Semantic Segmentation" published in CVPR 2021 as part of the ML Reproducibility Challenge 2021. The paper’s central claim revolves around the newly introduced Background Aware Pooling method to generate high-quality pseudo labels using bounding boxes as supervision and Noise Aware Loss to train a segmentation network using those noisy labels.

We started with the publicly available code-base provided by the authors and reproduced the results involving pseudo label generation. Further, we implemented Noise-Aware Loss and used it to train a semantic segmentation network, reproducing its claims. We performed many refactoring and upgrades on the author's code to include various procedures mentioned in the paper as well as reimplemented the code base in PyTorch Lightning for easier reproducibility.

To formulate the report, extensive experimentation on the author's code has been conducted to verify all the claims made by the author described in detail below. We also performed additional experiments to gauge the extent of improvement brought upon by the method over the state-of-the-art methods.

Paper Url: http://openaccess.thecvf.com//content/CVPR2021/papers/Oh_Background-Aware_Pooling_and_Noise-Aware_Loss_for_Weakly-Supervised_Semantic_Segmentation_CVPR_2021_paper.pdf

Paper Venue: CVPR 2021