arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.03311cs.CVcs.AIcs.LG
收藏

HASSOD:层次自适应自监督目标检测

HASSOD: Hierarchical Adaptive Self-Supervised Object Detection

University of Illinois at Urbana-Champaign(伊利诺伊大学厄巴纳-香槟分校);IBM Research(IBM研究院)

Shengcao Cao, Dhiraj Joshi, Liang-Yan Gui, Yu-Xiong Wang

AI总结:

HASSOD提出层次自适应聚类与均值教师框架,实现无监督目标检测及组成理解,在LVIS和SA-1B上显著提升Mask AR,推动自监督检测前沿。

摘要:

人类视觉感知系统在无显式监督的学习以及理解对象的局部到整体组成方面展现出卓越的能力。受这两种能力的启发,我们提出了层次自适应自监督目标检测(HASSOD),一种无需人工监督即可学习检测目标并理解其组成的新方法。HASSOD采用层次自适应聚类策略,基于自监督视觉表示将区域分组为目标掩码,并自适应地确定每幅图像中目标的数量。此外,HASSOD通过分析掩码之间的覆盖关系并构建树结构,识别目标在组成方面的层次级别。这一额外的自监督学习任务带来了检测性能的提升和可解释性的增强。最后,我们摒弃了先前方法中使用的低效的多轮自训练过程,转而采用半监督学习中的均值教师(Mean Teacher)框架,这使得训练过程更加平滑和高效。通过在常见图像数据集上的大量实验,我们证明了HASSOD相对于现有方法的优越性,从而推动了自监督目标检测领域的最新进展。值得注意的是,我们在LVIS上将Mask AR从20.2提升至22.5,在SA-1B上从17.0提升至26.0。项目页面:https://HASSOD-NeurIPS23.github.io。

英文摘要:

The human visual perception system demonstrates exceptional capabilities in learning without explicit supervision and understanding the part-to-whole composition of objects. Drawing inspiration from these two abilities, we propose Hierarchical Adaptive Self-Supervised Object Detection (HASSOD), a novel approach that learns to detect objects and understand their compositions without human supervision. HASSOD employs a hierarchical adaptive clustering strategy to group regions into object masks based on self-supervised visual representations, adaptively determining the number of objects per image. Furthermore, HASSOD identifies the hierarchical levels of objects in terms of composition, by analyzing coverage relations between masks and constructing tree structures. This additional self-supervised learning task leads to improved detection performance and enhanced interpretability. Lastly, we abandon the inefficient multi-round self-training process utilized in prior methods and instead adapt the Mean Teacher framework from semi-supervised learning, which leads to a smoother and more efficient training process. Through extensive experiments on prevalent image datasets, we demonstrate the superiority of HASSOD over existing methods, thereby advancing the state of the art in self-supervised object detection. Notably, we improve Mask AR from 20.2 to 22.5 on LVIS, and from 17.0 to 26.0 on SA-1B. Project page: https://HASSOD-NeurIPS23.github.io.

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

↑