![](https://d3ilqtpdwi981i.cloudfront.net/2mP-o8LJe3xotRRrhgTELRYj8Ec=/425x550/smart/https://bepress-attached-resources.s3.amazonaws.com/uploads/e5/43/7c/e5437ca3-1d09-4b83-8002-db8262f252dd/thumbnail_2e9f460b-188e-4cb5-8e48-91a53eb7fb72.jpg)
Context-specific Bayesian networks (i.e. directed acyclic graphs, DAGs) identify context-dependent relationships between variables, but the non-convexity induced by the acyclicity requirement makes it difficult to share information between context-specific estimators (e.g. with graph generator functions). For this reason, existing methods for inferring context-specific Bayesian networks have favored breaking datasets into subsamples, limiting statistical power and resolution, and preventing the use of multidimensional and latent contexts. To overcome this challenge, we propose NOTEARS-optimized Mixtures of Archetypal DAGs (NOTMAD). NOTMAD models context-specific Bayesian networks as the output of a function which learns to mix archetypal networks according to sample context. The archetypal networks are estimated jointly with the context-specific networks and do not require any prior knowledge. We encode the acyclicity constraint as a smooth regularization loss which is back-propagated to the mixing function; in this way, NOTMAD shares information between context-specific acyclic graphs, enabling the estimation of Bayesian network structures and parameters at even single-sample resolution. We demonstrate the utility of NOTMAD and sample-specific network inference through analysis and experiments, including patient-specific gene expression networks which correspond to morphological variation in cancer. © 2021, CC BY-NC-ND.
- Artificial Intelligence (cs.AI),
- Machine Learning (cs.LG),
- Machine Learning (stat.ML)
Preprint: arXiv;
Archived with thanks to arXiv;
Preprint License: CC by NC-ND 4.0;
Uploaded 05 May 2022