Weighted and Mumford-Shah losses for weakly supervised semantic segmentation

Yihao Zeng

doi:10.1117/12.3026432

9 February 2024 Weighted and Mumford-Shah losses for weakly supervised semantic segmentation

Yihao Zeng

Proceedings Volume 13073, Third International Conference on High Performance Computing and Communication Engineering (HPCCE 2023); 1307303 (2024) https://doi.org/10.1117/12.3026432
Event: Third International Conference on High Performance Computing and Communication Engineering (HPCCE 2023), 2023, Changsha, China

Abstract

In weakly supervised semantic segmentation (WSSS) tasks, the noise in pseudo labels poses a significant challenge to the training of segmentation networks. However, the widely used cross-entropy loss, which only considers individual pixel information in the images, is insufficient to address this issue. In order to effectively train segmentation networks using noisy labels, we propose a Weakly Supervised Semantic Segmentation method (WMS) based on weighted loss and Mumford-Shah loss. Firstly, we introduce a weighted loss with confidence weights for pseudo labels. This loss assigns weights to each pseudo label by incorporating a confidence indicator, enhancing the network's ability to resist noise interference. Additionally, we propose a Mumford-Shah (MS) loss based on variational segmentation. By leveraging the similarity between pixels in the original image, this loss introduces additional noise-free self-supervised information to assist in the training of the segmentation network, further suppressing the interference caused by noisy labels. Extensive experiments on the PASCAL VOC 2012 dataset demonstrate that the proposed method significantly improves WSSS performance.

(2024) Published by SPIE. Downloading of the abstract is permitted for personal use only.

Citation Download Citation

Yihao Zeng "Weighted and Mumford-Shah losses for weakly supervised semantic segmentation", Proc. SPIE 13073, Third International Conference on High Performance Computing and Communication Engineering (HPCCE 2023), 1307303 (9 February 2024); https://doi.org/10.1117/12.3026432

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available

Members: $17.00

Non-members: $21.00 ADD TO CART

PROCEEDINGS
9 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Image segmentation

Education and training

Semantics

Binary data

Neural networks

Ablation

Image classification

Show All Keywords

Keywords/Phrases

Search In:

Publication Years