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

分布の曲がり方に合わせて拡散モデルの誘導を安定化

Geometry-Aware Diffusion Guidance via Curvature-Adaptive Tubular Correction

Enze Jiang and Jinwei He and Zheng Ma

この論文をやさしく読む

ひとことで言うと

拡散モデルを目的の画像へ強く誘導すると、学習した分布から外れて品質が落ちることがあります。この研究は、分布の面の曲がり方を見ながら、誘導の方向と大きさを調整します。

何に役立つ?

画像の逆問題や条件付き生成で、誘導を強めたときの不安定さを抑える追加手法として役立つ可能性があります。既存サンプラーに追加学習なしで組み込む設計です。

この研究の面白いところ

面に沿う方向なら安全とみなすだけでなく、曲がった面では接方向の移動でも面から離れることを考慮します。一方、法線方向の移動もすべて捨てず、共通の予算の中で配分します。

どこまで分かった?

理論保証は局所的なものです。実験は指定した画像データ、七つの逆問題、InverseBenchなどの条件下で行われています。最小FIDは比較した手法と試した誘導強度の範囲での結果で、全用途の優位性を保証しません。

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

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

勾配で誘導する拡散サンプラーは、逆問題や条件付き生成に柔軟な事前分布を提供する。しかし強い誘導は、学習したスコアが十分に裏付けられていない領域へサンプリング軌道を動かすことがある。既存の接方向への射影戦略は、等密度面からの1次の逸脱を抑えるものの、有用な可能性がある法線方向の動きを捨て、曲面上の接方向移動が引き起こす2次の逸脱を見落とす。 本研究は、共通の雑音依存の幾何学的予算の中で両方の効果を調整する、追加学習不要のプラグイン、曲率適応型管状補正(CAT)を提案する。CATは誘導勾配を法線成分と接成分に分け、法線方向の変位を1次で、接方向の変位を方向曲率に応じて評価し、1次元の双対方程式から両者の大きさを同時に最適化する。Armijoのバックトラッキングにより、得られた有限ステップを実際の誘導目的関数に照らして調整する。また、行列を明示しない方向微分によって、スコアのヤコビ行列全体を構成せずに済む。 管状近似、補正の一意性、目的関数の十分な減少について、局所的な保証を示す。FFHQとImageNet上の七つの逆問題では、CATは評価対象となったピクセル空間および潜在空間の基盤サンプラーを改善し、とりわけ知覚指標で一貫した向上を示す。InverseBenchでのブラックホール再構成も改善する。また、彩度とコントラストの安定性を保ちつつ、試したすべての分類器不要ガイダンスの強度で、比較手法中の最小FIDを得る。これらの結果は、曲率を考慮した管状制御が、拡散の誘導を安定させる再利用可能な仕組みとなることを支持する。

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

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

更新履歴

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

原文の要旨

Gradient-guided diffusion samplers provide flexible priors for inverse problems and conditional generation, but strong guidance can move the sampling trajectory into regions where the learned score is poorly supported. Existing tangent-projection strategies limit first-order departure from an iso-density surface, yet discard potentially useful normal motion and overlook the second-order departure induced by tangent motion on a curved surface. We introduce curvature-adaptive tubular correction (CAT), a training-free plugin that regulates both effects within a shared, noise-dependent geometric budget. CAT decomposes the guidance gradient into normal and tangent components, charges normal displacement at first order and tangent displacement according to directional curvature, and obtains their jointly optimal magnitudes from a one-dimensional dual equation. Armijo backtracking calibrates the resulting finite step against the actual guidance objective, while matrix-free directional derivatives avoid constructing the full score Jacobian. We establish local guarantees for the tubular approximation, uniqueness of the correction, and sufficient objective decrease. Across seven inverse problems on FFHQ and ImageNet, CAT improves the evaluated pixel- and latent-space host samplers, with particularly consistent gains in perceptual metrics. It also improves black hole reconstruction on InverseBench and yields the lowest FID among the compared methods at every tested classifier-free guidance scale, while maintaining stable saturation and contrast. These results support curvature-aware tubular control as a reusable mechanism for stabilizing diffusion guidance.

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