English
Related papers

Related papers: Optimal Transport for Treatment Effect Estimation

200 papers

We study two-way-fixed-effects regressions (TWFE) with several treatment variables. Under a parallel trends assumption, we show that the coefficient on each treatment identifies a weighted sum of that treatment's effect, with possibly…

Econometrics · Economics 2023-04-18 Clément de Chaisemartin , Xavier D'Haultfœuille

This study examines the problem of determining whether to treat individuals based on observed covariates. The most common decision rule is the conditional empirical success (CES) rule proposed by Manski (2004), which assigns individuals to…

Econometrics · Economics 2025-10-08 Takuya Ishihara , Daisuke Kurisu

We study the statistical properties of the entropic optimal (self) transport problem for smooth probability measures. We provide an accurate description of the limit distribution for entropic (self-)potentials and plans as the…

Statistics Theory · Mathematics 2026-04-30 Gilles Mordant

Adaptive curricula in reinforcement learning (RL) have proven effective for producing policies robust to discrepancies between the train and test environment. Recently, the Unsupervised Environment Design (UED) framework generalized RL…

In this work, we introduce an optimal transport framework for inferring power distributions over both spatial location and temporal frequency. Recently, it has been shown that optimal transport is a powerful tool for estimating spatial…

Optimization and Control · Mathematics 2024-03-01 Isabel Haasler , Filip Elvander

The area under the curve (AUC) of the mean cumulative function (MCF) has recently been introduced as a novel estimand for evaluating treatment effects in recurrent event settings, offering an alternative to the commonly used…

Methodology · Statistics 2025-07-22 Jiren Sun , Tuo Wang , Yanyao Yi , Ting Ye , Jun Shao , Yu Du

Causal Inference has wide applications in various areas such as E-commerce and precision medicine, and its performance heavily relies on the accurate estimation of the Individual Treatment Effect (ITE). Conventionally, ITE is predicted by…

Machine Learning · Computer Science 2023-10-23 Kailiang Zhong , Fengtong Xiao , Yan Ren , Yaorong Liang , Wenqing Yao , Xiaofeng Yang , Ling Cen

Recent methodological research in causal inference has focused on effects of stochastic interventions, which assign treatment randomly, often according to subject-specific covariates. In this work, we demonstrate that the usual notion of…

Individual treatment effect (ITE) represents the expected improvement in the outcome of taking a particular action to a particular target, and plays important roles in decision making in various domains. However, its estimation problem is…

Machine Learning · Computer Science 2020-05-12 Shonosuke Harada , Hisashi Kashima

Counterfactual explanations (CEs) offer a human-understandable way to explain decisions by identifying specific changes to the input parameters of a base or present model that would lead to a desired change in the outcome. For optimization…

Optimization and Control · Mathematics 2026-01-06 Felix Engelhardt , Jannis Kurtz , Ş. İlker Birbil , Ted Ralphs

This study proposes a novel framework based on the RuleFit method to estimate Heterogeneous Treatment Effect (HTE) in a randomized clinical trial. To achieve this, we adopted S-learner of the metaalgorithm for our proposed framework. The…

Methodology · Statistics 2023-07-28 Mayu Hiraishi , Ke Wan , Kensuke Tanioka , Hiroshi Yadohisa , Toshio Shimokawa

This paper presents a weighted optimization framework that unifies the binary,multi-valued, continuous, as well as mixture of discrete and continuous treatment, under the unconfounded treatment assignment. With a general loss function, the…

Econometrics · Economics 2018-08-20 Chunrong Ai , Oliver Linton , Kaiji Motegi , Zheng Zhang

Individual Treatment Effects (ITE) estimation methods have risen in popularity in the last years. Most of the time, individual effects are better presented as Conditional Average Treatment Effects (CATE). Recently, representation balancing…

Machine Learning · Statistics 2022-03-30 Ayoub Abraich , Agathe Guilloux , Blaise Hanczar

To embed structured knowledge within labels into feature representations, prior work [Zeng et al., 2022] proposed to use the Cophenetic Correlation Coefficient (CPCC) as a regularizer during supervised learning. This regularizer calculates…

Machine Learning · Computer Science 2025-04-22 Siqi Zeng , Sixian Du , Makoto Yamada , Han Zhao

The purpose of this work is to improve the efficiency in estimating the average causal effect (ACE) on the survival scale where right-censoring exists and high-dimensional covariate information is available. We propose new estimators using…

Methodology · Statistics 2021-06-29 Ran Dai , Cheng Zheng , Mei-Jie Zhang

Counterfactual explanations (CE) explain model decisions by identifying input modifications that lead to different predictions. Most existing methods operate at the instance level. Distributional Counterfactual Explanations (DCE) extend…

Machine Learning · Computer Science 2026-03-18 Yikai Gu , Lele Cao , Bo Zhao , Lei Lei , Lei You

Causal analysis for time series data, in particular estimating individualized treatment effect (ITE), is a key task in many real-world applications, such as finance, retail, healthcare, etc. Real-world time series can include large-scale,…

Machine Learning · Computer Science 2023-03-07 Defu Cao , James Enouen , Yujing Wang , Xiangchen Song , Chuizheng Meng , Hao Niu , Yan Liu

Matching is a widely used causal inference design that aims to approximate a randomized experiment using observational data by forming matched sets of treated and control units based on similarities in their covariates. Ideally, treated…

Methodology · Statistics 2026-04-06 Jianan Zhu , Jeffrey Zhang , Zijian Guo , Siyu Heng

Observational studies of recurrent event rates are common in biomedical statistics. Broadly, the goal is to estimate differences in event rates under two treatments within a defined target population over a specified followup window.…

Methodology · Statistics 2024-11-13 Arman Oganisian , Anthony Girard , Jon A. Steingrimsson , Patience Moyo

The application of counterfactual explanation (CE) techniques in the realm of electroencephalography (EEG) classification has been relatively infrequent in contemporary research. In this study, we attempt to introduce and explore a novel…

Signal Processing · Electrical Eng. & Systems 2023-12-18 Kang Yin , Hye-Bin Shin , Hee-Dong Kim , Seong-Whan Lee