English
Related papers

Related papers: causalfe: Causal Forests with Fixed Effects in Pyt…

200 papers

Healthcare decision-making requires not only accurate predictions but also insights into how factors influence patient outcomes. While traditional Machine Learning (ML) models excel at predicting outcomes, such as identifying high risk…

Machine Learning · Computer Science 2025-01-28 Sheresh Zahoor , Pietro Liò , Gaël Dias , Mohammed Hasanuzzaman

Decision-making plays a pivotal role in shaping outcomes in various disciplines, such as medicine, economics, and business. This paper provides guidance to practitioners on how to implement a decision tree designed to address treatment…

Econometrics · Economics 2024-06-05 Hugo Bodory , Federica Mascolo , Michael Lechner

There has been considerable recent interest in estimating heterogeneous causal effects. In this paper, we study conditional average partial causal effects (CAPCE) to reveal the heterogeneity of causal effects with continuous treatment. We…

Machine Learning · Computer Science 2024-06-03 Yuta Kawakami , Manabu Kuroki , Jin Tian

In this paper we present a data-adaptive estimation procedure for estimation of average treatment effects in a time-to-event setting based on generalized random forests. In these kinds of settings, the definition of causal effect parameters…

Methodology · Statistics 2021-04-28 Helene C. W. Rytgaard , Claus T. Ekstrøm , Lars V. Kessing , Thomas A. Gerds

We propose causal effect estimators based on empirical Fr\'{e}chet means and operator-valued kernels, tailored to functional data spaces. These methods address the challenges of high-dimensionality, sequential ordering, and model complexity…

Methodology · Statistics 2025-06-04 Yordan P. Raykov , Hengrui Luo , Justin D. Strait , Wasiur R. KhudaBukhsh

Forest-based methods have recently gained in popularity for non-parametric treatment effect estimation. Building on this line of work, we introduce causal survival forests, which can be used to estimate heterogeneous treatment effects in a…

Methodology · Statistics 2023-03-01 Yifan Cui , Michael R. Kosorok , Erik Sverdrup , Stefan Wager , Ruoqing Zhu

Analyzing data from multiple sources offers valuable opportunities to improve the estimation efficiency of causal estimands. However, this analysis also poses many challenges due to population heterogeneity and data privacy constraints.…

Methodology · Statistics 2025-10-23 Rong Zhao , Jason Falvey , Xu Shi , Vernon M. Chinchilli , Chixiang Chen

This study proposes an end-to-end algorithm for policy learning in causal inference. We observe data consisting of covariates, treatment assignments, and outcomes, where only the outcome corresponding to the assigned treatment is observed.…

Econometrics · Economics 2025-12-30 Masahiro Kato

We propose a fully Bayesian approach for causal inference with multivariate categorical data based on staged tree models, a class of probabilistic graphical models capable of representing asymmetric and context-specific dependencies. To…

Methodology · Statistics 2025-11-06 Andrea Cremaschi , Manuele Leonelli , Gherardo Varando

This paper studies a panel data setting where the goal is to estimate causal effects of an intervention by predicting the counterfactual values of outcomes for treated units, had they not received the treatment. Several approaches have been…

Econometrics · Economics 2019-03-26 Susan Athey , Mohsen Bayati , Guido Imbens , Zhaonan Qu

Feature selection is a crucial preprocessing step in data analytics and machine learning. Classical feature selection algorithms select features based on the correlations between predictive features and the class variable and do not attempt…

Machine Learning · Computer Science 2019-11-19 Kui Yu , Xianjie Guo , Lin Liu , Jiuyong Li , Hao Wang , Zhaolong Ling , Xindong Wu

We present Causal Posterior Estimation (CPE), a novel method for Bayesian inference in simulator models, i.e., models where the evaluation of the likelihood function is intractable or too computationally expensive, but where one can…

Machine Learning · Computer Science 2025-05-28 Simon Dirmeier , Antonietta Mira

Research on causal effects often relies on synthetic data due to the scarcity of real-world datasets with ground-truth effects. Since current data-generating tools do not always meet all requirements for state-of-the-art research, ad-hoc…

Artificial Intelligence · Computer Science 2024-05-24 Andreas W M Sauter , Erman Acar , Aske Plaat

Federated causal inference enables multi-site treatment effect estimation without sharing individual-level data, offering a privacy-preserving solution for real-world evidence generation. However, data heterogeneity across sites, manifested…

Machine Learning · Computer Science 2025-05-06 Haoyang Li , Jie Xu , Kyra Gan , Fei Wang , Chengxi Zang

Causal inference has gained much popularity in recent years, with interests ranging from academic, to industrial, to educational, and all in between. Concurrently, the study and usage of neural networks has also grown profoundly (albeit at…

Machine Learning · Statistics 2024-05-07 Demetrios Papakostas , Andrew Herren , P. Richard Hahn , Francisco Castillo

Uplift modeling is a rapidly growing approach that utilizes causal inference and machine learning methods to directly estimate the heterogeneous treatment effects, which has been widely applied to various online marketplaces to assist…

Machine Learning · Statistics 2022-09-27 Shu Wan , Chen Zheng , Zhonggen Sun , Mengfan Xu , Xiaoqing Yang , Hongtu Zhu , Jiecheng Guo

We propose a novel method for estimating heterogeneous treatment effects based on the fused lasso. By first ordering samples based on the propensity or prognostic score, we match units from the treatment and control groups. We then run the…

The identification of heterogeneous treatment effects (HTE) across subgroups is of significant interest in clinical trial analysis. Several state-of-the-art HTE estimation methods, including causal forests, apply recursive partitioning for…

Methodology · Statistics 2025-06-10 Vik Shirvaikar , Andrea Storås , Xi Lin , Chris Holmes

When drawing causal inferences about the effects of multiple treatments on clustered survival outcomes using observational data, we need to address implications of the multilevel data structure, multiple treatments, censoring and unmeasured…

Methodology · Statistics 2022-02-18 Liangyuan Hu , Jiayi Ji , Ronald D. Ennis , Joseph W. Hogan

Investigating the relationships between two sets of variables helps to understand their interactions and can be done with canonical correlation analysis (CCA). However, the correlation between the two sets can sometimes depend on a third…