LLM2D
实例分割中的通用类发现
Generalized Class Discovery in Instance Segmentation
作者: Cuong Manh Hoang, Yeejin Lee, Byeongkeun Kang
发布日期: 5/12/2025
arXiv ID: oai:arXiv.org:2502.08149v2

摘要

arXiv:2502.08149v2 公告类型:替换-交叉 摘要:本文解决了实例分割中的泛化类发现(GCD)任务。目标是在给定标记和未标记数据的情况下,发现新类别并获得一种能够分割已知类别和新类别实例的模型。由于现实世界中存在大量长尾分布的对象,每个类别的实例分布本质上是不平衡的。为了解决这些不平衡分布的问题,我们提出了一种用于对比学习的实例级别温度分配(ITA)方法和一种用于伪标签的类别级别可靠性标准。ITA方法对属于头部类别的样本放松实例区分,以增强GCD。可靠性标准在使用从GCD获得的伪标签训练实例分割网络时,避免在尾部类别中排除大部分伪标签。此外,我们提出动态调整这些标准,在早期阶段利用多样化样本,而在后期阶段仅依赖可靠的伪标签。我们还引入了高效的软关注模块,用于为GCD编码对象特定的表示。最后,我们通过在两个设置上进行实验对提出的模型进行了评估:COCO_half + LVIS 和 LVIS + Visual Genome。实验结果表明,所提出的方法优于之前的最先进的方法。