arXiv reaDer
LDC-VAE:変分オートエンコーダへの潜在分布一貫性アプローチ
LDC-VAE: A Latent Distribution Consistency Approach to Variational AutoEncoders
生成モデルの重要な側面として、変分オートエンコーダー(VAE)は多くの研究関心を集め、多くの成功したアプリケーションに到達しました。ただし、証拠の下限(ELBO)を最適化するときに、学習した潜在分布と事前の潜在分布の間の一貫性を実現することは常に課題であり、最終的にデータ生成のパフォーマンスが不十分になります。この論文では、ELBO最適化における事後と事前の潜在分布間のこのような実質的な不一致を回避するために、潜在分布の一貫性アプローチを提案します。この方法を潜在分布整合性VAE(LDC-VAE)と名付けます。この目的は、潜在空間の実際の事後分布をギブス形式として想定し、エンコーダーを使用して近似することで実現します。ただし、このようなギブスの近似の後方分析ソリューションはなく、反復サンプリングベースのMCMCを使用するなど、従来の近似方法には時間がかかります。この問題に対処するために、スタイン変分勾配降下法(SVGD)を使用してギブス後方を近似します。一方、SVGDを使用して、ギブス後方から効率的なサンプルを取得できるサンプラーネットをトレーニングします。人気のある画像生成データセットに関する比較研究は、私たちの方法がVAEのいくつかの強力な改善と同等またはそれ以上のパフォーマンスを達成したことを示しています。
Variational autoencoders (VAEs), as an important aspect of generative models, have received a lot of research interests and reached many successful applications. However, it is always a challenge to achieve the consistency between the learned latent distribution and the prior latent distribution when optimizing the evidence lower bound (ELBO), and finally leads to an unsatisfactory performance in data generation. In this paper, we propose a latent distribution consistency approach to avoid such substantial inconsistency between the posterior and prior latent distributions in ELBO optimizing. We name our method as latent distribution consistency VAE (LDC-VAE). We achieve this purpose by assuming the real posterior distribution in latent space as a Gibbs form, and approximating it by using our encoder. However, there is no analytical solution for such Gibbs posterior in approximation, and traditional approximation ways are time consuming, such as using the iterative sampling-based MCMC. To address this problem, we use the Stein Variational Gradient Descent (SVGD) to approximate the Gibbs posterior. Meanwhile, we use the SVGD to train a sampler net which can obtain efficient samples from the Gibbs posterior. Comparative studies on the popular image generation datasets show that our method has achieved comparable or even better performance than several powerful improvements of VAEs.
updated: Sun May 08 2022 07:47:39 GMT+0000 (UTC)
published: Wed Sep 22 2021 10:34:40 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト