4 papers
Structure Learning on Clustered Data
Ryan Thompson, Matt P. Wand, Veerabhadran Baladandayuthapani
Recent algorithmic advances have made directed acyclic graph (DAG) structure learning scalable for causal discovery. Yet, the currently available techniques assume a completely hom…
Scalable Subset Selection in Linear Mixed Models
Ryan Thompson, Matt P. Wand, Joanna J. J. Wang
Linear mixed models (LMMs), which incorporate fixed and random effects, are key tools for analyzing heterogeneous data, such as in personalized medicine. Nowadays, this type of dat…
Arrow: A Foundation Model for Causal Discovery
Ryan Thompson, He Zhao, Daniel M. Steinberg +1
We introduce Arrow, a foundation model for zero-shot causal discovery on observational tabular data. Arrow factorizes a directed acyclic graph into an undirected skeleton and a top…
ProDAG: Projected Variational Inference for Directed Acyclic Graphs
Ryan Thompson, Edwin V. Bonilla, Robert Kohn
Directed acyclic graph (DAG) learning is a central task in structure discovery and causal inference. Although the field has witnessed remarkable advances over the past few years, i…