arXiv reaDer
可逆的注意を伴う生成フロー
Generative Flows with Invertible Attentions
フローベースの生成モデルは、一連の可逆変換を介してデータの確率密度関数を明示的に学習する優れた能力を示しています。しかし、フローの正規化に対する長期的な依存関係のモデル化は、まだ十分に研究されていません。ギャップを埋めるために、この論文では、生成フロー モデルの 2 種類の可逆的注意メカニズムを紹介します。正確には、無条件および条件付き生成フロー モデルのためのマップベースおよびスケーリングされたドット製品アテンションを提案します。重要なアイデアは、スプリット ベースのアテンション メカニズムを活用して、フロー フィーチャ マップの 2 つのスプリットごとにアテンションの重みと入力表現を学習することです。私たちの方法は、扱いやすいヤコビ行列式を備えた可逆アテンション モジュールを提供し、フローベース モデルの任意の位置でシームレスに統合できるようにします。提案されたアテンション メカニズムは、グローバル データの依存関係をモデル化することができ、より包括的なフロー モデルにつながります。複数の生成タスクの評価は、導入されたアテンション フローのアイデアが効率的なフロー モデルをもたらし、最先端の無条件および条件付き生成フロー メソッドに匹敵することを示しています。
Flow-based generative models have shown excellent ability to explicitly learn the probability density function of data via a sequence of invertible transformations. Yet, modeling long-range dependencies over normalizing flows remains understudied. To fill the gap, in this paper, we introduce two types of invertible attention mechanisms for generative flow models. To be precise, we propose map-based and scaled dot-product attention for unconditional and conditional generative flow models. The key idea is to exploit split-based attention mechanisms to learn the attention weights and input representations on every two splits of flow feature maps. Our method provides invertible attention modules with tractable Jacobian determinants, enabling seamless integration of it at any positions of the flow-based models. The proposed attention mechanism can model the global data dependencies, leading to more comprehensive flow models. Evaluation on multiple generation tasks demonstrates that the introduced attention flow idea results in efficient flow models and compares favorably against the state-of-the-art unconditional and conditional generative flow methods.
updated: Mon Jun 07 2021 20:43:04 GMT+0000 (UTC)
published: Mon Jun 07 2021 20:43:04 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アソシエイト