用双流适配器提升SAM在伪装目标检测中的表现
Improving SAM for Camouflaged Object Detection via Dual Stream Adapters
- 在SAM图像编码器中加入双流适配器,融合RGB与深度图信息
- 在四个基准上超越原始SAM,达到当前最优水平
- 适合需要高精度伪装目标检测的视觉任务研究者
分割一切模型(SAM)在自然图像上表现出色,但在伪装目标检测(COD)任务中表现不佳。本文提出SAM-COD,针对RGB-D输入实现伪装目标检测。在保持SAM架构不变的前提下,通过在图像编码器中扩展双流适配器,分别学习来自RGB图像和深度图像的互补信息,并微调掩码解码器及其深度副本以实现双流掩码预测。双流适配器以并行方式嵌入图像编码器的注意力模块,用于优化和修正两类图像嵌入。为缓解因双流嵌入不直接交互导致的通道差异,采用双向知识蒸馏,包括模型蒸馏器和模态蒸馏器。此外,通过混合两种图像嵌入,联合学习提示嵌入以更新初始提示,并输入掩码解码器,同步图像嵌入与提示嵌入的一致性。在四个COD基准测试中,SAM-COD在给定微调范式下显著优于SAM,并取得当前最优结果。
原文摘要 · Abstract (English)
Segment anything model (SAM) has shown impressive general-purpose segmentation performance on natural images, but its performance on camouflaged object detection (COD) is unsatisfactory. In this paper, we propose SAM-COD that performs camouflaged object detection for RGB-D inputs. While keeping the SAM architecture intact, dual stream adapters are expanded on the image encoder to learn potential complementary information from RGB images and depth images, and fine-tune the mask decoder and its depth replica to perform dual-stream mask prediction. In practice, the dual stream adapters are embedded into the attention block of the image encoder in a parallel manner to facilitate the refinement and correction of the two types of image embeddings. To mitigate channel discrepancies arising from dual stream embeddings that do not directly interact with each other, we augment the association of dual stream embeddings using bidirectional knowledge distillation including a model distiller and a modal distiller. In addition, to predict the masks for RGB and depth attention maps, we hybridize the two types of image embeddings which are jointly learned with the prompt embeddings to update the initial prompt, and then feed them into the mask decoders to synchronize the consistency of image embeddings and prompt embeddings. Experimental results on four COD benchmarks show that our SAM-COD achieves excellent detection performance gains over SAM and achieves state-of-the-art results with a given fine-tuning paradigm.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。