arXiv reaDer
Hat EBM を使用したジェネレーター潜在空間からの確率モデルの学習
Learning Probabilistic Models from Generator Latent Spaces with Hat EBM
この作業では、エネルギーベース モデル (EBM) の基礎として任意の発電機ネットワークを使用する方法を提案します。私たちの定式化では、観測された画像は、ジェネレーター ネットワークを通過した観測されていない潜在変数と、ジェネレーター出力と画像多様体の間のギャップにまたがる残差確率変数の合計であると仮定しています。次に、フォワード パスの一部としてジェネレーターを含む EBM を定義できます。これをハット EBM と呼びます。モデルは、観測データの潜在変数を推測したり、ジェネレータ ヤコビ行列式を計算したりせずにトレーニングできます。これにより、任意のタイプの発電機ネットワークの出力分布の明示的な確率モデリングが可能になります。実験では、(1) 128x128 解像度での無条件の ImageNet 合成、(2) 既存のジェネレーターの出力の改良、および (3) 非確率的ジェネレーターを組み込んだ EBM の学習において、提案された方法の強力なパフォーマンスが示されています。結果を再現するためのコードと事前トレーニング済みのモデルは、https://github.com/point0bar1/hat-ebm で入手できます。
This work proposes a method for using any generator network as the foundation of an Energy-Based Model (EBM). Our formulation posits that observed images are the sum of unobserved latent variables passed through the generator network and a residual random variable that spans the gap between the generator output and the image manifold. One can then define an EBM that includes the generator as part of its forward pass, which we call the Hat EBM. The model can be trained without inferring the latent variables of the observed data or calculating the generator Jacobian determinant. This enables explicit probabilistic modeling of the output distribution of any type of generator network. Experiments show strong performance of the proposed method on (1) unconditional ImageNet synthesis at 128x128 resolution, (2) refining the output of existing generators, and (3) learning EBMs that incorporate non-probabilistic generators. Code and pretrained models to reproduce our results are available at https://github.com/point0bar1/hat-ebm.
updated: Sat Oct 29 2022 03:55:34 GMT+0000 (UTC)
published: Sat Oct 29 2022 03:55:34 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アソシエイト