← Search

Arash Mehrjou

12 accepted papers

2025

Deriving Causal Order from Single-Variable Interventions: Guarantees & Algorithm

ICLR 2025poster

Targeted and uniform interventions to a system are crucial for unveiling causal relationships. While several methods have been developed to leverage interventional data for causal structure learning, their practical application in real-world scenarios often remains challenging. Recent benchmark stud…

2023

Diffusion Based Representation Learning

ICML 2023poster

Diffusion-based methods, represented as stochastic differential equations on a continuous-time domain, have recently proven successful as non-adversarial generative models. Training such models relies on denoising score matching, which can be seen as multi-scale denoising autoencoders. Here, we augm…

Cited by 35SourcePDFScholar
2023

DiscoBAX - Discovery of optimal intervention sets in genomic experiment design

ICML 2023poster

The discovery of therapeutics to treat genetically-driven pathologies relies on identifying genes involved in the underlying disease mechanism. Existing approaches search over the billions of potential interventions to maximize the expected influence on the target phenotype. However, to reduce the r…

2022

GalilAI: Out-of-Task Distribution Detection using Causal Active Experimentation for Safe Transfer RL

AISTATS 2022poster

Out-of-distribution (OOD) detection is a well-studied topic in supervised learning. Extending the successes in supervised learning methods to the reinforcement learning (RL) setting, however, is difficult due to the data generating process - RL agents actively query their environment for data and th…

Cited by 3SourcePDFScholar
2022

GeneDisco: A Benchmark for Experimental Design in Drug Discovery

ICLR 2022poster

In vitro cellular experimentation with genetic interventions, using for example CRISPR technologies, is an essential step in early-stage drug discovery and target validation that serves to assess initial hypotheses about causal associations between biological mechanisms and disease pathologies. With…

2021

Causal Curiosity: RL Agents Discovering Self-supervised Experiments for Causal Representation Learning

ICML 2021spotlight

Humans show an innate ability to learn the regularities of the world through interaction. By performing experiments in our environment, we are able to discern the causal factors of variation and infer how they affect the dynamics of our world. Analogously, here we attempt to equip reinforcement lear…

Cited by 81SourcePDFScholar
2020

Counterfactuals uncover the modular structure of deep generative models

ICLR 2020poster

Deep generative models can emulate the perceptual properties of complex image datasets, providing a latent representation of the data. However, manipulating such representation to perform meaningful and controllable transformations in the data space remains challenging without some form of supervisi…

Cited by 120SourceScholar
2019

The Incomplete Rosetta Stone problem: Identifiability results for Multi-view Nonlinear ICA

UAI 2019poster

We consider the problem of recovering a common latent source with independent components from multiple views. This applies to settings in which a variable is measured with multiple experimental modalities, and where the goal is to synthesize the disparate measurements into a single unified represent…

Cited by 101SourcePDFScholar
2018

Tempered Adversarial Networks

ICML 2018oral

Generative adversarial networks (GANs) have been shown to produce realistic samples from high-dimensional distributions, but training them is considered hard. A possible explanation for training instabilities is the inherent imbalance between the networks: While the discriminator is trained directly…