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

基于熵感知的加权通道归一化流用于深度生成模型

Entropy-Informed Weighting Channel Normalizing Flow for Deep Generative Models

School of Mathematics, South China University of Technology, Guangzhou, China(华南理工大学数学学院);Shenzhen International Graduate School, Tsinghua University, Shenzhen, China(清华大学深圳国际研究生院);School of Electronic and Information Engineering, South China University of Technology, Guangzhou, China(华南理工大学电子信息学院);Department of Electrical Engineering, Columbia University, New York, NY USA(哥伦比亚大学电气工程系)

Wei Chen, Shian Du, Shigui Li, Delu Zeng, John Paisley

AI总结:

本文提出了一种基于熵感知的加权通道归一化流,通过引入正则化的Shuffle操作提升深度生成模型的密度估计性能。

摘要:

归一化流(NFs)因其精确的似然估计和高效的采样而在深度生成模型中被广泛应用。然而,它们需要大量的内存,因为潜在空间与输入维度相匹配。多尺度架构通过逐步减少潜在维度同时保持可逆性来解决这一问题。现有的多尺度架构使用简单的静态通道分割,限制了表达能力。为改进这一点,我们引入了一种正则化的、基于特征的Shuffle操作,并将其整合到普通的多尺度架构中。该操作能够自适应地生成通道权重并重新排列潜在变量,在分割前进行洗牌。我们发现这种操作引导变量向熵增大的方向演化,因此我们将带有Shuffle操作的NFs称为熵感知加权通道归一化流(EIW-Flow)。在CIFAR-10、CelebA、ImageNet和LSUN上的广泛实验表明,EIW-Flow在深度生成建模中实现了最先进的密度估计和具有竞争力的样本质量,且计算开销极小。

英文摘要:

Normalizing Flows (NFs) are widely used in deep generative models for their exact likelihood estimation and efficient sampling. However, they require substantial memory since the latent space matches the input dimension. Multi-scale architectures address this by progressively reducing latent dimensions while preserving reversibility. Existing multi-scale architectures use simple, static channel-wise splitting, limiting expressiveness. To improve this, we introduce a regularized, feature-dependent $\mathtt{Shuffle}$ operation and integrate it into vanilla multi-scale architecture. This operation adaptively generates channel-wise weights and shuffles latent variables before splitting them. We observe that such operation guides the variables to evolve in the direction of entropy increase, hence we refer to NFs with the $\mathtt{Shuffle}$ operation as \emph{Entropy-Informed Weighting Channel Normalizing Flow} (EIW-Flow). Extensive experiments on CIFAR-10, CelebA, ImageNet, and LSUN demonstrate that EIW-Flow achieves state-of-the-art density estimation and competitive sample quality for deep generative modeling, with minimal computational overhead.

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

↑