arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2511.03219cs.CV
收藏

用于内窥镜图像分割的扩散引导掩码一致配对混合

Diffusion-Guided Mask-Consistent Paired Mixing for Endoscopic Image Segmentation

School of Intelligent Engineering, Sun Yat-sen University (Shenzhen Campus)(中山大学智能工程学院);Department of Otolaryngology, The First Affiliated Hospital of Sun Yat-sen University(中山大学附属第一医院耳鼻喉科);School of Mathematics and Statistics, Xi’an Jiaotong University(西安交通大学数学与统计学院)

Pengyu Jie, Wanquan Liu, Rui He, Yihui Wen, Deyu Meng, Chenqiang Gao

AI总结:

针对内窥镜图像分割中的数据增强问题,提出扩散引导的配对混合方法MCPMix,结合保留标签的混合与扩散合成多样性,并采用RLA自适应调整混合强度与损失权重,在多个数据集上实现了最先进的分割性能与鲁棒泛化能力。

摘要:

密集预测的数据增强通常依赖于样本混合或生成式合成。混合方法提高了鲁棒性,但未对齐的掩码会导致软标签模糊。扩散合成增加了表观多样性,但当作为普通样本训练时,忽略了掩码条件的结构优势,并引入了合成与真实域之间的偏移。我们提出了一种配对的扩散引导范式,融合了两者的优势。对于每张真实图像,在相同掩码下生成合成对应物,并将该对图像作为Mask-Consistent Paired Mixing (MCPMix)的可控输入,该方法仅混合图像外观,而监督始终使用原始硬掩码。这产生了一个连续的中间样本族,在共享几何下平滑地桥接合成与真实外观,在不损害像素级语义的情况下扩大了多样性。为保持学习与真实数据对齐,Real-Anchored Learnable Annealing (RLA)在训练过程中自适应调整混合强度和混合样本的损失权重,逐渐将优化重新锚定到真实数据,并缓解分布偏差。在Kvasir-SEG、PICCOLO、CVC-ClinicDB、一个私有NPC-LES队列以及ISIC 2017数据集上,该方法实现了最先进的分割性能,并在基线上取得了一致提升。结果表明,将保留标签的混合与扩散驱动的多样性相结合,加上自适应重新锚定,能够产生鲁棒且可泛化的内窥镜分割。

英文摘要:

Augmentation for dense prediction typically relies on either sample mixing or generative synthesis. Mixing improves robustness but misaligned masks yield soft label ambiguity. Diffusion synthesis increases apparent diversity but, when trained as common samples, overlooks the structural benefit of mask conditioning and introduces synthetic-real domain shift. We propose a paired, diffusion-guided paradigm that fuses the strengths of both. For each real image, a synthetic counterpart is generated under the same mask and the pair is used as a controllable input for Mask-Consistent Paired Mixing (MCPMix), which mixes only image appearance while supervision always uses the original hard mask. This produces a continuous family of intermediate samples that smoothly bridges synthetic and real appearances under shared geometry, enlarging diversity without compromising pixel-level semantics. To keep learning aligned with real data, Real-Anchored Learnable Annealing (RLA) adaptively adjusts the mixing strength and the loss weight of mixed samples over training, gradually re-anchoring optimization to real data and mitigating distributional bias. Across Kvasir-SEG, PICCOLO, CVC-ClinicDB, a private NPC-LES cohort, and ISIC 2017, the approach achieves state-of-the-art segmentation performance and consistent gains over baselines. The results show that combining label-preserving mixing with diffusion-driven diversity, together with adaptive re-anchoring, yields robust and generalizable endoscopic segmentation.

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

↑