← Search

Jeroen Berrevoets

14 accepted papers

2025

Active Feature Acquisition for Personalised Treatment Assignment

AISTATS 2025poster

Making treatment effect estimation actionable for personalized decision-making requires overcoming the costs and delays of acquiring necessary features. While many machine learning models estimate Conditional Average Treatment Effects (CATE), they mostly assume that _all_ relevant features are readi…

Cited by 0SourceScholar
2025

Differentiable Causal Structure Learning with Identifiability by NOTIME

AISTATS 2025poster

The introduction of the NOTEARS algorithm resulted in a wave of research on differentiable Directed Acyclic Graph (DAG) learning. Differentiable DAG learning transforms the combinatorial problem of identifying the DAG underlying a Structural Causal Model (SCM) into a constrained continuous optimizat…

Cited by 0SourceScholar
2024

DAGnosis: Localized Identification of Data Inconsistencies using Structures

AISTATS 2024poster

Identification and appropriate handling of inconsistencies in data at deployment time is crucial to reliably use machine learning models. While recent data-centric methods are able to identify such inconsistencies with respect to the training set, they suffer from two key limitations: (1) suboptimal…

2024

ODE Discovery for Longitudinal Heterogeneous Treatment Effects Inference

ICLR 2024spotlight

Inferring unbiased treatment effects has received widespread attention in the machine learning community. In recent years, our community has proposed numerous solutions in standard settings, high-dimensional treatment settings, and even longitudinal settings. While very diverse, the solution has mos…

Cited by 7SourcePDFScholar
2023

AllSim: Simulating and Benchmarking Resource Allocation Policies in Multi-User Systems

NeurIPS 2023poster

Numerous real-world systems, ranging from healthcare to energy grids, involve users competing for finite and potentially scarce resources. Designing policies for resource allocation in such real-world systems is challenging for many reasons, including the changing nature of user types and their (pos…

Cited by 5SourcePDFScholar
2023

Differentiable and Transportable Structure Learning

ICML 2023poster

Directed acyclic graphs (DAGs) encode a lot of information about a particular distribution in their structure. However, compute required to infer these structures is typically super-exponential in the number of variables, as inference requires a sweep of a combinatorially large space of potential st…

2023

GOGGLE: Generative Modelling for Tabular Data by Learning Relational Structure

ICLR 2023poster

Deep generative models learn highly complex and non-linear representations to generate realistic synthetic data. While they have achieved notable success in computer vision and natural language processing, similar advances have been less demonstrable in the tabular domain. This is partially because…

2023

Learning Representations without Compositional Assumptions

ICML 2023poster

This paper addresses unsupervised representation learning on tabular data containing multiple views generated by distinct sources of measurement. Traditional methods, which tackle this problem using the multi-view framework, are constrained by predefined assumptions that assume feature sets share th…

Cited by 2SourcePDFScholar
2023

To Impute or not to Impute? Missing Data in Treatment Effect Estimation

AISTATS 2023poster

Missing data is a systemic problem in practical scenarios that causes noise and bias when estimating treatment effects. This makes treatment effect estimation from data with missingness a particularly tricky endeavour. A key reason for this is that standard assumptions on missingness are rendered in…

2022

Identifiable Energy-based Representations: An Application to Estimating Heterogeneous Causal Effects

AISTATS 2022poster

Conditional average treatment effects (CATEs) allow us to understand the effect heterogeneity across a large population of individuals. However, typical CATE learners assume all confounding variables are measured in order for the CATE to be identifiable. This requirement can be satisfied by collecti…

2021

DECAF: Generating Fair Synthetic Data Using Causally-Aware Generative Networks

NeurIPS 2021poster

Machine learning models have been criticized for reflecting unfair biases in the training data. Instead of solving for this by introducing fair learning algorithms directly, we focus on generating fair synthetic data, such that any downstream learner is fair. Generating fair synthetic data from unf…

2021

Learning Queueing Policies for Organ Transplantation Allocation using Interpretable Counterfactual Survival Analysis

ICML 2021spotlight

Organ transplantation is often the last resort for treating end-stage illnesses, but managing transplant wait-lists is challenging because of organ scarcity and the complexity of assessing donor-recipient compatibility. In this paper, we develop a data-driven model for (real-time) organ allocation u…

2020

OrganITE: Optimal transplant donor organ offering using an individual treatment effect

NeurIPS 2020poster

Transplant-organs are a scarce medical resource. The uniqueness of each organ and the patients' heterogeneous responses to the organs present a unique and challenging machine learning problem. In this problem there are two key challenges: (i) assigning each organ "optimally" to a patient in the queu…

Cited by 49SourcePDFScholar