arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2308.05695cs.CV

作为自监督表示学习器的掩码扩散模型

Masked Diffusion as Self-supervised Representation Learner

University of Notre Dame(圣母大学);Leibniz-Institut für Analytische Wissenschaften - ISAS - e.V.(莱布尼茨分析科学研究所(ISAS - e.V.))

Zixuan Pan, Jianxu Chen, Yiyu Shi

AI总结:

该研究拆解扩散模型生成能力与表示学习能力的关联,提出用掩码机制替代传统高斯噪声的掩码扩散模型(MDM)作为自监督表示学习器,在医学与自然图像语义分割尤其是少样本场景下性能超越现有基准。

摘要:

去噪扩散概率模型近期展现出了最优的生成性能,也已被用作强大的像素级表示学习器。本文拆解了扩散模型固有生成能力与表示学习能力之间的相互关系。我们提出掩码扩散模型(MDM),这是一种面向语义分割的可扩展自监督表示学习器,用掩码机制替代了传统扩散模型中常规的加性高斯噪声。我们提出的方法显著超越了现有基准,在医学图像和自然图像的语义分割任务中,尤其是在少样本场景下,均展现出了显著的性能提升。

英文摘要:

Denoising diffusion probabilistic models have recently demonstrated state-of-the-art generative performance and have been used as strong pixel-level representation learners. This paper decomposes the interrelation between the generative capability and representation learning ability inherent in diffusion models. We present the masked diffusion model (MDM), a scalable self-supervised representation learner for semantic segmentation, substituting the conventional additive Gaussian noise of traditional diffusion with a masking mechanism. Our proposed approach convincingly surpasses prior benchmarks, demonstrating remarkable advancements in both medical and natural image semantic segmentation tasks, particularly in few-shot scenarios.

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

↑