无条件去噪网络能自发提取图像语义特征,无需标签
Unconditional CNN denoisers contain sparse semantic representation of images
- 用纯卷积无条件UNet做去噪,中间层激活通道稀疏
- 通道空间均值构成非线性语义表示,欧氏距离具语义意义
- 自引导重建算法揭示模型隐含语义结构,适合研究生成机制
生成式扩散模型通过神经网络估计得分以学习多样图像数据集的概率密度。尽管其在生成高质量图像方面表现卓越,但底层得分网络的内部机制仍不清晰。本文研究了全卷积无条件UNet中得分估计产生的图像表征。我们发现,UNet中间块将单个图像分解为稀疏激活的通道子集,这些通道的空间平均值可形成对原始干净图像的非线性表示。该表示空间中的欧氏距离具有语义意义,即使训练时未提供任何条件信息。我们提出一种新型随机重建算法,利用该表示自引导无条件模型进行图像合成:基于同一模型提取的表示,重建样本集合中的共性模式揭示了中间块捕捉到的特征。这些结果首次表明,仅通过去噪目标即可无监督地涌现出语义相似性度量。
原文摘要 · Abstract (English)
Generative diffusion models learn probability densities over diverse image datasets by estimating the score with a neural network trained to remove noise. Despite their remarkable success in generating high-quality images, the internal mechanisms of the underlying score networks are not well understood. Here, we examine the image representation that arises from score estimation in a {fully-convolutional unconditional UNet}. We show that the middle block of the UNet decomposes individual images into sparse subsets of active channels, and that the vector of spatial averages of these channels can provide a nonlinear representation of the underlying clean images. Euclidean distances in this representation space are semantically meaningful, even though no conditioning information is provided during training. We develop a novel algorithm for stochastic reconstruction of images conditioned on this representation: The synthesis using the unconditional model is "self-guided" by the representation extracted from that very same model. For a given representation, the common patterns in the set of reconstructed samples reveal the features captured in the middle block of the UNet. Together, these results show, for the first time, that a measure of semantic similarity emerges, unsupervised, solely from the denoising objective.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。