arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.04031cs.CVcs.LG

Polyp-DDPM:用于增强分割的基于扩散的语义息肉合成

Polyp-DDPM: Diffusion-Based Semantic Polyp Synthesis for Enhanced Segmentation

National Taiwan University of Science and Technology(台湾科技大学);National Taiwan University College of Medicine(台湾大学医学院)

Zolnamar Dorjsembe, Hsing-Kuo Pao, Furen Xiao

AI总结:

本文提出 Polyp-DDPM,利用分割掩码条件扩散模型生成逼真多样的胃肠道息肉图像,以更低成本增强训练并提升分割性能。

摘要:

本研究介绍了 Polyp-DDPM,这是一种基于扩散的方法,用于在掩码条件下生成逼真的息肉图像,旨在增强胃肠道(GI tract)息肉分割。我们的方法解决了与医学图像相关的数据有限、标注成本高以及隐私问题等挑战。通过将扩散模型以分割掩码(表示异常区域的二值掩码)为条件,Polyp-DDPM 在图像质量和分割性能方面优于最先进的方法:其 Fréchet Inception Distance(FID)得分为 78.47,而其他方法得分高于 83.79;其 Intersection over Union(IoU)为 0.7156,而基线模型生成的合成图像的 IoU 低于 0.6694,真实数据的 IoU 为 0.7067。我们的方法生成高质量、多样化的合成数据集用于训练,从而增强息肉分割模型,使其效果可与真实图像相媲美,并提供更强的数据增强能力以改进分割模型。Polyp-DDPM 的源代码和预训练权重已公开发布于 https://github.com/mobaidoctor/polyp-ddpm。

英文摘要:

This study introduces Polyp-DDPM, a diffusion-based method for generating realistic images of polyps conditioned on masks, aimed at enhancing the segmentation of gastrointestinal (GI) tract polyps. Our approach addresses the challenges of data limitations, high annotation costs, and privacy concerns associated with medical images. By conditioning the diffusion model on segmentation masks-binary masks that represent abnormal areas-Polyp-DDPM outperforms state-of-the-art methods in terms of image quality (achieving a Frechet Inception Distance (FID) score of 78.47, compared to scores above 83.79) and segmentation performance (achieving an Intersection over Union (IoU) of 0.7156, versus less than 0.6694 for synthetic images from baseline models and 0.7067 for real data). Our method generates a high-quality, diverse synthetic dataset for training, thereby enhancing polyp segmentation models to be comparable with real images and offering greater data augmentation capabilities to improve segmentation models. The source code and pretrained weights for Polyp-DDPM are made publicly available at https://github.com/mobaidoctor/polyp-ddpm.

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

↑