Timezone: »
Poster
Globally optimal score-based learning of directed acyclic graphs in high-dimensions
Bryon Aragam · Arash Amini · Qing Zhou
Tue Dec 10 10:45 AM -- 12:45 PM (PST) @ East Exhibition Hall B + C #222
We prove that $\Omega(s\log p)$ samples suffice to learn a sparse Gaussian directed acyclic graph (DAG) from data, where $s$ is the maximum Markov blanket size. This improves upon recent results that require $\Omega(s^{4}\log p)$ samples in the equal variance case. To prove this, we analyze a popular score-based estimator that has been the subject of extensive empirical inquiry in recent years and is known to achieve state-of-the-art results. Furthermore, the approach we study does not require strong assumptions such as faithfulness that existing theory for score-based learning crucially relies on. The resulting estimator is based around a difficult nonconvex optimization problem, and its analysis may be of independent interest given recent interest in nonconvex optimization in machine learning. Our analysis overcomes the drawbacks of existing theoretical analyses, which either fail to guarantee structure consistency in high-dimensions (i.e. learning the correct graph with high probability), or rely on restrictive assumptions. In contrast, we give explicit finite-sample bounds that are valid in the important $p\gg n$ regime.
Author Information
Bryon Aragam (University of Chicago)
Arash Amini (UCLA)
Qing Zhou (UCLA)
More from the Same Authors
-
2023 Poster: Uncovering Meanings of Embeddings via Markov Boundaries »
Yibo Jiang · Bryon Aragam · Victor Veitch -
2023 Poster: Learning Linear Causal Representations from Interventions under General Nonlinear Mixing »
Simon Buchholz · Goutham Rajendran · Elan Rosenfeld · Bryon Aragam · Bernhard Schölkopf · Pradeep Ravikumar -
2023 Poster: Learning Latent Causal Graphs with Unknown Interventions »
Yibo Jiang · Bryon Aragam -
2023 Poster: Assumption violations in causal discovery and the robustness of score matching »
Francesco Montagna · Atalanti Mastakouri · Elias Eulig · Nicoletta Noceti · Lorenzo Rosasco · Dominik Janzing · Bryon Aragam · Francesco Locatello -
2023 Poster: Identifying Causal Mechanism Shifts among Nonlinear Additive Noise Models »
Tianyu Chen · Kevin Bello · Bryon Aragam · Pradeep Ravikumar -
2023 Poster: Global Optimality in Bivariate Gradient-based DAG Learning »
Chang Deng · Kevin Bello · Pradeep Ravikumar · Bryon Aragam -
2023 Oral: Learning Linear Causal Representations from Interventions under General Nonlinear Mixing »
Simon Buchholz · Goutham Rajendran · Elan Rosenfeld · Bryon Aragam · Bernhard Schölkopf · Pradeep Ravikumar -
2023 Poster: Fundamental Limits and Tradeoffs in Invariant Representation Learning »
Han Zhao · Chen Dan · Bryon Aragam · Tommi Jaakkola · Geoffrey Gordon · Pradeep Ravikumar -
2022 Spotlight: Identifiability of deep generative models without auxiliary information »
Bohdan Kivva · Goutham Rajendran · Pradeep Ravikumar · Bryon Aragam -
2022 Poster: Target alignment in truncated kernel ridge regression »
Arash Amini · Richard Baumgartner · Dai Feng -
2022 Poster: DAGMA: Learning DAGs via M-matrices and a Log-Determinant Acyclicity Characterization »
Kevin Bello · Bryon Aragam · Pradeep Ravikumar -
2022 Poster: Identifiability of deep generative models without auxiliary information »
Bohdan Kivva · Goutham Rajendran · Pradeep Ravikumar · Bryon Aragam -
2021 Poster: Label consistency in overfitted generalized $k$-means »
Linfan Zhang · Arash Amini -
2021 Poster: Learning latent causal graphs via mixture oracles »
Bohdan Kivva · Goutham Rajendran · Pradeep Ravikumar · Bryon Aragam -
2021 Poster: Structure learning in polynomial time: Greedy algorithms, Bregman information, and exponential families »
Goutham Rajendran · Bohdan Kivva · Ming Gao · Bryon Aragam -
2021 Poster: Efficient Bayesian network structure learning via local Markov boundary search »
Ming Gao · Bryon Aragam -
2020 Poster: The Potts-Ising model for discrete multivariate data »
Zahra Razaee · Arash Amini -
2020 Poster: A polynomial-time algorithm for learning nonparametric causal graphs »
Ming Gao · Yi Ding · Bryon Aragam -
2019 Poster: Learning Sample-Specific Models with Low-Rank Personalized Regression »
Ben Lengerich · Bryon Aragam · Eric Xing -
2018 Poster: The Sample Complexity of Semi-Supervised Learning with Nonparametric Mixture Models »
Chen Dan · Liu Leqi · Bryon Aragam · Pradeep Ravikumar · Eric Xing -
2018 Poster: DAGs with NO TEARS: Continuous Optimization for Structure Learning »
Xun Zheng · Bryon Aragam · Pradeep Ravikumar · Eric Xing -
2018 Spotlight: DAGs with NO TEARS: Continuous Optimization for Structure Learning »
Xun Zheng · Bryon Aragam · Pradeep Ravikumar · Eric Xing -
2017 : Posters »
Reihaneh Rabbany · Tianxi Li · Jacob Carroll · Yin Cheng Ng · Xueyu Mao · Alexandre Hollocou · Jeric Briones · James Atwood · John Santerre · Natalie Klein · Pranamesh Chakraborty · Zahra Razaee · Chandan Singh · Arun Suggala · Beilun Wang · Andrew R. Lawrence · Aditya Grover · FARSHAD HARIRCHI · radhika arava · Qing Zhou · Takatomi Kubo · Josue Orellana · Govinda Kamath · Vivek Kumar Bagaria -
2017 Poster: Variable Importance Using Decision Trees »
Jalil Kazemitabar · Arash Amini · Adam Bloniarz · Ameet S Talwalkar -
2013 Poster: Bayesian inference as iterated random functions with applications to sequential inference in graphical models »
Arash Amini · XuanLong Nguyen -
2013 Spotlight: Bayesian inference as iterated random functions with applications to sequential inference in graphical models »
Arash Amini · XuanLong Nguyen