One-shot active learning for image segmentation via contrastive learning and diversity-based sampling
Knowledge-Based Systems(2022)
Abstract
Image segmentation tasks based on deep learning usually require a large number of labeled samples to obtain great performance of Convolutional Neural Networks (CNNs). However, even if samples are abundant, a major issue remains that labeling samples is usually time-consuming and costly. Active learning can select valuable samples for annotation, so as to reduce the annotation cost as much as possible while maintaining the performance of CNNs. Most existing active learning approaches work in an iterative way. However, this iterative scheme needs more interaction with experts, more labor and more computing resources. In this paper, we propose a one-shot active learning framework, i.e. Contrastive Annotation (CA) based on contrastive self-supervised learning and diversity-based query strategy, aiming to select valuable samples in one-shot. Extensive experiments on three segmentation datasets, i.e., skin lesion segmentation, remote sensing image segmentation and chest X-ray segmentation, show that our proposed CA framework outperforms state-of-the-art methods by large margins.
MoreTranslated text
Key words
Active learning,Contrastive learning,Image segmentation
AI Read Science
Must-Reading Tree
Example
![](https://originalfileserver.aminer.cn/sys/aminer/pubs/mrt_preview.jpeg)
Generate MRT to find the research sequence of this paper
Chat Paper
Summary is being generated by the instructions you defined