← Search

Niki Kilbertus

23 accepted papers

2026

Identifiability Challenges in Sparse Linear Ordinary Differential Equations

ICLR 2026poster

Dynamical systems modeling is a core pillar of scientific inquiry across natural and life sciences. Increasingly, dynamical system models are learned from data, rendering identifiability a paramount concept. For systems that are not identifiable from data, no guarantees can be given about their beha…

Cited by 0SourceScholar
2025

An Analysis of Causal Effect Estimation using Outcome Invariant Data Augmentation

NeurIPS 2025spotlight

The technique of data augmentation (DA) is often used in machine learning for regularization purposes to better generalize under i.i.d. settings. In this work, we present a unifying framework with topics in causal inference to make a case for the use of DA beyond just the i.i.d. setting, but for gen…

Cited by 0SourceScholar
2025

Generative Intervention Models for Causal Perturbation Modeling

ICML 2025poster

We consider the problem of predicting perturbation effects via causal models. In many applications, it is a priori unknown which mechanisms of a system are modified by an external perturbation, even though the features of the perturbation are available. For example, in genomics, some properties of a…

Cited by 1SourcePDFScholar
2025

Learning Representations of Instruments for Partial Identification of Treatment Effects

ICML 2025poster

Reliable estimation of treatment effects from observational data is important in many disciplines such as medicine. However, estimation is challenging when unconfoundedness as a standard assumption in the causal inference literature is violated. In this work, we leverage arbitrary (potentially high-…

2025

Signature Kernel Conditional Independence Tests in Causal Discovery for Stochastic Processes

ICLR 2025spotlight

Inferring the causal structure underlying stochastic dynamical systems from observational data holds great promise in domains ranging from science and health to finance. Such processes can often be accurately modeled via stochastic differential equations (SDEs), which naturally imply causal relation…

Cited by 14SourcePDFScholar
2025

Whole Genome Transformer for Gene Interaction Effects in Microbiome Habitat Specificity

AAAI 2025technical

Leveraging the vast genetic diversity within microbiomes offers unparalleled insights into complex phenotypes, yet the task of accurately predicting and understanding such traits from genomic data remains challenging. We propose a framework taking advantage of existing large models for gene vectoriz…

2024

ODEFormer: Symbolic Regression of Dynamical Systems with Transformers

ICLR 2024spotlight

We introduce ODEFormer, the first transformer able to infer multidimensional ordinary differential equation (ODE) systems in symbolic form from the observation of a single solution trajectory. We perform extensive evaluations on two datasets: (i) the existing ‘Strogatz’ dataset featuring two-dimensi…

2024

Unbalancedness in Neural Monge Maps Improves Unpaired Domain Translation

ICLR 2024poster

In optimal transport (OT), a Monge map is known as a mapping that transports a source distribution to a target distribution in the most cost-efficient way. Recently, multiple neural estimators for Monge maps have been developed and applied in diverse unpaired domain translation tasks, e.g. in single…

2023

Modeling content creator incentives on algorithm-curated platforms

ICLR 2023top-5%

Content creators compete for user attention. Their reach crucially depends on algorithmic choices made by developers on online platforms. To maximize exposure, many creators adapt strategically, as evidenced by examples like the sprawling search engine optimization industry. This begets competition…

Cited by 44SourcePDFScholar
2023

Predicting Ordinary Differential Equations with Transformers

ICML 2023poster

We develop a transformer-based sequence-to-sequence model that recovers scalar ordinary differential equations (ODEs) in symbolic form from irregularly sampled and noisy observations of a single solution trajectory. We demonstrate in extensive empirical evaluations that our model performs better or…

Cited by 16SourcePDFScholar
2023

Sequential Underspecified Instrument Selection for Cause-Effect Estimation

ICML 2023oral

Instrumental variable (IV) methods are used to estimate causal effects in settings with unobserved confounding, where we cannot directly experiment on the treatment variable. Instruments are variables which only affect the outcome indirectly via the treatment variable(s). Most IV applications focus…

2023

Stabilized Neural Differential Equations for Learning Dynamics with Explicit Constraints

NeurIPS 2023poster

Many successful methods to learn dynamical systems from data have recently been introduced. However, ensuring that the inferred dynamics preserve known constraints, such as conservation laws or restrictions on the allowed system states, remains challenging. We propose stabilized neural differential…

2022

Predicting Cellular Responses to Novel Drug Perturbations at a Single-Cell Resolution

NeurIPS 2022accept

Single-cell transcriptomics enabled the study of cellular heterogeneity in response to perturbations at the resolution of individual cells. However, scaling high-throughput screens (HTSs) to measure cellular responses for many drugs remains a challenge due to technical limitations and, more importan…

2022

Sparsity in Continuous-Depth Neural Networks

NeurIPS 2022accept

Neural Ordinary Differential Equations (NODEs) have proven successful in learning dynamical systems in terms of accurately recovering the observed trajectories. While different types of sparsity have been proposed to improve robustness, the generalization properties of NODEs for dynamical systems be…

2021

On Component Interactions in Two-Stage Recommender Systems

NeurIPS 2021poster

Thanks to their scalability, two-stage recommenders are used by many of today's largest online platforms, including YouTube, LinkedIn, and Pinterest. These systems produce recommendations in two steps: (i) multiple nominators—tuned for low prediction latency—preselect a small subset of candidates fr…

Cited by 39SourcePDFScholar
2021

On Disentangled Representations Learned from Correlated Data

ICML 2021oral

The focus of disentanglement approaches has been on identifying independent factors of variation in data. However, the causal variables underlying real-world observations are often not statistically independent. In this work, we bridge the gap to real-world scenarios by analyzing the behavior of the…

2020

A Class of Algorithms for General Instrumental Variable Models

NeurIPS 2020poster

Causal treatment effect estimation is a key problem that arises in a variety of real-world settings, from personalized medicine to governmental policy making. There has been a flurry of recent work in machine learning on estimating causal effects when one has access to an instrument. However, to ach…

2020

Fair Decisions Despite Imperfect Predictions

AISTATS 2020poster

Consequential decisions are increasingly informed by sophisticated data-driven predictive models. However, consistently learning accurate predictive models requires access to ground truth labels. Unfortunately, in practice, labels may only exist conditional on certain decisions—if a loan is denied,…

2019

The Sensitivity of Counterfactual Fairness to Unmeasured Confounding

UAI 2019poster

Causal approaches to fairness have seen substantial recent interest, both from the machine learning community and from wider parties interested in ethical prediction algorithms. In no small part, this has been due to the fact that causal models allow one to simultaneously leverage data and expert kn…

2018

Blind Justice: Fairness with Encrypted Sensitive Attributes

ICML 2018oral

Recent work has explored how to train machine learning models which do not discriminate against any subgroup of the population as determined by sensitive attributes such as gender or race. To avoid disparate treatment, sensitive attributes should not be considered. On the other hand, in order to avo…

2018

Learning Independent Causal Mechanisms

ICML 2018oral

Statistical learning relies upon data sampled from a distribution, and we usually do not care what actually generated it in the first place. From the point of view of causal modeling, the structure of each distribution is induced by physical mechanisms that give rise to dependences between observabl…

Cited by 206SourcePDFScholar
2017

Avoiding Discrimination through Causal Reasoning

NeurIPS 2017poster

Recent work on fairness in machine learning has focused on various statistical discrimination criteria and how they trade off. Most of these criteria are observational: They depend only on the joint distribution of predictor, protected attribute, features, and outcome. While convenient to work with,…

Cited by 792SourcePDFScholar