MEAL: Manifold Embedding-based Active Learning

Abstract

Image segmentation is a common and challenging task in autonomous driving. Availability of sufficient pixel-level annotations for the training data is a hurdle. Active learning helps learning from small amounts of data by suggesting the most promising samples for labeling. In this work, we propose a new pool-based method for active learning, which proposes promising patches extracted from full image, in each acquisition step. The problem is framed in an exploration-exploitation framework by combining an embedding based on Uniform Manifold Approximation to model representativeness with entropy as uncertainty measure to model informativeness. We applied our proposed method to the autonomous driving datasets CamVid and Cityscapes and performed a quantitative comparison with state-of-the-art baselines. We find that our active learning method achieves better performance compared to previous methods.

References in corpus (7)

• PyTorch: An Imperative Style, High-Performance Deep Learning Library
• MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications
• UMAP: Uniform Manifold Approximation and Projection for Dimension Reduction
• Rethinking Atrous Convolution for Semantic Image Segmentation
• Recent progress in semantic image segmentation
• Reinforced active learning for image segmentation
• Importance of Self-Consistency in Active Learning for Semantic Segmentation