English
Related papers

Related papers: Causal rule ensemble approach for multi-arm data

200 papers

We propose a matching method for observational data that matches units with others in unit-specific, hyper-box-shaped regions of the covariate space. These regions are large enough that many matches are created for each unit and small…

Methodology · Statistics 2020-08-11 Marco Morucci , Vittorio Orlandi , Sudeepa Roy , Cynthia Rudin , Alexander Volfovsky

Causal inference and model interpretability are gaining increasing attention, particularly in the biomedical domain. Despite recent advance, decorrelating features in nonlinear environments with human-interpretable representations remains…

Machine Learning · Computer Science 2024-11-12 Junda Wang , Weijian Li , Han Wang , Hanjia Lyu , Caroline P. Thirukumaran , Addisu Mesfin , Hong Yu , Jiebo Luo

The simultaneous application of multiple treatments is increasingly common in many fields, such as healthcare and marketing. In such scenarios, it is important to estimate the single treatment effects and the interaction treatment effects…

Methodology · Statistics 2025-11-14 Yuki Murakami , Takumi Hattori , Kohsuke Kubota

We introduce an algorithm for identifying interpretable subgroups with elevated treatment effects, given an estimate of individual or conditional average treatment effects (CATE). Subgroups are characterized by ``rule sets'' --…

Machine Learning · Statistics 2025-07-15 Albert Chiu

We consider the problem of estimating the average treatment effect (ATE) in a semi-supervised learning setting, where a very small proportion of the entire set of observations are labeled with the true outcome but features predictive of the…

Methodology · Statistics 2020-10-27 David Cheng , Ashwin Ananthakrishnan , Tianxi Cai

Statistical inference of heterogeneous treatment effects (HTEs) across predefined subgroups is challenging when units interact because treatment effects may vary by pre-treatment variables, post-treatment exposure variables (that measure…

Econometrics · Economics 2024-10-02 Julius Owusu

Estimating heterogeneous treatment effects across individuals has attracted growing attention as a statistical tool for performing critical decision-making. We propose a Bayesian inference framework that quantifies the uncertainty in…

Methodology · Statistics 2023-12-19 Shunsuke Horii , Yoichi Chikahara

We study targeted maximum likelihood estimation (TMLE) of the average treatment effect in a semiparametric regression model whose mean function is indexed by a finite-dimensional parameter, while the additive error distribution is left…

Methodology · Statistics 2026-04-20 Mijeong Kim

The machine learning toolbox for estimation of heterogeneous treatment effects from observational data is expanding rapidly, yet many of its algorithms have been evaluated only on a very limited set of semi-synthetic benchmark datasets. In…

Machine Learning · Computer Science 2021-07-29 Alicia Curth , Mihaela van der Schaar

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

Evaluating heterogeneity of treatment effects (HTE) across subgroups is common in both randomized trials and observational studies. Although several statistical challenges of HTE analyses including low statistical power and multiple…

Methodology · Statistics 2024-07-10 Noorie Hyun , Abisola E. Idu , Andrea J. Cook , Jennifer F. Bobb

Understanding treatment effect heterogeneity has become an increasingly popular task in various fields, as it helps design personalized advertisements in e-commerce or targeted treatment in biomedical studies. However, most of the existing…

Methodology · Statistics 2024-07-12 Waverly Wei , Xinwei Ma , Jingshen Wang

Every design choice will have different effects on different units. However traditional A/B tests are often underpowered to identify these heterogeneous effects. This is especially true when the set of unit-level attributes is…

Artificial Intelligence · Computer Science 2016-11-09 Alexander Peysakhovich , Akos Lada

There is currently a dearth of appropriate methods to estimate the causal effects of multiple treatments when the outcome is binary. For such settings, we propose the use of nonparametric Bayesian modeling, Bayesian Additive Regression…

Methodology · Statistics 2020-03-02 Chenyang Gu , Michael J. Lopez , Liangyuan Hu

We present unexpected findings from a large-scale benchmark study evaluating Conditional Average Treatment Effect (CATE) estimation algorithms, i.e., CATE models. By running 16 modern CATE models on 12 datasets and 43,200 sampled variants…

Machine Learning · Statistics 2025-02-21 Haining Yu , Yizhou Sun

The Average Treatment Effect (ATE) is a foundational metric in causal inference, widely used to assess intervention efficacy in randomized controlled trials (RCTs). However, in many applications -- particularly in healthcare -- this static…

Machine Learning · Computer Science 2025-07-23 Julianna Piskorz , Krzysztof Kacprzyk , Harry Amad , Mihaela van der Schaar

We introduce a robust framework for heterogeneous treatment effect (HTE) estimation tailored to high-dimensional low sample size (HDLSS) settings. By combining Graph Attention Networks (GAT) to capture structural dependencies among…

Methodology · Statistics 2025-09-16 Byeonghee Lee , Joonsung Kang

We consider the estimation of treatment effects in settings when multiple treatments are assigned over time and treatments can have a causal effect on future outcomes or the state of the treated unit. We propose an extension of the…

Econometrics · Economics 2021-06-18 Greg Lewis , Vasilis Syrgkanis

Treatment effect estimation, which refers to the estimation of causal effects and aims to measure the strength of the causal relationship, is of great importance in many fields but is a challenging problem in practice. As present,…

Machine Learning · Computer Science 2021-07-20 Zhenyu Guo , Shuai Zheng , Zhizhe Liu , Kun Yan , Zhenfeng Zhu

Estimating heterogeneous treatment effects is important to tailor treatments to those individuals who would most likely benefit. However, conditional average treatment effect predictors may often be trained on one population but possibly…

Machine Learning · Computer Science 2024-10-18 Christoph Kern , Michael Kim , Angela Zhou