arXiv:2506.10712cs.CV2025-06被引 4

用不确定性掩码引导扩散模型,精准优化伪装目标检测结果。

Uncertainty-Masked Bernoulli Diffusion for Camouflaged Object Detection Refinement

  • 基于不确定性掩码的伯努利扩散,仅对分割差区域进行生成式修复。
  • 在多个数据集上实现MAE降低5.5%、加权F-measure提升3.2%。
  • 可无缝集成现有检测模型,适合需要高精度分割的场景。

伪装目标检测(COD)因目标与背景视觉差异微弱而面临挑战。尽管已有方法取得进展,但后处理优化仍待深入。为此,我们提出首个专为COD设计的生成式精修框架——不确定性掩码伯努利扩散(UMBD)。该模型引入不确定性引导的掩码机制,仅对分割质量差的残差区域应用伯努利扩散,实现精准修复同时保留正确分割部分。为此,设计了混合不确定性量化网络(HUQNet),采用多分支结构融合多源不确定性信息,提升估计精度,实现生成采样过程中的自适应引导。所提UMBD框架可无缝集成于多种基于编码器-解码器的COD模型,结合其判别能力与扩散模型的生成优势。大量实验表明,该框架在多个COD基准上均带来稳定提升,平均实现MAE下降5.5%、加权F-measure提升3.2%,计算开销极小。代码将公开。

原文摘要 · Abstract (English)

Camouflaged Object Detection (COD) presents inherent challenges due to the subtle visual differences between targets and their backgrounds. While existing methods have made notable progress, there remains significant potential for post-processing refinement that has yet to be fully explored. To address this limitation, we propose the Uncertainty-Masked Bernoulli Diffusion (UMBD) model, the first generative refinement framework specifically designed for COD. UMBD introduces an uncertainty-guided masking mechanism that selectively applies Bernoulli diffusion to residual regions with poor segmentation quality, enabling targeted refinement while preserving correctly segmented areas. To support this process, we design the Hybrid Uncertainty Quantification Network (HUQNet), which employs a multi-branch architecture and fuses uncertainty from multiple sources to improve estimation accuracy. This enables adaptive guidance during the generative sampling process. The proposed UMBD framework can be seamlessly integrated with a wide range of existing Encoder-Decoder-based COD models, combining their discriminative capabilities with the generative advantages of diffusion-based refinement. Extensive experiments across multiple COD benchmarks demonstrate consistent performance improvements, achieving average gains of 5.5% in MAE and 3.2% in weighted F-measure with only modest computational overhead. Code will be released.

伪装检测扩散模型生成精修

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。