Maturity-Aware Active Learning for Semantic Segmentation

A Hierarchically-Adaptive Sample Assessment Approach

British Machine Vision Conference 2023

ABSTRACT

Active Learning (AL) for semantic segmentation is challenging due to heavy class imbalance and different ways of defining "sample" (pixels, areas, etc.), leaving the interpretation of the data distribution ambiguous. We propose ``Maturity-Aware Distribution Breakdown-based Active Learning'' (MADBAL), an AL method that benefits from a hierarchical approach to define a multiview data distribution, which takes into account the different "sample" definitions jointly, hence able to select the most impactful segmentation pixels with comprehensive understanding. MADBAL also features a novel uncertainty formulation, where AL supporting modules are included to sense the features' maturity whose weighted influence continuously contributes to the uncertainty detection. In this way, MADBAL makes significant performance leaps even in the early AL stage, hence reducing the training burden significantly. It outperforms state-of-the-art methods on Cityscapes and PASCAL VOC datasets as verified in our extensive experiments.

Code

Experimental and implementation details of MADBAL can be found here.

Paper can be found here.

MADBAL


Figure 1. The proposed AL modules.


Figure 2. The proposed hierarchical sample selection approach.

Results


Figure 3. Comparison with SOTA on Cityscapes and VOC dataset using different backbones.


Figure 4. Comparison with SOTA on Cityscapes w.r.t. the number of clicks (left) and the visualization results on both datasets.

Related Publications

  1. Yazdani, Amirsaeed, et al. "Maturity-Aware Active Learning for Semantic Segmentation with Hierarchically-Adaptive Sample Assessment," Accepted to the 34th British Machine Vision Conference 2023. [BMVC]

Email
ipal.psu@gmail.com

Address
104 Electrical Engineering East,
University Park, PA 16802, USA

Lab Phone:
814-863-7810
814-867-4564