Skip to yearly menu bar Skip to main content


Poster

CoSW: Conditional Sample Weighting for Smoke Segmentation with Label Noise

Lujian Yao · Haitao Zhao · Zhongze Wang · Kaijie Zhao · Jingchao Peng

East Exhibit Hall A-C #1003
[ ]
Wed 11 Dec 4:30 p.m. PST — 7:30 p.m. PST

Abstract:

Smoke segmentation is of great importance in precisely identifying the smoke location, enabling timely fire rescue and gas leak detection. However, due to the visual diversity and blurry edges of the non-grid smoke, noisy labels are almost inevitable in large-scale pixel-level smoke datasets. Noisy labels significantly impact the robustness of the model, and may lead to serious accidents. However, currently, there are no specific methods for addressing noisy labels in smoke segmentation. Smoke differs from regular objects as its transparency varies, causing inconsistent features in the noisy labels. In this paper, we propose a conditional sample weighting (CoSW). CoSW utilizes a multi-prototype framework, where prototypes serve as prior information to apply different weighting metrics to the different feature clusters. A novel regularized within-prototype entropy (RWE) is introduced to achieve CoSW and stable prototypes update. CoSW is concise and does not require data with clean labels during the training. The experiments show that our approach achieves SOTA performance on both real-world and synthetic noisy smoke segmentation datasets.

Live content is unavailable. Log in and register to view live content