arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2210.04845cs.CVcs.AI
收藏

FS-DETR:带提示且无需重新训练的小样本检测Transformer

FS-DETR: Few-Shot DEtection TRansformer with prompting and without re-training

Samsung AI Cambridge(三星剑桥人工智能研究院);Technical University of Iasi(雅西技术大学);Queen Mary University of London(伦敦玛丽女王大学)

Adrian Bulat, Ricardo Guerrero, Brais Martinez, Georgios Tzimiropoulos

AI总结:

本文针对小样本目标检测的实用需求,提出基于视觉提示的FS-DETR框架,无需测试阶段微调即可处理任意数量新类别与样本,在主流基准上精度超越现有无微调方法,甚至能匹配并超过当前最先进的微调方法。

摘要:

本文研究小样本目标检测(Few-Shot Object Detection,FSOD),该任务的目标是:给定少量描绘训练阶段未见过的新类别的模板(样本),在一组图像中检测出该类别的所有出现实例。从实用角度出发,FSOD系统必须满足以下核心要求:(a)可直接使用,测试阶段无需任何微调;(b)能够同时处理任意数量的新类别目标,同时支持每个类别对应任意数量的样本;(c)检测精度需与封闭集系统相当。为满足上述(a)-(c)项要求,本文做出如下贡献:我们首次提出了一种简洁却性能强大的、基于视觉提示的小样本检测Transformer(FS-DETR),可同时满足要求(a)和(b)。该系统以DETR框架为基础,通过两个核心思路进行扩展:(1)测试阶段将提供的新类别视觉模板作为视觉提示输入模型;(2)为这些提示“加盖”伪类别嵌入(类似于软提示),该嵌入随后在解码器的输出端被预测。值得注意的是,我们的系统不仅比现有方法更灵活,还朝着满足要求(c)迈进了一步。具体而言,它的精度显著高于所有无需微调的方法,在最成熟的基准测试集(PASCAL VOC与MSCOCO)上,其性能甚至能匹配并超越当前最先进的基于微调的方法。

英文摘要:

This paper is on Few-Shot Object Detection (FSOD), where given a few templates (examples) depicting a novel class (not seen during training), the goal is to detect all of its occurrences within a set of images. From a practical perspective, an FSOD system must fulfil the following desiderata: (a) it must be used as is, without requiring any fine-tuning at test time, (b) it must be able to process an arbitrary number of novel objects concurrently while supporting an arbitrary number of examples from each class and (c) it must achieve accuracy comparable to a closed system. Towards satisfying (a)-(c), in this work, we make the following contributions: We introduce, for the first time, a simple, yet powerful, few-shot detection transformer (FS-DETR) based on visual prompting that can address both desiderata (a) and (b). Our system builds upon the DETR framework, extending it based on two key ideas: (1) feed the provided visual templates of the novel classes as visual prompts during test time, and (2) ``stamp'' these prompts with pseudo-class embeddings (akin to soft prompting), which are then predicted at the output of the decoder. Importantly, we show that our system is not only more flexible than existing methods, but also, it makes a step towards satisfying desideratum (c). Specifically, it is significantly more accurate than all methods that do not require fine-tuning and even matches and outperforms the current state-of-the-art fine-tuning based methods on the most well-established benchmarks (PASCAL VOC & MSCOCO).

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

↑