arXiv:2412.10176cs.CV2024-12中稿 · AAAI被引 5

通过联合监督提升未知物体检测中的通用表征能力。

UN-DETR: Promoting Objectness Learning via Joint Supervision for Unknown Object Detection

  • 设计联合监督机制,融合位置与类别潜空间信息增强对象性学习
  • 在多个基准上达到领先性能,显著提升未知物体识别率
  • 适合关注开放世界目标检测与鲁棒表征学习的研究者

未知物体检测(UOD)旨在识别未见类别物体,突破传统检测依赖封闭世界假设的局限。其核心在于学习跨已知与未知类别的通用表征——即不依赖类别的对象性,以区分物体与背景。然而,现有方法将对象性学习与定位或分类信息分离,导致性能不佳。为此,本文提出基于Transformer的UOD框架UN-DETR,引入实例存在得分(IPS)表示物体存在的概率。为实现信息互补,IPS采用联合监督策略,整合来自位置与类别潜空间的通用对象性特征作为监督信号。为增强学习效果,提出一对多分配策略引入更多监督,并设计无偏查询选择以提供优质解码器初始查询向量。此外,提出基于IPS的后处理策略,用于过滤冗余框并修正已知与未知物体的分类预测。最后,通过无监督预训练获得对象性先验。UN-DETR在多个UOD及已知检测基准上全面评估,表现优异,达到当前最优水平。

原文摘要 · Abstract (English)

Unknown Object Detection (UOD) aims to identify objects of unseen categories, differing from the traditional detection paradigm limited by the closed-world assumption. A key component of UOD is learning a generalized representation, i.e. objectness for both known and unknown categories to distinguish and localize objects from the background in a class-agnostic manner. However, previous methods obtain supervision signals for learning objectness in isolation from either localization or classification information, leading to poor performance for UOD. To address this issue, we propose a transformer-based UOD framework, UN-DETR. Based on this, we craft Instance Presence Score (IPS) to represent the probability of an object's presence. For the purpose of information complementarity, IPS employs a strategy of joint supervised learning, integrating attributes representing general objectness from the positional and the categorical latent space as supervision signals. To enhance IPS learning, we introduce a one-to-many assignment strategy to incorporate more supervision. Then, we propose Unbiased Query Selection to provide premium initial query vectors for the decoder. Additionally, we propose an IPS-guided post-process strategy to filter redundant boxes and correct classification predictions for known and unknown objects. Finally, we pretrain the entire UN-DETR in an unsupervised manner, in order to obtain objectness prior. Our UN-DETR is comprehensively evaluated on multiple UOD and known detection benchmarks, demonstrating its effectiveness and achieving state-of-the-art performance.

未知检测目标检测自监督Transformer

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。