制御されたパス生成のための解析的ブリッジ拡散

arXiv cs.LG / 2026/5/6

📰 ニュースIdeas & Deep AnalysisModels & Research

要点

  • この論文は「LQ-GM-PID」(制約はあるが十分広く、解析的に解けるクラス)を提案し、スコア、中間時刻の分布、プロトコル勾配を閉形式で計算できるため、最適化ループ内ではニューラルネットワークや内部の確率シミュレーションループを不要にしています。
  • 古典的な線形–二次–ガウス(LQG)確率制御を、Path Integral Diffusion(PID)型の輸送問題として再解釈し、終端の状態一致だけでなく、所望の終端確率密度に基づく「パスの形状(path shaping)」へブリッジ拡散を拡張します。
  • LQGの枠組み(線形ダイナミクスと二次コスト)を維持しつつ、終端状態の制御を終端密度の指定に置き換え、初期分布・終端分布の双方をガウス混合(GM)として扱えるようにしています。
  • 2Dの回廊タスク、2Dの複数入口による輸送タスク、さらに高次元(d=32、M=16の終端ガウス混合モード)で検証し、ノートPCで解析的な事前計算がサブ50msであることを示します。
  • LQ-GM-PIDは、ニューラルなブリッジ拡散や生成輸送手法をベンチマークするための、正確に検証可能な解析参照モデルとして位置づけられており、スコア推定やパス形状目的、プロトコル学習を厳密値と比較できます。

Abstract

ほとんどの現代的なブリッジ拡散手法は、補間、Schr"odinger-bridge、または確率的制御の目的を指定して有限時間の輸送を達成し、その後、関連するスコアまたはドリフト場をニューラルネットワークで学習することで実現します。これに対して本研究では、制約されたものの十分に広く、かつ解析的に解けるクラスを同定し、そのクラスでは、スコア、中間分布、プロトコル勾配が、内部の確率的シミュレーション・ループや最適化ループ内のニューラルネットワークなしで、閉形式で利用可能であることを示します。私たちは、古典的な線形--二次--ガウス(LQG)の確率的制御構造を、Path Integral Diffusion(PID)型の輸送問題として再定式化します。古典的なLQG制御では、線形ダイナミクス、ガウス雑音、二次コストがリカッチ方程式と閉形式の最適フィードバックを導きます。LQ-GM-PIDでは、線形--二次の確率的制御の骨格を保持しつつ、終端状態の制御を所与の終端確率密度で置き換え、初期法則と終端法則の両方をガウス混合(GM)として許します。 さらに、LQ-GM-PIDは、ブリッジ拡散を終端ターゲットのマッチングのための道具にとどまらず、経路形成(path shaping)のための道具へと変えます。これは、2次元の回廊タスク、2次元の複数入口の輸送タスク、ならびにd=32かつM=16のガウス混合終端モードに対する高次元のスケーリング研究で示します。いずれも、ノートPC上でサブ50\,msの解析的な事前計算で実行できます。私たちは、LQ-GM-PIDを、最先端のニューラル・ブリッジ拡散および生成的輸送手法に対する、解析的に解ける参照モデルとして位置づけます。すなわち、ニューラル近似、スコア推定、経路形成のための目的関数、そしてプロトコル学習手続きが、厳密な量と比較して検証できる制御された設定です。