[論文レビュー] Cost-Optimal Learning of Causal Graphs

Murat Kocaoglu, Alexandros G. Dimakis+1 more|arXiv (Cornell University)|Mar 8, 2017

Bayesian Modeling and Causal Inference参考文献 2被引用数 4

ひとこと要約

本稿では、各変数に異なるコストが設定された介入を用いた因果グラフの学習のためのコスト最適フレームワークを提案する。木構造およびクリークツリーに対して多項式時間アルゴリズムを導入し、一般の弦的グラフに対しては貪欲法を適用するが、区間グラフに対しては改善を加えた。グラフ彩色と分離系を用いて、与えられた制約下で因果構造を一意に同定する総コストを最小化する。

ABSTRACT

We consider the problem of learning a causal graph over a set of variables with interventions. We study the cost-optimal causal graph learning problem: For a given skeleton (undirected version of the causal graph), design the set of interventions with minimum total cost, that can uniquely identify any causal graph with the given skeleton. We show that this problem is solvable in polynomial time. Later, we consider the case when the number of interventions is limited. For this case, we provide polynomial time algorithms when the skeleton is a tree or a clique tree. For a general chordal skeleton, we develop an efficient greedy algorithm, which can be improved when the causal graph skeleton is an interval graph.

研究の動機と目的

• 固定されたスケルトンと変数の介入コストが与えられたもとで、因果グラフを一意に同定するための介入集合を、総コストを最小化するように設計すること。
• 介入回数が制限されている場合の、介入コストと介入回数のトレードオフを扱うこと。
• 木構造、クリークツリー、区間グラフなどの構造的グラフ族におけるコスト最適介入設計のための効率的アルゴリズムを開発すること。
• 変数コストを組み込み、総コストを最小化するように一般化された非適応的介入設計手法を拡張すること。
• コストおよび介入予算制約下での因果グラフ学習の多項式時間解法可能なフレームワークを確立すること。

提案手法

• 総コストを最小化する制約下で、コスト最適介入設計を整数線形計画問題として定式化する。
• 介入集合をグラフの分離系と適切な頂点彩色でモデル化し、各色クラスが同時に介入される変数の集合に対応する。
• 修正された貪欲アルゴリズムを適用：最初の $2^m - \chi$ ステップでは最大重み付き独立集合を用い、残りの頂点に対しては $\chi$ 色を用いて $\chi$-彩色性を保証する。
• 区間グラフの場合、独立集合に基づく彩色の代わりに最大重み付き $k$-彩色部分グラフを用いることで、貪欲法を改善し、この部分問題の多項式時間解法を活用する。
• 弦的グラフにおける最大重み付き独立集合のためのフレンクの線形時間アルゴリズムをサブルーチンとして用いる。
• i.i.d. 指数分布または一様分布のコストを有するランダムに抽出された弦的グラフおよび区間グラフに対して、正規化および性能評価を実施する。

実験結果

リサーチクエスチョン

• RQ1 一般のスケルトンに対して、コスト最適介入設計問題は多項式時間で解けるか？
• RQ2 介入コストが変動する場合、因果グラフを一意に同定するために必要な最小総コストは何か？
• RQ3 介入コストはグラフのスパarsity（スパarsityパラメータ $d$）および許容可能な介入回数にどのように依存するか？
• RQ4 区間グラフのような特定のグラフクラスに対して、貪欲アルゴリズムを改善できるか？
• RQ5 グラフ構造（例：木、弦的グラフ、区間グラフ）は、総介入コストにどのような影響を及けるか？

主な発見

• 一般のスケルトンに対して、コスト最適介入設計問題は多項式時間で解けることが示され、基礎的なアルゴリズム的結果が確立された。
• 木構造およびクリークツリーに対しては、その構造的性質のおかげで多項式時間解法が可能である。
• 修正された彩色戦略を用いた貪欲アルゴリズムにより、弦的グラフに対しても $\chi$-彩色性を保ちながら多項式時間で実行可能であり、$m$ が大きくても問題ない。
• 区間グラフでは、最大重み付き $k$-彩色部分グラフを用いた改善された貪欲アルゴリズムにより、従来の独立集合ベースの彩色よりも優れたコスト性能が達成された。
• 実験の結果、平均的な介入コストはグラフ密度（スパarsityパラメータ $d$）の増加に伴い上昇し、スパースなグラフでは低下することが判明した。これは、介入予算が制限されていても同様に成り立つ。
• スパースなグラフでは正規化コストが低く保たれ、最大3回の介入ですら、提案手法が最小限のコストしか発生させないことが示された。