通过增强区域多样性,用极少标注像素实现接近全监督的分割效果
Exploring Spatial Diversity for Region-based Active Learning
- 在区域主动学习中引入空间多样性约束,与不确定性联合优化
- 仅用5-9%标注像素即达全监督模型95%性能
- 适合标注成本高的密集分割任务,如自动驾驶场景
当前语义分割方法依赖大规模标注数据集,而像素级标注成本高昂。本文提出基于区域的主动学习策略,通过选择高信息量图像区域而非整图进行标注,以降低标注成本。关键创新在于引入局部空间多样性机制,将其与传统不确定性等选择标准统一整合到优化框架中。在Cityscapes和PASCAL VOC数据集上的实验表明,该方法显著提升基于不确定性和特征多样性的主动学习性能。仅需5%-9%的标注像素即可达到全监督方法95%的性能,优于所有现有区域级主动学习方法。
原文摘要 · Abstract (English)
State-of-the-art methods for semantic segmentation are based on deep neural networks trained on large-scale labeled datasets. Acquiring such datasets would incur large annotation costs, especially for dense pixel-level prediction tasks like semantic segmentation. We consider region-based active learning as a strategy to reduce annotation costs while maintaining high performance. In this setting, batches of informative image regions instead of entire images are selected for labeling. Importantly, we propose that enforcing local spatial diversity is beneficial for active learning in this case, and to incorporate spatial diversity along with the traditional active selection criterion, e.g., data sample uncertainty, in a unified optimization framework for region-based active learning. We apply this framework to the Cityscapes and PASCAL VOC datasets and demonstrate that the inclusion of spatial diversity effectively improves the performance of uncertainty-based and feature diversity-based active learning methods. Our framework achieves $95\%$ performance of fully supervised methods with only $5-9\%$ of the labeled pixels, outperforming all state-of-the-art region-based active learning methods for semantic segmentation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。