入力分布の変化に強いシュレーディンガー橋を学習する
Distributionally Robust Schrödinger Bridge
この論文をやさしく読む
ひとことで言うと
ある分布から別の分布へデータを運ぶモデルを、入力の分布が変わっても目標に近づけるように学習する方法です。想定範囲内の最悪の入力分布まで考えて制御器を作ります。
何に役立つ？
考えられる用途は、入力の雑音や分布変化がある画像変換や確率的輸送です。2次元輸送と画像間変換で入力摂動への頑健性を評価しています。
この研究の面白いところ
初期分布を不利な方向へ動かす更新と、制御器の学習を交互に行います。通常条件での性能を保つだけでなく、分布がずれたときの最悪値を目的に含めています。
どこまで分かった？
頑健性の改善には、基準条件での性能とのトレードオフがあります。固定水準の雑音拡張を上回った結果は、ガウス混合輸送で試した二つの未見雑音水準に関するものです。要旨には性能差の具体値はありません。
v1のアブストラクトに基づくAI解説。日本語訳とは別に、用途の解釈を含みます。
アブストラクトの日本語訳
シュレーディンガー橋（SB）は、指定された初期分布と目標分布の間の確率的な輸送を学習する。テスト時に初期分布が変化すると、学習済みの動力学では目標分布を再現できなくなることがある。本研究では、初期分布の不確かさを考慮する単一の制御器を学習する、分布ロバスト・シュレーディンガー橋（DRSB）を導入する。DRSBの目的関数は、制御エネルギーと、得られる終端分布と目標分布の間のKLペナルティからなる。初期分布が基準分布の周囲の曖昧性集合内で変わるとき、この目的関数の最悪値を最小化する単一の制御器を求める。 この目的関数の厳密な変分定式化を導き、終端コストを固定した部分問題を、確率的最適制御および分布ロバスト最適化と結びつける。この定式化に基づき、敵対的な初期分布の更新、終端の対数密度比の推定、制御器の学習を交互に行うアルゴリズムを構成する。確率的制御の最適性条件を用いて敵対的更新に必要な勾配を近似し、Wasserstein型とSinkhorn型の変種を開発する。 2次元の輸送タスクと画像間変換の実験では、通常のSBと比べて入力の摂動に対する頑健性が改善する一方、基準条件での性能とのトレードオフがあることを示す。ガウス混合分布の輸送では、Sinkhorn DRSBは、評価した未見の二つの雑音水準の両方で、固定水準の雑音を用いたデータ拡張よりも平均スライスWasserstein距離が小さかった。
v1の要旨から自動生成。本文の精読・人による確認は未実施。
- 初稿
- 2026-10-01（UTC）
- 最新改訂
- 2026-10-01 · v1
- 査読・掲載
- 査読状況未確認
更新履歴
- v1 2026-10-01 この版を読む
取得できた版を表示。版の更新は査読済みを意味しません。過去版の本文差分は未解析です。
原文の要旨
Schrödinger bridge (SB) learns stochastic transport between prescribed initial and target distributions. When the initial distribution shifts at test time, the learned dynamics can fail to recover the target distribution. We introduce the Distributionally Robust Schrödinger Bridge (DRSB), which learns a single controller that accounts for uncertainty in the initial distribution. The DRSB objective consists of control energy and a KL penalty between the resulting terminal distribution and the target distribution. DRSB seeks a single controller that minimizes the worst-case value of this objective as the initial distribution varies within an ambiguity set around the nominal distribution. We derive an exact variational formulation of this objective and connect its fixed-terminal-cost subproblem to stochastic optimal control and distributionally robust optimization. This formulation motivates an alternating algorithm that updates the adversarial initial distribution, estimates the terminal log-density ratio, and trains the controller. We develop Wasserstein and Sinkhorn variants using stochastic control optimality conditions to approximate the gradients required for adversarial updates. Experiments on two-dimensional transport tasks and image-to-image translation show improved robustness to input perturbations relative to standard SB, with a tradeoff in nominal performance. On Gaussian mixture transport, Sinkhorn DRSB also achieves lower mean sliced Wasserstein distance than fixed-level noise augmentation at both tested unseen noise levels.
著者のコメント
30 pages, 5 figures
arXiv ID: 2610.02043 ／ 要約の誤りについて