English
Related papers

Related papers: Improving Precision of RCT-Based CATE Estimation u…

200 papers

Recent advances in natural language processing have enabled the increasing use of text data in causal inference, particularly for adjusting confounding factors in treatment effect estimation. Although high-dimensional text can encode rich…

Machine Learning · Computer Science 2025-12-08 Lijinghua Zhang , Hengrui Cai

One of the central goals of causal machine learning is the accurate estimation of heterogeneous treatment effects from observational data. In recent years, meta-learning has emerged as a flexible, model-agnostic paradigm for estimating…

Artificial Intelligence · Computer Science 2024-11-14 Henri Arno , Paloma Rabaey , Thomas Demeester

Randomized Controlled Trials (RCT)s are relied upon to assess new treatments, but suffer from limited power to guide personalized treatment decisions. On the other hand, observational (i.e., non-experimental) studies have large and diverse…

Methodology · Statistics 2023-03-07 Zeshan Hussain , Ming-Chieh Shih , Michael Oberst , Ilker Demirel , David Sontag

In cluster-randomized trials (CRTs), missing data can occur in various ways, including missing values in outcomes and baseline covariates at the individual or cluster level, or completely missing information for non-participants. Among the…

Methodology · Statistics 2025-11-06 Bingkai Wang , Fan Li , Rui Wang

State-level policy studies often conduct heterogeneity analyses that quantify how treatment effects vary across state characteristics. These analyses may be used to inform state-specific policy decisions, or to infer how the effect of a…

The conditional tail average treatment effect (CTATE) is defined as a difference between the conditional tail expectations of potential outcomes, which can capture heterogeneity and deliver aggregated local information on treatment effects…

Applications · Statistics 2024-05-21 Le-Yu Chen , Yu-Min Yen

In industrial scenarios, data augmentation is an effective approach to improve model performance. However, its benefits are not unidirectionally beneficial. There is no theoretical research or established estimation for the optimal sample…

Machine Learning · Computer Science 2026-02-19 Mingchun Sun , Rongqiang Zhao , Zhennan Huang , Songyu Ding , Jie Liu

Parameter estimation of mixture regression model using the expectation maximization (EM) algorithm is highly sensitive to outliers. Here we propose a fast and efficient robust mixture regression algorithm, called Component-wise Adaptive…

Methodology · Statistics 2021-04-20 Wennan Chang , Xinyu Zhou , Yong Zang , Chi Zhang , Sha Cao

Time-to-event outcomes are often subject to left truncation and right censoring. While many survival analysis methods have been developed to handle truncation and censoring, majority of the past works require strong independence…

Methodology · Statistics 2025-07-28 Yuyao Wang , Andrew Ying , Ronghui Xu

The conditional average treatment effect (CATE) is a commonly targeted statistical parameter for measuring the effect of a treatment conditional on covariates. However, the CATE will fail to capture effects of treatments beyond differences…

Methodology · Statistics 2026-04-03 Jeffrey Näf , Junhyung Park , Herbert Susmann

Assume that one is interested in estimating an average treatment effect (ATE), equal to a weighted average of $S$ conditional average treatment effects (CATEs). One has unbiased estimators of the CATEs. One could just average the CATE…

Econometrics · Economics 2025-11-11 Clément de Chaisemartin

Artificial intelligence (AI) has significantly improved medical screening accuracy, particularly in cancer detection and risk assessment. However, traditional classification metrics often fail to account for imbalanced data, varying…

Machine Learning · Computer Science 2025-10-28 Longfei Wei , Fang Sheng , Jianfei Zhang

Randomized controlled trials (RCTs) often include subgroup analyses to assess whether treatment effects vary across pre-specified patient populations. However, these analyses frequently suffer from small sample sizes which limit the power…

Offline reinforcement learning is important in many settings with available observational data but the inability to deploy new policies online due to safety, cost, and other concerns. Many recent advances in causal inference and machine…

Machine Learning · Statistics 2025-07-08 Defu Cao , Angela Zhou

In cluster-randomized trials (CRTs), entire clusters of individuals are randomized to treatment, and outcomes within a cluster are typically correlated. While frequentist approaches are standard practice for CRT analysis, Bayesian methods…

Methodology · Statistics 2025-11-27 Ruyi Liu , Joshua L. Warren , Yuki Ohnishi , Donna Spiegelman , Liangyuan Hu , Fan Li

Estimation of individualized treatment effects (ITE), also known as conditional average treatment effects (CATE), is an active area of methodology development. However, much less attention has been paid to the quantification of uncertainty…

Methodology · Statistics 2025-04-08 Daijiro Kabata , Nicholas C. Henderson , Ravi Varadhan

Accurate heterogeneous treatment effect (HTE) estimation is essential for personalized recommendations, making it important to evaluate and compare HTE estimators. Traditional assessment methods are inapplicable due to missing…

Methodology · Statistics 2024-12-30 Zijun Gao

Causal decomposition analysis aims to assess the effect of modifying risk factors on reducing social disparities in outcomes. Recently, this analysis has incorporated individual characteristics when modifying risk factors by utilizing…

Machine Learning · Statistics 2025-09-16 Soojin Park , Suyeon Kang , Chioun Lee

We consider estimation of the target population average treatment effect (TATE) when outcome information is unavailable. Instead, we observe the outcome in multiple source populations and wish to combine the treatment effects therein to…

Methodology · Statistics 2025-05-16 Zehao Su , Helene Charlotte Rytgaard , Henrik Ravn , Frank Eriksson

Quantization has become a predominant approach for model compression, enabling deployment of large models trained on GPUs onto smaller form-factor devices for inference. Quantization-aware training (QAT) optimizes model parameters with…

Machine Learning · Computer Science 2022-12-13 Zheng Wang , Juncheng B Li , Shuhui Qu , Florian Metze , Emma Strubell