arXiv:2511.12633cs.CV2025-11被引 4

提出新方法提升视觉Transformer自编码器生成质量与收敛速度。

Denoising Vision Transformer Autoencoder with Spectral Self-Regularization

  • 通过频谱自正则化抑制高维隐变量中的高频噪声。
  • 在ImageNet上实现rFID=0.28、PSNR=27.26的重建质量,生成质量gFID=1.82。
  • 无需外部视觉模型,使扩散模型收敛速度提升约2倍。

变分自编码器(VAE)通常将图像编码到紧凑的隐空间,降低计算成本,但带来优化困境:更高维度的隐空间虽能提升重建保真度,却常损害生成性能。现有方法尝试利用外部视觉基础模型(VFMs)对高维隐空间进行正则化,但高维隐变量如何影响生成模型优化仍不明确。我们首次揭示,高维隐空间中冗余的高频成分会阻碍扩散模型训练收敛,从而降低生成质量。为此,我们提出频谱自正则化策略,在保留重建质量的同时抑制冗余高频噪声。所提出的去噪-VAE是一种基于ViT的自编码器,不依赖外部模型,生成更干净、低噪声的隐变量,显著提升生成质量和优化收敛速度。我们进一步引入频谱对齐策略,促进基于Denoising-VAE的生成模型优化。完整方法使扩散模型收敛速度比SD-VAE快约2倍,同时在ImageNet 256×256基准上达到最先进的重建质量(rFID = 0.28,PSNR = 27.26)和具有竞争力的生成性能(gFID = 1.82)。

原文摘要 · Abstract (English)

Variational autoencoders (VAEs) typically encode images into a compact latent space, reducing computational cost but introducing an optimization dilemma: a higher-dimensional latent space improves reconstruction fidelity but often hampers generative performance. Recent methods attempt to address this dilemma by regularizing high-dimensional latent spaces using external vision foundation models (VFMs). However, it remains unclear how high-dimensional VAE latents affect the optimization of generative models. To our knowledge, our analysis is the first to reveal that redundant high-frequency components in high-dimensional latent spaces hinder the training convergence of diffusion models and, consequently, degrade generation quality. To alleviate this problem, we propose a spectral self-regularization strategy to suppress redundant high-frequency noise while simultaneously preserving reconstruction quality. The resulting Denoising-VAE, a ViT-based autoencoder that does not rely on VFMs, produces cleaner, lower-noise latents, leading to improved generative quality and faster optimization convergence. We further introduce a spectral alignment strategy to facilitate the optimization of Denoising-VAE-based generative models. Our complete method enables diffusion models to converge approximately 2$\times$ faster than with SD-VAE, while achieving state-of-the-art reconstruction quality (rFID = 0.28, PSNR = 27.26) and competitive generation performance (gFID = 1.82) on the ImageNet 256$\times$256 benchmark.

自编码器扩散模型频谱正则化视觉Transformer

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。