English
Related papers

Related papers: Dynamic Inter-treatment Information Sharing for In…

200 papers

Recent development in the data-driven decision science has seen great advances in individualized decision making. Given data with individual covariates, treatment assignments and outcomes, policy makers best individualized treatment rule…

Machine Learning · Statistics 2020-06-29 Weibin Mo , Zhengling Qi , Yufeng Liu

We investigate the problem of machine learning-based (ML) predictive inference on individual treatment effects (ITEs). Previous work has focused primarily on developing ML-based meta-learners that can provide point estimates of the…

Machine Learning · Computer Science 2023-08-30 Ahmed Alaa , Zaid Ahmad , Mark van der Laan

Designing faithful yet accurate AI models is challenging, particularly in the field of individual treatment effect estimation (ITE). ITE prediction models deployed in critical settings such as healthcare should ideally be (i) accurate, and…

Machine Learning · Computer Science 2024-06-04 Yinjun Wu , Mayank Keoliya , Kan Chen , Neelay Velingker , Ziyang Li , Emily J Getzen , Qi Long , Mayur Naik , Ravi B Parikh , Eric Wong

Determining causal effects of interventions onto outcomes from real-world, observational (non-randomized) data, e.g., treatment repurposing using electronic health records, is challenging due to underlying bias. Causal deep learning has…

Machine Learning · Computer Science 2025-10-23 Shantanu Ghosh , Zheng Feng , Jiang Bian , Kevin Butler , Mattia Prosperi

Uncertainty quantification for individual treatment effects (ITEs) is a daunting challenge in causal inference. Motivated by recent advances in conformal prediction, several works aim to construct distribution-free prediction sets for ITEs…

Methodology · Statistics 2026-05-07 Chongguang Tao , Zheng Zhou , Yuhong Yang

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

Despite the impressive advancements achieved using deep-learning for functional brain activity analysis, the heterogeneity of functional patterns and scarcity of imaging data still pose challenges in tasks such as prediction of future onset…

Image and Video Processing · Electrical Eng. & Systems 2023-12-25 Wenhui Cui , Haleh Akrami , Ganning Zhao , Anand A. Joshi , Richard M. Leahy

We investigate the problem of estimating the average treatment effect (ATE) under a very general setup where the covariates can be high-dimensional, highly correlated, and can have sparse nonlinear effects on the propensity and outcome…

Machine Learning · Statistics 2025-08-26 Jianqing Fan , Soham Jana , Sanjeev Kulkarni , Qishuo Yin

Estimating heterogeneous treatment effect is an important task in causal inference with wide application fields. It has also attracted increasing attention from machine learning community in recent years. In this work, we reinterpret the…

Methodology · Statistics 2018-10-26 Ran Chen , Hanzhong Liu

Replicating causal estimates across different cohorts is crucial for increasing the integrity of epidemiological studies. However, strong assumptions regarding unmeasured confounding and effect modification often hinder this goal. By…

Methodology · Statistics 2024-09-23 Roy S. Zawadzki , Daniel L. Gillen

This paper studies treatment effect models in which individuals are classified into unobserved groups based on heterogeneous treatment rules. Using a finite mixture approach, we propose a marginal treatment effect (MTE) framework in which…

Econometrics · Economics 2022-05-24 Tadao Hoshino , Takahide Yanagi

Many medical datasets have recently been created for medical image segmentation tasks, and it is natural to question whether we can use them to sequentially train a single model that (1) performs better on all these datasets, and (2)…

Computer Vision and Pattern Recognition · Computer Science 2022-08-02 Chenyu You , Jinlin Xiang , Kun Su , Xiaoran Zhang , Siyuan Dong , John Onofrey , Lawrence Staib , James S. Duncan

We propose a simple and efficient approach for training the BERT model. Our approach exploits the special structure of BERT that contains a stack of repeated modules (i.e., transformer encoders). Our proposed approach first trains BERT with…

Machine Learning · Computer Science 2021-10-11 Shuo Yang , Le Hou , Xiaodan Song , Qiang Liu , Denny Zhou

Understanding the effects of quarantine policies in populations with underlying social networks is crucial for public health, yet most causal inference methods fail here due to their assumption of independent individuals. We introduce…

Artificial Intelligence · Computer Science 2024-12-09 Suhan Guo , Furao Shen , Ni Li

The estimation of individual treatment effects (ITE) focuses on predicting the outcome changes that result from a change in treatment. A fundamental challenge in observational data is that while we need to infer outcome differences under…

Machine Learning · Computer Science 2025-12-23 Zichuan Lin , Xiaokai Huang , Jiate Liu , Yuxuan Han , Jia Chen , Xiapeng Wu , Deheng Ye

Accurately estimating personalized treatment effects within a study site (e.g., a hospital) has been challenging due to limited sample size. Furthermore, privacy considerations and lack of resources prevent a site from leveraging…

Machine Learning · Statistics 2022-06-17 Xiaoqing Tan , Chung-Chou H. Chang , Ling Zhou , Lu Tang

Deep learning has emerged as a technique of choice for rapid feature extraction across imaging disciplines, allowing rapid conversion of the data streams to spatial or spatiotemporal arrays of features of interest. However, applications of…

Data Analysis, Statistics and Probability · Physics 2021-01-25 Ayana Ghosh , Bobby G. Sumpter , Ondrej Dyck , Sergei V. Kalinin , Maxim Ziatdinov

This paper develops a sensitivity analysis framework that transfers the average total treatment effect (ATTE) from source data with a fully observed network to target data whose network is completely unknown. The ATTE represents the average…

Methodology · Statistics 2025-10-17 Tadao Hoshino

Treatment effect estimation (TEE) is the task of determining the impact of various treatments on patient outcomes. Current TEE methods fall short due to reliance on limited labeled data and challenges posed by sparse and high-dimensional…

Machine Learning · Computer Science 2024-03-07 Ruoqi Liu , Lingfei Wu , Ping Zhang

To promote precision medicine, individualized treatment regimes (ITRs) are crucial for optimizing the expected clinical outcome based on patient-specific characteristics. However, existing ITR research has primarily focused on scenarios…

Methodology · Statistics 2024-02-20 Chang Wang , Lu Wang