English
Related papers

Related papers: M-learner:A Flexible And Powerful Framework To Stu…

200 papers

Estimating heterogeneous treatment effects (HTEs) over time is crucial in many disciplines such as personalized medicine. For example, electronic health records are commonly collected over several time periods and then used to personalize…

Machine Learning · Computer Science 2025-02-07 Dennis Frauen , Konstantin Hess , Stefan Feuerriegel

In this paper we propose a heterogeneous modeling framework which achieves individual-wise feature selection and individualized covariates' effects subgrouping simultaneously. In contrast to conventional model selection approaches, the new…

Methodology · Statistics 2019-06-11 Xiwei Tang , Fei Xue , Annie Qu

Machine learning methods for estimating heterogeneous treatment effects (HTE) facilitate large-scale personalized decision-making across various domains such as healthcare, policy making, education, and more. Current machine learning…

Machine Learning · Computer Science 2024-06-25 Disha Makhija , Joydeep Ghosh , Yejin Kim

Mediation analysis plays a crucial role in causal inference as it can investigate the pathways through which treatment influences outcome. Most existing mediation analysis assumes that mediation effects are static and homogeneous within…

Methodology · Statistics 2026-03-02 Yijiao Zhang , Yubai Yuan , Yuexia Zhang , Zhongyi Zhu , Annie Qu

Understanding causal mechanisms is crucial for explaining and generalizing empirical phenomena. Causal mediation analysis offers statistical techniques to quantify the mediation effects. Although numerous methods have been developed for…

Methodology · Statistics 2026-05-12 Jiawei Fu

Estimating heterogeneous treatment effects is critical in domains such as personalized medicine, resource allocation, and policy evaluation. A central challenge lies in identifying subpopulations that respond differently to interventions,…

Machine Learning · Statistics 2025-09-18 Zilong Wang , Turgay Ayer , Shihao Yang

For mental disorders, patients' underlying mental states are non-observed latent constructs which have to be inferred from observed multi-domain measurements such as diagnostic symptoms and patient functioning scores. Additionally,…

Machine Learning · Computer Science 2020-11-03 Yuan Chen , Donglin Zeng , Tianchen Xu , Yuanjia Wang

Understanding treatment effect heterogeneity is vital for scientific and policy research. However, identifying and evaluating heterogeneous treatment effects pose significant challenges due to the typically unknown subgroup structure.…

Methodology · Statistics 2024-11-05 Kwangho Kim , Jisu Kim , Larry A. Wasserman , Edward H. Kennedy

Learning heterogeneous treatment effects (HTEs) is an important problem across many fields. Most existing methods consider the setting with a single treatment arm and a single outcome metric. However, in many real world domains, experiments…

Machine Learning · Computer Science 2022-06-13 Leon Yao , Caroline Lo , Israel Nir , Sarah Tan , Ariel Evnine , Adam Lerer , Alex Peysakhovich

Flexible estimation of heterogeneous treatment effects lies at the heart of many statistical challenges, such as personalized medicine and optimal resource allocation. In this paper, we develop a general class of two-step algorithms for…

Machine Learning · Statistics 2020-08-07 Xinkun Nie , Stefan Wager

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

The R-learner is widely used in causal inference due to its flexibility and efficiency in estimating the conditional average treatment effect. However, extending the R-learner framework from binary to continuous treatments introduces a…

Methodology · Statistics 2025-04-21 Yichi Zhang , Dehan Kong , Shu Yang

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

Mediation analysis is difficult when the number of potential mediators is larger than the sample size. In this paper we propose new inference procedures for the indirect effect in the presence of high-dimensional mediators for linear…

Methodology · Statistics 2019-10-29 Ruixuan Rachel Zhou , Liewei Wang , Sihai Dave Zhao

Recently, from the personalized medicine perspective, there has been an increased demand to identify subgroups of subjects for whom treatment is effective. Consequently, the estimation of heterogeneous treatment effects (HTE) has been…

Methodology · Statistics 2024-08-02 Ryoma Hieda , Shintaro Yuki , Kensuke Tanioka , Hiroshi Yadohisa

Quantifying the heterogeneity of treatment effect is important for understanding how a commercial product or medical treatment affects different population subgroups. While much of treatment effect heterogeneity analysis focuses on the…

We introduce a new instrumental variable (IV) estimator for heterogeneous treatment effects in the presence of endogeneity. Our estimator is based on double/debiased machine learning (DML) and uses efficient machine learning instruments…

Methodology · Statistics 2026-02-06 Cyrill Scheidegger , Zijian Guo , Peter Bühlmann

Randomized trials are typically designed to detect average treatment effects but often lack the statistical power to uncover individual-level treatment effect heterogeneity, limiting their value for personalized decision-making. To address…

Machine Learning · Statistics 2026-03-19 Rickard Karlsson , Piersilvio De Bartolomeis , Issa J. Dahabreh , Jesse H. Krijthe

Robust estimation of heterogeneous treatment effects is a fundamental challenge for optimal decision-making in domains ranging from personalized medicine to educational policy. In recent years, predictive machine learning has emerged as a…

Machine Learning · Statistics 2025-06-23 Maximilian Schuessler , Erik Sverdrup , Robert Tibshirani

We study treatment effect modifiers for causal analysis in a social network, where neighbors' characteristics or network structure may affect the outcome of a unit, and the goal is to identify sub-populations with varying treatment effects…

Social and Information Networks · Computer Science 2021-11-09 Amir Gilad , Harsh Parikh , Sudeepa Roy , Babak Salimi