Annotation-Efficient Polyp Segmentation via Active Learning
CoRR(2024)
摘要
Deep learning-based techniques have proven effective in polyp segmentation
tasks when provided with sufficient pixel-wise labeled data. However, the high
cost of manual annotation has created a bottleneck for model generalization. To
minimize annotation costs, we propose a deep active learning framework for
annotation-efficient polyp segmentation. In practice, we measure the
uncertainty of each sample by examining the similarity between features masked
by the prediction map of the polyp and the background area. Since the
segmentation model tends to perform weak in samples with indistinguishable
features of foreground and background areas, uncertainty sampling facilitates
the fitting of under-learning data. Furthermore, clustering image-level
features weighted by uncertainty identify samples that are both uncertain and
representative. To enhance the selectivity of the active selection strategy, we
propose a novel unsupervised feature discrepancy learning mechanism. The
selection strategy and feature optimization work in tandem to achieve optimal
performance with a limited annotation budget. Extensive experimental results
have demonstrated that our proposed method achieved state-of-the-art
performance compared to other competitors on both a public dataset and a
large-scale in-house dataset.
更多查看译文
AI 理解论文
溯源树
样例
![](https://originalfileserver.aminer.cn/sys/aminer/pubs/mrt_preview.jpeg)
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要