HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

CW-BASS: Confidence-Weighted Boundary-Aware Learning for Semi-Supervised Semantic Segmentation

Ebenezer Tarubinga Jenifer Kalafatovich Seong-Whan Lee

CW-BASS: Confidence-Weighted Boundary-Aware Learning for Semi-Supervised Semantic Segmentation

Abstract

Semi-supervised semantic segmentation (SSSS) aims to improve segmentation performance by utilizing large amounts of unlabeled data with limited labeled samples. Existing methods often suffer from coupling, where over-reliance on initial labeled data leads to suboptimal learning; confirmation bias, where incorrect predictions reinforce themselves repeatedly; and boundary blur caused by limited boundary-awareness and ambiguous edge cues. To address these issues, we propose CW-BASS, a novel framework for SSSS. In order to mitigate the impact of incorrect predictions, we assign confidence weights to pseudo-labels. Additionally, we leverage boundary-delineation techniques, which, despite being extensively explored in weakly-supervised semantic segmentation (WSSS), remain underutilized in SSSS. Specifically, our method: (1) reduces coupling via a confidence-weighted loss that adjusts pseudo-label influence based on their predicted confidence scores, (2) mitigates confirmation bias with a dynamic thresholding mechanism that learns to filter out pseudo-labels based on model performance, (3) tackles boundary blur using a boundary-aware module to refine segmentation near object edges, and (4) reduces label noise through a confidence decay strategy that progressively refines pseudo-labels during training. Extensive experiments on Pascal VOC 2012 and Cityscapes demonstrate that CW-BASS achieves state-of-the-art performance. Notably, CW-BASS achieves a 65.9% mIoU on Cityscapes under a challenging and underexplored 1/30 (3.3%) split (100 images), highlighting its effectiveness in limited-label settings. Our code is available at https://github.com/psychofict/CW-BASS.

Code Repositories

psychofict/CW-BASS
Official
pytorch

Benchmarks

BenchmarkMethodologyMetrics
semi-supervised-semantic-segmentation-on-1CW-BASS (DeepLab v3+ with ResNet-50)
Validation mIoU: 78.43%
semi-supervised-semantic-segmentation-on-2CW-BASS (DeepLab v3+ with ResNet-50)
Validation mIoU: 77.20%
semi-supervised-semantic-segmentation-on-22CW-BASS (DeepLab v3+ with ResNet-50)
Validation mIoU: 75.00
semi-supervised-semantic-segmentation-on-27CW-BASS (DeepLab v3+ with ResNet-50)
Validation mIoU: 72.8
semi-supervised-semantic-segmentation-on-3CW-BASS (DeepLab v3+ with ResNet-50)
Validation mIoU: 65.87
semi-supervised-semantic-segmentation-on-4CW-BASS (DeepLab v3+ with ResNet-50)
Validation mIoU: 75.81%
semi-supervised-semantic-segmentation-on-44CW-BASS (DeepLab v3+ with ResNet-50)
Validation mIoU: 77.15
semi-supervised-semantic-segmentation-on-9CW-BASS (DeepLab v3+ with ResNet-50)
Validation mIoU: 76.2

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