← Search

Kyle Cranmer

9 accepted papers

2021

Cluster Trellis: Data Structures & Algorithms for Exact Inference in Hierarchical Clustering

AISTATS 2021poster

Hierarchical clustering is a fundamental task often used to discover meaningful structures in data. Due to the combinatorial number of possible hierarchical clusterings, approximate algorithms are typically used for inference. In contrast to existing methods, we present novel dynamic-programming alg…

2021

Exact and approximate hierarchical clustering using A*

UAI 2021poster

Hierarchical clustering is a critical task in numerous domains. Many approaches are based on heuristics and the properties of the resulting clusterings are studied post hoc. However, in several applications, there is a natural cost function that can be used to characterize the quality of the cluster…

Cited by 5SourcePDFScholar
2020

Discovering Symbolic Models from Deep Learning with Inductive Biases

NeurIPS 2020poster

We develop a general approach to distill symbolic representations of a learned deep model by introducing strong inductive biases. We focus on Graph Neural Networks (GNNs). The technique works as follows: we first encourage sparse latent representations when we train a GNN in a supervised setting, th…

2020

Normalizing Flows on Tori and Spheres

ICML 2020poster

Normalizing flows are a powerful tool for building expressive distributions in high dimensions. So far, most of the literature has concentrated on learning flows on Euclidean spaces. Some problems however, such as those involving angles, are defined on spaces with more complex geometries, such as to…

Cited by 181SourcePDFScholar
2020

Set2Graph: Learning Graphs From Sets

NeurIPS 2020poster

Many problems in machine learning (ML) can be cast as learning functions from sets to graphs, or more generally to hypergraphs; in short, Set2Graph functions. Examples include clustering, learning vertex and edge features on graphs, and learning features on triplets in a collection.

2019

Adversarial Variational Optimization of Non-Differentiable Simulators

AISTATS 2019poster

Complex computer simulators are increasingly used across fields of science as generative models tying parameters of an underlying theory to experimental observations. Inference in this setup is often difficult, as simulators rarely admit a tractable density or likelihood function. We introduce Adver…

Cited by 76SourcePDFScholar
2019

Efficient Probabilistic Inference in the Quest for Physics Beyond the Standard Model

NeurIPS 2019poster

We present a novel probabilistic programming framework that couples directly to existing large-scale simulators through a cross-platform probabilistic execution protocol, which allows general-purpose inference engines to record and control random number draws within simulators in a language-agnostic…