Towards Adaptive Open-Set Object Detection via Category-Level Collaboration Knowledge Mining
AI 摘要
提出一种基于类别级协作知识挖掘的自适应开放集目标检测方法。
主要贡献
- 构建聚类记忆库,增强类别级知识表示
- 设计源域到新类别的选择度量,初始化新类别分类器
- 设计自适应特征分配策略,缓解源域偏差
方法论
通过类别级协作知识挖掘,利用跨域的类间和类内关系,提升模型在新类别的检测性能。
原文摘要
Existing object detectors often struggle to generalize across domains while adapting to emerging novel categories. Adaptive open-set object detection (AOOD) addresses this challenge by training on base categories in the source domain and adapting to both base and novel categories in the target domain without target annotations. However, current AOOD methods remain limited by weak cross-domain representations, ambiguity among novel categories, and source-domain feature bias. To address these issues, we propose a category-level collaboration knowledge mining strategy that exploits both inter-class and intra-class relationships across domains. Specifically, we construct a clustering-based memory bank to encode class prototypes, auxiliary features, and intra-class disparity information, and iteratively update it via unsupervised clustering to enhance category-level knowledge representation. We further design a base-to-novel selection metric to discover source-domain features related to novel categories and use them to initialize novel-category classifiers. In addition, an adaptive feature assignment strategy transfers the learned category-level knowledge to the target domain and asynchronously updates the memory bank to alleviate source-domain bias. Extensive experiments on multiple benchmarks show that our method consistently surpasses state-of-the-art AOOD methods by 1.1-5.5 mAP.