DCILP: A Distributed Approach for Large-Scale Causal Structure Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Dong, Shuyu, Sebag, Michèle, Uemura, Kento, Fujii, Akito, Chang, Shuang, Koyanagi, Yusuke, Maruhashi, Koji
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916816065921024
author Dong, Shuyu
Sebag, Michèle
Uemura, Kento
Fujii, Akito
Chang, Shuang
Koyanagi, Yusuke
Maruhashi, Koji
author_facet Dong, Shuyu
Sebag, Michèle
Uemura, Kento
Fujii, Akito
Chang, Shuang
Koyanagi, Yusuke
Maruhashi, Koji
contents Causal learning tackles the computationally demanding task of estimating causal graphs. This paper introduces a new divide-and-conquer approach for causal graph learning, called DCILP. In the divide phase, the Markov blanket MB($X_i$) of each variable $X_i$ is identified, and causal learning subproblems associated with each MB($X_i$) are independently addressed in parallel. This approach benefits from a more favorable ratio between the number of data samples and the number of variables considered. In counterpart, it can be adversely affected by the presence of hidden confounders, as variables external to MB($X_i$) might influence those within it. The reconciliation of the local causal graphs generated during the divide phase is a challenging combinatorial optimization problem, especially in large-scale applications. The main novelty of DCILP is an original formulation of this reconciliation as an integer linear programming (ILP) problem, which can be delegated and efficiently handled by an ILP solver. Through experiments on medium to large scale graphs, and comparisons with state-of-the-art methods, DCILP demonstrates significant improvements in terms of computational complexity, while preserving the learning accuracy on real-world problem and suffering at most a slight loss of accuracy on synthetic problems.
format Preprint
id arxiv_https___arxiv_org_abs_2406_10481
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle DCILP: A Distributed Approach for Large-Scale Causal Structure Learning
Dong, Shuyu
Sebag, Michèle
Uemura, Kento
Fujii, Akito
Chang, Shuang
Koyanagi, Yusuke
Maruhashi, Koji
Machine Learning
Optimization and Control
Methodology
Causal learning tackles the computationally demanding task of estimating causal graphs. This paper introduces a new divide-and-conquer approach for causal graph learning, called DCILP. In the divide phase, the Markov blanket MB($X_i$) of each variable $X_i$ is identified, and causal learning subproblems associated with each MB($X_i$) are independently addressed in parallel. This approach benefits from a more favorable ratio between the number of data samples and the number of variables considered. In counterpart, it can be adversely affected by the presence of hidden confounders, as variables external to MB($X_i$) might influence those within it. The reconciliation of the local causal graphs generated during the divide phase is a challenging combinatorial optimization problem, especially in large-scale applications. The main novelty of DCILP is an original formulation of this reconciliation as an integer linear programming (ILP) problem, which can be delegated and efficiently handled by an ILP solver. Through experiments on medium to large scale graphs, and comparisons with state-of-the-art methods, DCILP demonstrates significant improvements in terms of computational complexity, while preserving the learning accuracy on real-world problem and suffering at most a slight loss of accuracy on synthetic problems.
title DCILP: A Distributed Approach for Large-Scale Causal Structure Learning
topic Machine Learning
Optimization and Control
Methodology
url https://arxiv.org/abs/2406.10481