Diffusion models have demonstrated impressive potential in semantic segmentation tasks. However, these models cannot accurately segment hidden polyps with complex structures owing to the absence of multi-scale conditional features for guiding the reverse process of diffusion models. To address this issue, we propose a dynamic multi-scale conditional diffusion model (PolypSegDiff). First, we design a dynamic multi-scale integration module to fuse the noise segmentation mask and the original image, dynamically extract multi-scale conditional features, and strengthen the network’s ability of identifying polyp areas. Second, we design a hierarchical feature enhancement module to extract and combine image features at different levels. This module significantly enriches the semantic diversity of conditional features, enabling the denoising network to more accurately understand the semantic relationships between polyps and the surrounding normal tissues. Experimental results across five publicly available polyp segmentation datasets demonstrate that PolypSegDiff outperforms existing popular methods in segmentation accuracy, achieving outstanding performance and robust generalization.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

PolypSegDiff: Dynamic Multi-scale Conditional Diffusion Model for Polyp Segmentation

  • Xiaogang Du,
  • Yipeng Jiao,
  • Tao Lei,
  • Xuejun Zhang,
  • Yingbo Wang,
  • Asoke K. Nandi

摘要

Diffusion models have demonstrated impressive potential in semantic segmentation tasks. However, these models cannot accurately segment hidden polyps with complex structures owing to the absence of multi-scale conditional features for guiding the reverse process of diffusion models. To address this issue, we propose a dynamic multi-scale conditional diffusion model (PolypSegDiff). First, we design a dynamic multi-scale integration module to fuse the noise segmentation mask and the original image, dynamically extract multi-scale conditional features, and strengthen the network’s ability of identifying polyp areas. Second, we design a hierarchical feature enhancement module to extract and combine image features at different levels. This module significantly enriches the semantic diversity of conditional features, enabling the denoising network to more accurately understand the semantic relationships between polyps and the surrounding normal tissues. Experimental results across five publicly available polyp segmentation datasets demonstrate that PolypSegDiff outperforms existing popular methods in segmentation accuracy, achieving outstanding performance and robust generalization.