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

学习背景提示以发现开放词汇目标检测的隐式知识

Learning Background Prompts to Discover Implicit Knowledge for Open Vocabulary Object Detection

School of Computer Science and Engineering, Sun Yat-sen University(中山大学计算机科学与工程学院);The University of Hong Kong(香港大学);SECE, Shenzhen Graduate School, Peking University(北京大学深圳研究生院电子与计算机工程学院);Institute of Artificial Intelligence, Beihang University(北京航空航天大学人工智能研究院);Research Institute, Sun Yat-sen University, Shenzhen(中山大学深圳研究院)

Jiaming Li, Jiacheng Zhang, Jichang Li, Ge Li, Si Liu, Liang Lin, Guanbin Li

AI总结:

针对开放词汇目标检测中背景解释和模型过拟合导致背景知识丢失的问题,提出LBP框架,通过学习背景提示并设计三个模块来发现和利用隐式背景知识,在OV-COCO和OV-LVIS上优于现有方法。

摘要:

开放词汇目标检测(OVD)旨在寻求一种最优的目标检测器,能够识别基础类别和新类别中的目标。最近的进展利用知识蒸馏,将预训练的大规模视觉-语言模型中的深刻知识转移到目标检测任务中,显著地将检测器识别更多未知目标类别的强大能力进行了泛化。然而,这些方法在背景解释和模型过拟合方面面临重大挑战,因此经常导致关键背景知识的丢失,从而导致检测器的推理性能次优。为了缓解这些问题,我们提出了一种名为LBP的新型OVD框架,提出通过学习背景提示来利用探索到的隐式背景知识,从而增强基础类别和新类别的检测性能。具体而言,我们设计了三个模块:背景类别特定提示、背景目标发现和推理概率修正,以赋能检测器发现、表示和利用从背景提议中探索到的隐式目标知识。在两个基准数据集OV-COCO和OV-LVIS上的评估表明,我们提出的方法在处理OVD任务方面优于现有的最先进方法。

英文摘要:

Open vocabulary object detection (OVD) aims at seeking an optimal object detector capable of recognizing objects from both base and novel categories. Recent advances leverage knowledge distillation to transfer insightful knowledge from pre-trained large-scale vision-language models to the task of object detection, significantly generalizing the powerful capabilities of the detector to identify more unknown object categories. However, these methods face significant challenges in background interpretation and model overfitting and thus often result in the loss of crucial background knowledge, giving rise to sub-optimal inference performance of the detector. To mitigate these issues, we present a novel OVD framework termed LBP to propose learning background prompts to harness explored implicit background knowledge, thus enhancing the detection performance w.r.t. base and novel categories. Specifically, we devise three modules: Background Category-specific Prompt, Background Object Discovery, and Inference Probability Rectification, to empower the detector to discover, represent, and leverage implicit object knowledge explored from background proposals. Evaluation on two benchmark datasets, OV-COCO and OV-LVIS, demonstrates the superiority of our proposed method over existing state-of-the-art approaches in handling the OVD tasks.

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

↑