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

因子グラフでメッセージを直接近似する推論法

Direct Message Approximation (DMA): A Consistency-Based Framework for Tractable Approximate Inference on Factor Graphs

Ralf Herbrich, Rainer Schlosser, Jan Lemcke, Johann Ukrow, Anna Kazachkova, Nicolas Alder, Leonhard Hennicke, Theo Bardey, Nico Grimm, Luca Kleinschmidt, Philipp Kolbe, Cezary Kujath, Johanna Schlimme, Karl Matti Schütz

この論文をやさしく読む

ひとことで言うと

因子グラフの周辺分布ではなく、因子からのメッセージを直接近似する確率推論法。

何に役立つ?

ベイズニューラルネットワークなどで、反復や不適切なメッセージを避けながら不確実性を推定する設計に役立つ。

この研究の面白いところ

任意のグラフに対するKLの保証を示し、積とleaky-ReLU因子の明示的なメッセージへ適用した。

どこまで分かった?

要旨に示す実証は構成したBNNでの予測不確実性など。計算時間や精度の全比較は記載されていない。

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

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

因子グラフでの近似メッセージ伝播には、期待伝播(EP)と変分メッセージ伝播(VMP)という主な二系統がある。両者は各因子の辺での周辺分布を近似するため、反復する巡回型の更新順序を必要とし、精度が負のメッセージを生じる危険がある。またVMPではDiracデルタ因子で点推定へ潰れる。著者らは、周辺分布ではなく因子から変数へ送るメッセージを直接近似するDirect Message Approximation(DMA)を導入する。正規化できる因子については、他の入力メッセージがすべてDiracデルタなら厳密に一致するという整合条件を定め、メッセージの構成に使う。任意のグラフで適切なメッセージを扱い、メッセージのKLから周辺分布のKLを抑える基本定理を証明する。そこから、Dirac入力との整合性、EP型の内部反復が不要であること、精度が負のメッセージがないことという3つの構造的な帰結を得る。さらに、それ自体は適切な確率分布でない積の因子の逆向きメッセージについて、補完的なO(1/r²)の保証を証明する。この因子を閉形式で扱うことは従来困難だった。具体例として積とleaky-ReLUの因子のDMAメッセージを明示的に導き、学習例ごとに前後方向を一往復するだけで、勾配の学習率というハイパーパラメーターを持たないベイズニューラルネットワーク(BNN)の推論アルゴリズムを構成する。モデルが合っていない場合も含め、データの乏しい領域で予測不確実性が広がることを検証し、構造的保証が実際の予測へ反映されると示す。

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

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

更新履歴

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

原文の要旨

Approximate message passing on factor graphs underlies two dominant families of probabilistic inference algorithms: expectation propagation (EP) and variational message passing (VMP). Both methods approximate the marginal at each factor edge, forcing an iterative round-robin schedule, risking negative-precision messages, and, for VMP, collapsing to point estimates at Dirac-delta factors. We introduce Direct Message Approximation (DMA), which approximates factor-to-variable messages directly rather than the marginal. For normalisable factors, we define a consistency condition (requiring exactness when all other incoming messages are Dirac deltas) to guide message construction. We prove a master theorem (proper messages, any graph) bounding marginal KL from message KL, with three structural corollaries: Dirac-input consistency, no EP-style inner-loop iteration, and no negative-precision messages. Further, we prove a complementary $O(1/r^2)$ guarantee for the inherently improper backward message of the product factor, whose closed-form treatment has resisted prior work. As a concrete instantiation, we derive explicit DMA messages for the product and leaky-ReLU factors and assemble a Bayesian neural network (BNN) inference algorithm with one forward/backward sweep per training example and no gradient learning-rate hyperparameter, validating that the structural guarantees translate to predictive uncertainty that widens in data-sparse regions, including under model mismatch.

著者のコメント

Submitted to ICLR 2027

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