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

神奇动物在哪里:用 Dual SAM 分割任意海洋动物

Fantastic Animals and Where to Find Them: Segment Any Marine Animal with Dual SAM

School of Future Technology(未来技术学院);School of Artificial Intelligence(人工智能学院);Dalian University of Technology(大连理工大学)

Pingping Zhang, Tianyu Yan, Yang Liu, Huchuan Lu

AI总结:

针对海洋动物分割中长程特征与像素连通性不足、SAM 缺乏水下先验的问题,提出 Dual-SAM 框架,结合多层耦合提示、扩张融合注意力和十字交叉连通性预测,在五个 MAS 数据集上取得最优性能。

摘要:

作为水下智能的重要支柱,海洋动物分割(Marine Animal Segmentation, MAS)涉及在海洋环境中分割动物。以往的方法不擅长提取长程上下文特征,并且忽视了离散像素之间的连通性。近年来,Segment Anything Model(SAM)为通用分割任务提供了一个通用框架。遗憾的是,由于 SAM 是用自然图像训练的,它没有获得来自海洋图像的先验知识。此外,SAM 的单位置提示对于先验引导非常不足。为了解决这些问题,我们提出了一种新颖的特征学习框架,称为 Dual-SAM,用于高性能 MAS。为此,我们首先引入了一种具有 SAM 范式的双结构,以增强海洋图像的特征学习。然后,我们提出了一种多层耦合提示(Multi-level Coupled Prompt, MCP)策略,以注入全面的水下先验信息,并通过适配器增强 SAM 编码器的多层特征。随后,我们设计了一个扩张融合注意力模块(Dilated Fusion Attention Module, DFAM),以逐步整合来自 SAM 编码器的多层特征。最后,我们不是直接预测海洋动物的掩码,而是提出了一种十字交叉连通性预测(Criss-Cross Connectivity Prediction, C$^3$P)范式,以捕获离散像素之间的相互连通性。借助双解码器,它生成伪标签并实现互补特征表示的相互监督,从而相比以往技术取得了显著改进。大量实验验证了我们提出的方法在五个广泛使用的 MAS 数据集上达到了最先进的性能。代码可在 https://github.com/Drchip61/Dual_SAM 获取。

英文摘要:

As an important pillar of underwater intelligence, Marine Animal Segmentation (MAS) involves segmenting animals within marine environments. Previous methods don't excel in extracting long-range contextual features and overlook the connectivity between discrete pixels. Recently, Segment Anything Model (SAM) offers a universal framework for general segmentation tasks. Unfortunately, trained with natural images, SAM does not obtain the prior knowledge from marine images. In addition, the single-position prompt of SAM is very insufficient for prior guidance. To address these issues, we propose a novel feature learning framework, named Dual-SAM for high-performance MAS. To this end, we first introduce a dual structure with SAM's paradigm to enhance feature learning of marine images. Then, we propose a Multi-level Coupled Prompt (MCP) strategy to instruct comprehensive underwater prior information, and enhance the multi-level features of SAM's encoder with adapters. Subsequently, we design a Dilated Fusion Attention Module (DFAM) to progressively integrate multi-level features from SAM's encoder. Finally, instead of directly predicting the masks of marine animals, we propose a Criss-Cross Connectivity Prediction (C$^3$P) paradigm to capture the inter-connectivity between discrete pixels. With dual decoders, it generates pseudo-labels and achieves mutual supervision for complementary feature representations, resulting in considerable improvements over previous techniques. Extensive experiments verify that our proposed method achieves state-of-the-art performances on five widely-used MAS datasets. The code is available at https://github.com/Drchip61/Dual_SAM.

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

↑