用多提示机制提升分割模型对隐蔽目标的检测能力
COMPrompter: reconceptualized segment anything model with multiprompt network for camouflaged object detection
- 引入边界梯度掩码作为新提示,增强边缘感知
- 通过框与边界提示互导,提升特征提取精度,使平均性能领先2.2%
- 适合需要高精度隐蔽目标分割的医学图像分析场景
我们重新思考了通用分割模型SAM,并提出一种新型多提示网络COMPrompter,用于隐蔽目标检测(COD)。SAM具备零样本泛化能力,为COD提供了理想框架。本文将SAM的单提示策略升级为多提示策略:设计边缘梯度提取模块,生成包含隐蔽目标边界梯度信息的掩码,作为新的边界提示;构建框-边界互引导模块,通过二者协同实现更精准、全面的特征提取;同时利用离散小波变换从图像嵌入中提取高频特征,作为多提示系统的补充。实验表明,COMPrompter在COD10K等基准上平均性能超越当前最优模型2.2%,在息肉分割任务中也优于顶级方法。代码将开源。
原文摘要 · Abstract (English)
We rethink the segment anything model (SAM) and propose a novel multiprompt network called COMPrompter for camouflaged object detection (COD). SAM has zero-shot generalization ability beyond other models and can provide an ideal framework for COD. Our network aims to enhance the single prompt strategy in SAM to a multiprompt strategy. To achieve this, we propose an edge gradient extraction module, which generates a mask containing gradient information regarding the boundaries of camouflaged objects. This gradient mask is then used as a novel boundary prompt, enhancing the segmentation process. Thereafter, we design a box-boundary mutual guidance module, which fosters more precise and comprehensive feature extraction via mutual guidance between a boundary prompt and a box prompt. This collaboration enhances the model's ability to accurately detect camouflaged objects. Moreover, we employ the discrete wavelet transform to extract high-frequency features from image embeddings. The high-frequency features serve as a supplementary component to the multiprompt system. Finally, our COMPrompter guides the network to achieve enhanced segmentation results, thereby advancing the development of SAM in terms of COD. Experimental results across COD benchmarks demonstrate that COMPrompter achieves a cutting-edge performance, surpassing the current leading model by an average positive metric of 2.2% in COD10K. In the specific application of COD, the experimental results in polyp segmentation show that our model is superior to top-tier methods as well. The code will be made available at https://github.com/guobaoxiao/COMPrompter.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。