arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2407.09842cs.CV

消除小样本分割中的特征歧义

Eliminating Feature Ambiguity for Few-Shot Segmentation

Nanyang Technological University(南洋理工大学);University of Cologne(科隆大学);SenseTime Research(商汤科技研究院)

Qianxiong Xu, Guosheng Lin, Chen Change Loy, Cheng Long, Ziyue Li, Rui Zhao

AI总结:

针对小样本分割中交叉注意力机制因深层特征混有背景信息导致前景匹配不佳、支持信息利用不足的问题,提出可即插即用的歧义消除网络AENet,通过挖掘判别性查询前景区域修正歧义特征,显著提升多个基线模型性能。

摘要:

小样本分割(FSS)领域的最新进展采用了查询特征与支持特征之间的逐像素匹配方法,该方法通常基于交叉注意力机制,能够选择性激活与同类支持前景(FG)特征相对应的查询前景特征。然而,由于骨干网络深层的感受野较大,提取出的查询和支持前景特征不可避免地会混入背景(BG)特征,这阻碍了交叉注意力中的前景-前景匹配。因此,查询前景特征仅能与较少的支持前景特征融合,即支持信息未得到充分利用。本文提出了一种名为歧义消除网络(AENet)的新型即插即用模块,可嵌入到任何现有的基于交叉注意力的小样本分割方法中。其核心思路是挖掘具有判别性的查询前景区域,以修正存在歧义的前景特征,提高前景信息的占比,从而抑制混入的背景特征带来的负面影响。通过这种方式,前景-前景匹配自然得到增强。我们将AENet嵌入到CyCTR、SCCAN和HDMNet三个基线模型中进行评估,它们的得分均得到大幅提升,例如SCCAN的1-shot性能在PASCAL-5$^i$和COCO-20$^i$上均可提升3.0%以上。代码可在https://github.com/Sam1224/AENet获取。

英文摘要:

Recent advancements in few-shot segmentation (FSS) have exploited pixel-by-pixel matching between query and support features, typically based on cross attention, which selectively activate query foreground (FG) features that correspond to the same-class support FG features. However, due to the large receptive fields in deep layers of the backbone, the extracted query and support FG features are inevitably mingled with background (BG) features, impeding the FG-FG matching in cross attention. Hence, the query FG features are fused with less support FG features, i.e., the support information is not well utilized. This paper presents a novel plug-in termed ambiguity elimination network (AENet), which can be plugged into any existing cross attention-based FSS methods. The main idea is to mine discriminative query FG regions to rectify the ambiguous FG features, increasing the proportion of FG information, so as to suppress the negative impacts of the doped BG features. In this way, the FG-FG matching is naturally enhanced. We plug AENet into three baselines CyCTR, SCCAN and HDMNet for evaluation, and their scores are improved by large margins, e.g., the 1-shot performance of SCCAN can be improved by 3.0%+ on both PASCAL-5$^i$ and COCO-20$^i$. The code is available at https://github.com/Sam1224/AENet.

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

↑