arXiv論文メモ
新着一覧
cs.LG / cs.AI · 査読状況未確認

離散状態の確率流を学び、一段階でデータを生成する

Discrete Wasserstein Flows for One-Step Generative Modeling

Alessandro Micheli, Andrea Zerio, Samir Bhatt

この論文をやさしく読む

ひとことで言うと

離散的なデータを目標分布へ近づける反復過程を学習時に使い、生成時は一度の処理で出力する方法です。

何に役立つ?

構造を持つ離散データの高速生成へつなげるための基礎的な枠組みになります。大規模な実用データへの拡張は今後の方向として示されています。

この研究の面白いところ

確率分布の理論的な流れと、粒子の跳躍による実現が一致するかを、厳密解を計算できる条件で確かめています。

どこまで分かった?

検証は分布と輸送を厳密計算できる統制設定です。要旨は基本構成の妥当性を示す段階であり、大規模な構造化離散データでの性能実証までは報告していません。

v1のアブストラクトに基づくAI解説。日本語訳とは別に、用途の解釈を含みます。

アブストラクトの日本語訳

有限状態空間上での1段階生成モデリングの新しい枠組みを導入する。ドリフティングを連続領域の外へ拡張するため、離散ワッサースタイン幾何を使い、可逆マルコフ核の遷移上で、目標分布に対するKL勾配流を定義する。この確率流をマルコフ跳躍によって粒子レベルで実現し、そこで得た輸送更新を潜在変数に条件付けた生成器へ学習させる。これにより、反復的な動力学は学習時だけに必要となり、推論は1段階のままにできる。 基礎となる分布と輸送動力学を厳密に計算できる統制された設定で、KLの減衰、粒子動力学と確率流の整合性、予測された数値的スケーリングを確認する。さらに、表現容量が有限のニューラル生成器でも、1段階生成を維持しながら、この厳密な輸送目標を追跡できることを示す。これらの結果は基本構成の妥当性を検証し、Discrete Driftingを構造化された離散データへ拡張する基盤となる。

v1の要旨から自動生成。本文の精読・人による確認は未実施。

初稿
2026-10-01(UTC)
最新改訂
2026-10-01 · v1
査読・掲載
査読状況未確認
arXivで読むPDF

更新履歴

取得できた版を表示。版の更新は査読済みを意味しません。過去版の本文差分は未解析です。

原文の要旨

We introduce a new framework for one-step generative modelling on finite state spaces. To extend drifting beyond continuous domains, we use discrete Wasserstein geometry to define a target-relative KL gradient flow over the transitions of a reversible Markov kernel. We realize this probability flow at the particle level through Markov jumps and amortize the resulting transport updates into a latent-conditioned generator, so that the iterative dynamics are required only during training while inference remains one-step. In a controlled setting where the underlying distributions and transport dynamics can be computed exactly, we verify KL dissipation, consistency between the particle dynamics and the probability flow, and the predicted numerical scaling. We further show that a finite-capacity neural generator can track these exact transport targets while retaining one-step generation. These results validate the basic construction and provide a foundation for scaling Discrete Drifting to structured discrete data.

arXiv ID: 2610.01355 / 要約の誤りについて