arXiv reaDer
NCP-VAE:ノイズ対照的な事前分布を備えた変分オートエンコーダ
NCP-VAE: Variational Autoencoders with Noise Contrastive Priors
変分オートエンコーダー(VAE)は、さまざまなドメインでアプリケーションを使用する強力な尤度ベースの生成モデルの1つです。ただし、特にサンプルが焼き戻しなしで以前から取得された場合、高品質の画像を生成するのに苦労します。 VAEの生成品質が低いことの1つの説明は、事前の穴の問題です。事前の分布は、後の集計の近似と一致しません。この不一致により、潜在空間には、エンコードされた画像に対応しない事前の密度の高い領域が存在します。これらの領域からのサンプルは、破損した画像にデコードされます。この問題に取り組むために、ベースの事前分布と再重み付け係数の積によって定義されるエネルギーベースの事前分布を提案します。これは、ベースを集合体の後方に近づけるように設計されています。ノイズコントラスト推定によって再重み付け係数をトレーニングし、それを多くの潜在変数グループを持つ階層VAEに一般化します。私たちの実験では、提案されたノイズコントラストの事前設定により、MNIST、CIFAR-10、CelebA 64、およびCelebA HQ256データセットで最先端のVAEの生成パフォーマンスが大幅に向上することが確認されています。
Variational autoencoders (VAEs) are one of the powerful likelihood-based generative models with applications in various 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.
updated: Mon Nov 01 2021 17:11:48 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アソシエイト