Learning the structure of Bayesian Networks via the bootstrap

Learning the structure of dependencies among multiple random variables is a problem of considerable theoretical and practical interest. Within the context of Bayesian Networks, a practical and surprisingly successful solution to this learning problem is achieved by adopting score-functions optimisation schema, augmented with multiple restarts to avoid local optima. Yet, the conditions under which such strategies work well are poorly understood, and there are also some intrinsic limitations to learning the directionality of the interaction among the variables. Following an early intuition of Friedman and Koller, we propose to decouple the learning problem into two steps: first, we identify a partial ordering among input variables which constrains the structural learning problem, and then propose an effective bootstrap-based algorithm to simulate augmented data sets, and select the most important dependencies among the variables. By using several synthetic data sets, we show that our algorithm yields better recovery performance than the state of the art, increasing the chances of identifying a globally-optimal solution to the learning problem, and solving also well-known identifiability issues that affect the standard approach. We use our new algorithm to infer statistical dependencies between cancer driver somatic mutations detected by high-throughput genome sequencing data of multiple colorectal cancer patients. In this way, we also show how the proposed methods can shade new insights about cancer initiation, and progression. Code: https://github.com/caravagn/Bootstrap-based-Learning

[1]  K. Polyak,et al.  Tumor heterogeneity: causes and consequences. , 2010, Biochimica et biophysica acta.

[2]  David Maxwell Chickering,et al.  Learning Equivalence Classes of Bayesian Network Structures , 1996, UAI.

[3]  P. Nowell The clonal evolution of tumor cell populations. , 1976, Science.

[4]  Daphne Koller,et al.  Ordering-Based Search: A Simple and Effective Algorithm for Learning Bayesian Networks , 2005, UAI.

[5]  James Cussens,et al.  Bayesian Network Structure Learning with Integer Programming: Polytopes, Facets and Complexity , 2017, J. Artif. Intell. Res..

[6]  H. Akaike,et al.  Information Theory and an Extension of the Maximum Likelihood Principle , 1973 .

[7]  Marco Scutari,et al.  Learning Bayesian Networks with the bnlearn R Package , 2009, 0908.3817.

[8]  B. Vogelstein,et al.  A genetic model for colorectal tumorigenesis , 1990, Cell.

[9]  Judea Pearl,et al.  A Theory of Inferred Causation , 1991, KR.

[10]  Steven J. M. Jones,et al.  Comprehensive molecular characterization of human colon and rectal cancer , 2012, Nature.

[11]  Mikko Koivisto,et al.  Exact Bayesian Structure Discovery in Bayesian Networks , 2004, J. Mach. Learn. Res..

[12]  Jeffrey S. Morris,et al.  The Consensus Molecular Subtypes of Colorectal Cancer , 2015, Nature Medicine.

[13]  Varda Rotter,et al.  Mutations in the p53 Tumor Suppressor Gene: Important Milestones at the Various Steps of Tumorigenesis. , 2011, Genes & cancer.

[14]  David Maxwell Chickering,et al.  Large-Sample Learning of Bayesian Networks is NP-Hard , 2002, J. Mach. Learn. Res..

[15]  Gregory F. Cooper,et al.  The ALARM Monitoring System: A Case Study with two Probabilistic Inference Techniques for Belief Networks , 1989, AIME.

[16]  Sach Mukherjee,et al.  A Gibbs Sampler for Learning DAGs , 2016, J. Mach. Learn. Res..

[17]  Liviu Iftode,et al.  Finding hierarchy in directed online social networks , 2011, WWW.

[18]  Fabio Stella,et al.  A survey on Bayesian network structure learning from data , 2019, Progress in Artificial Intelligence.

[19]  D. Haughton On the Choice of a Model to Fit Data from an Exponential Family , 1988 .

[20]  Richard M. Karp,et al.  Reducibility Among Combinatorial Problems , 1972, 50 Years of Integer Programming.

[21]  Kevin P. Murphy,et al.  Exact Bayesian structure learning from uncertain interventions , 2007, AISTATS.

[22]  James Cussens,et al.  Bayesian network learning with cutting planes , 2011, UAI.

[23]  Jose Miguel Puerta,et al.  Learning Bayesian networks by hill climbing: efficient methods based on progressive restriction of the neighborhood , 2010, Data Mining and Knowledge Discovery.

[24]  David Heckerman,et al.  Learning Gaussian Networks , 1994, UAI.

[25]  K. Kinzler,et al.  Cancer genes and the pathways they control , 2004, Nature Medicine.

[26]  Daniele Ramazzotti,et al.  Modeling Cumulative Biological Phenomena with Suppes-Bayes Causal Networks , 2016, bioRxiv.

[27]  José Manuel Gutiérrez,et al.  Who learns better Bayesian network structures: Accuracy and speed of structure learning algorithms , 2018, Int. J. Approx. Reason..

[28]  R. W. Robinson Counting unlabeled acyclic digraphs , 1977 .

[29]  Nikolaj Tatti,et al.  Hierarchies in Directed Networks , 2015, 2015 IEEE International Conference on Data Mining.

[30]  Marco Grzegorczyk,et al.  Improving the structure MCMC sampler for Bayesian networks by introducing a new edge reversal move , 2008, Machine Learning.

[31]  G. Schwarz Estimating the Dimension of a Model , 1978 .

[32]  Nir Friedman,et al.  Data Analysis with Bayesian Networks: A Bootstrap Approach , 1999, UAI.

[33]  M. Kenward,et al.  An Introduction to the Bootstrap , 2007 .

[34]  Nir Friedman,et al.  Probabilistic Graphical Models - Principles and Techniques , 2009 .

[35]  C. Swanton Intratumor heterogeneity: evolution through space and time. , 2012, Cancer research.

[36]  Giancarlo Mauri,et al.  Algorithmic methods to infer the evolutionary trajectories in cancer progression , 2015, Proceedings of the National Academy of Sciences.

[37]  M. Nowak,et al.  Dynamics of cancer progression , 2004, Nature Reviews Cancer.

[38]  Giusi Moffa,et al.  Partition MCMC for Inference on Acyclic Digraphs , 2015, 1504.05006.

[39]  Xin-She Yang,et al.  Introduction to Algorithms , 2021, Nature-Inspired Optimization Algorithms.

[40]  F. Markowetz,et al.  Cancer Evolution: Mathematical Models and Computational Inference , 2014, Systematic biology.

[41]  Gregory F. Cooper,et al.  A Bayesian method for the induction of probabilistic networks from data , 1992, Machine-mediated learning.

[42]  Nir Friedman,et al.  Being Bayesian About Network Structure. A Bayesian Approach to Structure Discovery in Bayesian Networks , 2004, Machine Learning.

[43]  Judea Pearl,et al.  Probabilistic reasoning in intelligent systems - networks of plausible inference , 1991, Morgan Kaufmann series in representation and reasoning.

[44]  Giancarlo Mauri,et al.  CAPRI: Efficient Inference of Cancer Progression Models from Cross-sectional Data , 2014, bioRxiv.

[45]  A. Foran,et al.  Quicksort , 1962, Comput. J..