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