神经损失会改变VAE潜空间的信息量与几何结构。
How Neural Losses Shape VAE Latents

- 用神经损失替代点对点重建,降低潜变量信息量。
- 神经损失使潜空间更均匀,各维度不确定性分布更均衡。
- 适合关注潜空间本质机制的研究者阅读。
现代变分自编码器(VAE)通常不使用标准β-VAE目标所隐含的逐点似然进行训练。实践中,逐点重建常与感知损失和对抗损失结合,但尚不清楚这种组合如何改变模型的潜空间动态。我们证明并实证验证:将神经类损失(如感知损失、对抗损失)加入逐点重建,会减少潜变量中存储的信息量。此外,神经重建损失系统性地改变了潜空间的几何结构:使表示更各向同性,且在各潜维度间更均匀地分布不确定性,形成不同的后验方差分布。这些发现表明,率失真权衡并非理解VAE行为的完整视角,我们提出一种更机制化的分析方法,以研究失真度量的选择如何重塑优化问题。
原文摘要 · Abstract (English)
Modern VAEs are rarely trained with the pointwise likelihood implied by the standard $β$-VAE objective. In practice, pointwise reconstruction is often combined with perceptual and adversarial losses, despite a lack of understanding of how this changes the latent dynamics of the model. We show that the choice of reconstruction loss reshapes the rate-distortion problem itself, altering both the information content and the geometry of the learned latent space in ways that may be invisible from reconstructions alone. First, we prove and verify empirically that augmenting pointwise reconstruction with neural terms, such as perceptual and adversarial objectives, reduces the amount of information stored in the latent representations. Second, we show that neural reconstruction losses systematically change the geometry of the latent space: they make representations more isotropic and distribute uncertainty more evenly across latent dimensions, producing different posterior variance profiles. These findings highlight how the rate-distortion tradeoff is not a comprehensive lens to understand the behavior of VAEs, and we propose a more mechanistic approach to investigate how the choice of a distortion metric reshapes the optimization problem.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。