arXiv reaDer
変分オートエンコーダの事前トレーニングのための対照学習アプローチ
A Contrastive Learning Approach for Training Variational Autoencoder Priors
変分オートエンコーダー(VAE)は、多くのドメインでアプリケーションを使用する強力な尤度ベースの生成モデルの1つです。ただし、特にサンプルが焼き戻しなしで以前から取得された場合、高品質の画像を生成するのに苦労します。 VAEの生成品質が低いことの1つの説明は、事前の穴の問題です。事前の分布は、事後の総計と一致しません。この不一致のために、事前分布の下で高密度の潜在空間に、エンコードされた画像に対応しない領域が存在します。これらの領域からのサンプルは、破損した画像にデコードされます。この問題に取り組むために、ベースの事前分布と再重み付け係数の積によって定義されるエネルギーベースの事前分布を提案します。これは、ベースを集合体の事後確率に近づけるように設計されています。ノイズコントラスト推定によって再重み付け係数をトレーニングし、それを多くの潜在変数グループを持つ階層VAEに一般化します。私たちの実験では、提案されたノイズ対照事前分布が、MNIST、CIFAR-10、CelebA 64、およびCelebA HQ256データセットで最先端のVAEの生成パフォーマンスを大幅に改善することを確認しています。私たちの方法は単純で、さまざまなVAEに適用して、事前分布の表現度を向上させることができます。
Variational autoencoders (VAEs) are one of the powerful likelihood-based generative models with applications in many domains. However, they struggle to generate high-quality images, especially when samples are obtained from the prior without any tempering. One explanation for VAEs' poor generative quality is the prior hole problem: the prior distribution fails to match the aggregate approximate posterior. Due to this mismatch, there exist areas in the latent space with high density under the prior that do not correspond to any encoded image. Samples from those areas are decoded to corrupted images. To tackle this issue, we propose an energy-based prior defined by the product of a base prior distribution and a reweighting factor, designed to bring the base closer to the aggregate posterior. We train the reweighting factor by noise contrastive estimation, and we generalize it to hierarchical VAEs with many latent variable groups. Our experiments confirm that the proposed noise contrastive priors improve the generative performance of state-of-the-art VAEs by a large margin on the MNIST, CIFAR-10, CelebA 64, and CelebA HQ 256 datasets. Our method is simple and can be applied to a wide variety of VAEs to improve the expressivity of their prior distribution.
updated: Wed Nov 03 2021 20:01:38 GMT+0000 (UTC)
published: Tue Oct 06 2020 17:59:02 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アソシエイト