arXiv reaDer
マスクされたベクトルの量子化
Masked Vector Quantization
離散潜在表現を持つ生成モデルは、複雑な高次元データ分布を学習する印象的な能力を最近実証しました。ただし、それらのパフォーマンスは、インスタンスごとのトークンの長いシーケンスと多数のコードブック エントリに依存しているため、サンプリング時間が長くなり、カテゴリ事後分布に適合するためにかなりの計算が必要になります。これらの問題に対処するために、Multiple Hypothese Dropout (MH-Dropout) と呼ばれる確率的勝者総取りトレーニング体制を介してマスク構成を学習することにより、各コード ベクトルの表現能力を向上させる Masked Vector Quantization (MVQ) フレームワークを提案します。 ImageNet 64×64 では、MVQ は既存のベクトル量子化アーキテクチャの FID をインスタンスあたり 2 トークンで最大 68%、5 トークンで最大 57% 削減します。これらの改善は、コードブック エントリが減少するにつれて広がり、推論中のトークン サンプリングで 7 ~ 45 倍のスピードアップが可能になります。追加の利点として、潜在空間が小さいと、複数をスムーズに組み合わせることができる転送可能な視覚的表現を識別する MVQ につながることがわかりました。
Generative models with discrete latent representations have recently demonstrated an impressive ability to learn complex high-dimensional data distributions. However, their performance relies on a long sequence of tokens per instance and a large number of codebook entries, resulting in long sampling times and considerable computation to fit the categorical posterior. To address these issues, we propose the Masked Vector Quantization (MVQ) framework which increases the representational capacity of each code vector by learning mask configurations via a stochastic winner-takes-all training regime called Multiple Hypothese Dropout (MH-Dropout). On ImageNet 64×64, MVQ reduces FID in existing vector quantization architectures by up to 68% at 2 tokens per instance and 57% at 5 tokens. These improvements widen as codebook entries is reduced and allows for 7--45× speed-up in token sampling during inference. As an additional benefit, we find that smaller latent spaces lead to MVQ identifying transferable visual representations where multiple can be smoothly combined.
updated: Mon Jan 16 2023 22:30:53 GMT+0000 (UTC)
published: Mon Jan 16 2023 22:30:53 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アソシエイト