English
Related papers

Related papers: Identifiable Energy-based Representations: An Appl…

200 papers

Recently, conditional average treatment effect (CATE) estimation has been attracting much attention due to its importance in various fields such as statistics, social and biomedical sciences. This study proposes a partially linear…

Methodology · Statistics 2022-01-31 Shunsuke Horii

This study investigates the estimation and the statistical inference about Conditional Average Treatment Effects (CATEs), which have garnered attention as a metric representing individualized causal effects. In our data-generating process,…

Methodology · Statistics 2024-03-07 Masahiro Kato

Estimating the conditional average treatment effect (CATE) from observational data is relevant for many applications such as personalized medicine. Here, we focus on the widespread setting where the observational data come from multiple…

Machine Learning · Computer Science 2024-06-05 Jonas Schweisthal , Dennis Frauen , Mihaela van der Schaar , Stefan Feuerriegel

We propose a framework for testing the homogeneity of conditional average treatment effects (CATEs) across multiple experimental and observational studies. Our approach leverages multiple randomized trials to assess whether treatment…

Econometrics · Economics 2026-02-25 Ana Armendariz , Martin Huber

Conditional Average Treatment Effects (CATE) estimation is one of the main challenges in causal inference with observational data. In addition to Machine Learning based-models, nonparametric estimators called meta-learners have been…

Machine Learning · Statistics 2023-06-06 Naoufal Acharki , Ramiro Lugo , Antoine Bertoncello , Josselin Garnier

Energy-based learning is a powerful learning paradigm that encapsulates various discriminative and generative approaches. An energy-based model (EBM) is typically formed of inner-model(s) that learn a combination of the different features…

Machine Learning · Computer Science 2023-06-05 Firas Laakom , Jenni Raitoharju , Alexandros Iosifidis , Moncef Gabbouj

Multi-attribute classification generalizes classification, presenting new challenges for making accurate predictions and quantifying uncertainty. We build upon recent work and show that architectures for multi-attribute prediction can be…

Computer Vision and Pattern Recognition · Computer Science 2021-08-10 Jacob Kelly , Richard Zemel , Will Grathwohl

Treatment effect heterogeneity plays an important role in many areas of causal inference and within recent years, estimation of the conditional average treatment effect (CATE) has received much attention in the statistical community. While…

Methodology · Statistics 2024-12-17 Simon Christoffer Ziersen , Torben Martinussen

Additive noise models (ANMs) are an important setting studied in causal inference. Most of the existing works on ANMs assume causal sufficiency, i.e., there are no unobserved confounders. This paper focuses on confounded ANMs, where a set…

Methodology · Statistics 2024-07-16 Muhammad Qasim Elahi , Mahsa Ghasemi , Murat Kocaoglu

Estimation of conditional average treatment effects (CATEs) plays an essential role in modern medicine by informing treatment decision-making at a patient level. Several metalearners have been proposed recently to estimate CATEs in an…

Applications · Statistics 2022-09-07 Yizhe Xu , Nikolaos Ignatiadis , Erik Sverdrup , Scott Fleming , Stefan Wager , Nigam Shah

When estimating heterogeneous treatment effects, missing outcome data can complicate treatment effect estimation, causing certain subgroups of the population to be poorly represented. In this work, we discuss this commonly overlooked…

Machine Learning · Statistics 2025-04-15 Matthew Pryce , Karla Diaz-Ordaz , Ruth H. Keogh , Stijn Vansteelandt

Energy-based models (EBMs) provide a powerful and flexible way of learning a joint probability distribution over data by constructing an energy surface. This energy surface enables insight extraction and conditional sampling. We apply EBMs…

Plasma Physics · Physics 2026-05-12 Phil Travis , Troy Carter

Accurately predicting conditional average treatment effects (CATEs) is crucial in personalized medicine and digital platform analytics. Since the treatments of interest often cannot be directly randomized, observational data is leveraged to…

Methodology · Statistics 2024-11-05 Miruna Oprescu , Nathan Kallus

Estimating treatment effects from observational data is challenging due to two main reasons: (a) hidden confounding, and (b) covariate mismatch (control and treatment groups not having identical distributions). Long lines of works exist…

Machine Learning · Computer Science 2025-04-30 Praharsh Nanavati , Ranjitha Prasad , Karthikeyan Shanmugam

Causal representation learning aims to unveil latent high-level causal representations from observed low-level data. One of its primary tasks is to provide reliable assurance of identifying these latent causal models, known as…

Machine Learning · Computer Science 2024-12-02 Yuhang Liu , Zhen Zhang , Dong Gong , Mingming Gong , Biwei Huang , Anton van den Hengel , Kun Zhang , Javen Qinfeng Shi

In this paper, we propose conjugate energy-based models (CEBMs), a new class of energy-based models that define a joint density over data and latent variables. The joint density of a CEBM decomposes into an intractable distribution over…

Machine Learning · Computer Science 2021-06-28 Hao Wu , Babak Esmaeili , Michael Wick , Jean-Baptiste Tristan , Jan-Willem van de Meent

Conditional Average Treatment Effect (CATE) estimation, at the heart of counterfactual reasoning, is a crucial challenge for causal modeling both theoretically and applicatively, in domains such as healthcare, sociology, or advertising.…

Machine Learning · Computer Science 2025-01-27 Armand Lacombe , Michèle Sebag

The estimation of Conditional Average Treatment Effects (CATE) is crucial for understanding the heterogeneity of treatment effects in clinical trials. We evaluate the performance of common methods, including causal forests and various…

Methodology · Statistics 2024-07-12 Oshri Machluf , Tzviel Frostig , Gal Shoham , Tomer Milo , Elad Berkman , Raviv Pryluk

Estimates of heterogeneous treatment effects such as conditional average treatment effects (CATEs) and conditional quantile treatment effects (CQTEs) play an important role in real-world decision making. Given this importance, one should…

Machine Learning · Statistics 2025-05-02 Justin Whitehouse , Christopher Jung , Vasilis Syrgkanis , Bryan Wilder , Zhiwei Steven Wu

In the last decades, energy-based models (EBMs) have become an important class of probabilistic models in which a component of the likelihood is intractable and therefore cannot be evaluated explicitly. Consequently, parameter estimation in…

Computational Engineering, Finance, and Science · Computer Science 2026-04-10 Luca Martino