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

AgMTR:用于遥感少样本分割的智能体挖掘Transformer

AgMTR: Agent Mining Transformer for Few-shot Segmentation in Remote Sensing

Aerospace Information Research Institute, Chinese Academy of Sciences(中国科学院空天信息创新研究院);School of Electronic, Electrical and Communication Engineering, University of Chinese Academy of Sciences(中国科学院大学电子电气与通信工程学院);University of Chinese Academy of Sciences(中国科学院大学);Institute of Electronics, Chinese Academy of Sciences(中国科学院电子学研究所);Tsinghua University(清华大学)

Hanbo Bi, Yingchao Feng, Yongqiang Mao, Jianning Pei, Wenhui Diao, Hongqi Wang, Xian Sun

AI总结:

针对遥感场景中像素级少样本分割易产生语义模糊的问题,提出AgMTR,通过挖掘局部感知智能体构建智能体级语义关联,有效提升前景与背景的语义清晰度并达到SOTA性能。

摘要:

少样本分割(FSS)旨在仅利用少量标注样本(即支持图像)来分割查询图像中的目标对象。以往方案利用支持-查询像素对之间的相似性来构建像素级语义关联。然而,在具有极端类内差异和背景杂乱的遥感场景中,这种像素级关联会产生大量不匹配,导致查询前景(FG)和背景(BG)像素之间出现语义模糊。为解决此问题,我们提出了一种新颖的智能体挖掘Transformer(AgMTR),它自适应地挖掘一组局部感知智能体来构建智能体级语义关联。与像素级语义相比,这些智能体配备了局部上下文信息并具有更广的感受野。由此,不同的查询像素可以选择性地聚合不同智能体的细粒度局部语义,从而增强查询FG和BG像素之间的语义清晰度。具体而言,首先提出了智能体学习编码器(ALE),建立最优传输计划,安排不同智能体在不同局部区域下聚合支持语义。然后,为进一步优化智能体,构建了智能体聚合解码器(AAD)和语义对齐解码器(SAD),分别突破有限支持集的限制,从未标注数据源和查询图像本身挖掘有价值的特定类语义。在遥感基准iSAID上的大量实验表明,所提方法达到了最先进的性能。令人惊讶的是,当扩展到更常见的自然场景(即PASCAL-5i和COCO-20i)时,我们的方法仍保持相当强的竞争力。

英文摘要:

Few-shot Segmentation (FSS) aims to segment the interested objects in the query image with just a handful of labeled samples (i.e., support images). Previous schemes would leverage the similarity between support-query pixel pairs to construct the pixel-level semantic correlation. However, in remote sensing scenarios with extreme intra-class variations and cluttered backgrounds, such pixel-level correlations may produce tremendous mismatches, resulting in semantic ambiguity between the query foreground (FG) and background (BG) pixels. To tackle this problem, we propose a novel Agent Mining Transformer (AgMTR), which adaptively mines a set of local-aware agents to construct agent-level semantic correlation. Compared with pixel-level semantics, the given agents are equipped with local-contextual information and possess a broader receptive field. At this point, different query pixels can selectively aggregate the fine-grained local semantics of different agents, thereby enhancing the semantic clarity between query FG and BG pixels. Concretely, the Agent Learning Encoder (ALE) is first proposed to erect the optimal transport plan that arranges different agents to aggregate support semantics under different local regions. Then, for further optimizing the agents, the Agent Aggregation Decoder (AAD) and the Semantic Alignment Decoder (SAD) are constructed to break through the limited support set for mining valuable class-specific semantics from unlabeled data sources and the query image itself, respectively. Extensive experiments on the remote sensing benchmark iSAID indicate that the proposed method achieves state-of-the-art performance. Surprisingly, our method remains quite competitive when extended to more common natural scenarios, i.e., PASCAL-5i and COCO-20i.

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

↑