arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2008.05723cs.CV
收藏

主动学习的情境多样性

Contextual Diversity for Active Learning

IIIT-Delhi(印度德里信息技术学院);Flixstock Inc.(Flixstock公司);Indian Institute of Technology Delhi(印度理工学院德里分校)

Sharat Agarwal, Himanshu Arora, Saket Anand, Chetan Arora

摘要:

大型注释数据集的需求限制了深度卷积神经网络(CNN)在许多实际应用中的使用。该问题可以通过使用主动学习(AL)技术来缓解,该技术在给定的注释预算下允许选择在微调时产生最大准确度的数据子集。最先进的人工智能方法通常依赖于视觉多样性或预测不确定性的测量,这是无法有效地捕捉空间背景的变化。另一方面,现代CNN架构大量使用空间上下文来实现高度准确的预测。由于上下文是难以评估的情况下,地面实况标签,我们引入了上下文多样性的概念,捕捉与空间共现类相关联的混乱。上下文多样性(CD)取决于一个关键的观察,即由CNN预测的感兴趣区域的概率向量通常包含来自更大感受野的信息。利用这一观察结果,我们在两个AL框架中使用拟议的CD度量:(1)基于核心集的策略和(2)基于强化学习的策略,用于主动帧选择。我们广泛的经验评估建立了语义分割、目标检测和图像分类的基准数据集上的主动学习的最新结果。我们的消融研究显示了使用情境多样性进行主动学习的明显优势。源代码和其他结果可在https://github.com/sharat29ag/CDAL上获得。

英文摘要:

Requirement of large annotated datasets restrict the use of deep convolutional neural networks (CNNs) for many practical applications. The problem can be mitigated by using active learning (AL) techniques which, under a given annotation budget, allow to select a subset of data that yields maximum accuracy upon fine tuning. State of the art AL approaches typically rely on measures of visual diversity or prediction uncertainty, which are unable to effectively capture the variations in spatial context. On the other hand, modern CNN architectures make heavy use of spatial context for achieving highly accurate predictions. Since the context is difficult to evaluate in the absence of ground-truth labels, we introduce the notion of contextual diversity that captures the confusion associated with spatially co-occurring classes. Contextual Diversity (CD) hinges on a crucial observation that the probability vector predicted by a CNN for a region of interest typically contains information from a larger receptive field. Exploiting this observation, we use the proposed CD measure within two AL frameworks: (1) a core-set based strategy and (2) a reinforcement learning based policy, for active frame selection. Our extensive empirical evaluation establish state of the art results for active learning on benchmark datasets of Semantic Segmentation, Object Detection and Image Classification. Our ablation studies show clear advantages of using contextual diversity for active learning. The source code and additional results are available at https://github.com/sharat29ag/CDAL.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑