English
Related papers

Related papers: A Doubly Robust Machine Learning Approach for Dise…

200 papers

Robust loss functions are crucial for training deep neural networks in the presence of label noise, yet existing approaches require extensive, dataset-specific hyperparameter tuning. In this work, we introduce Fractional Classification Loss…

Machine Learning · Computer Science 2025-08-11 Mert Can Kurucu , Tufan Kumbasar , İbrahim Eksin , Müjde Güzelkaya

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

Reliable estimation of treatment effects from observational data is important in many disciplines such as medicine. However, estimation is challenging when unconfoundedness as a standard assumption in the causal inference literature is…

Machine Learning · Computer Science 2024-10-15 Jonas Schweisthal , Dennis Frauen , Maresa Schröder , Konstantin Hess , Niki Kilbertus , Stefan Feuerriegel

Estimating how a treatment affects different individuals, known as heterogeneous treatment effect estimation, is an important problem in empirical sciences. In the last few years, there has been a considerable interest in adapting machine…

Machine Learning · Computer Science 2024-10-18 Christopher Tran , Keith Burghardt , Kristina Lerman , Elena Zheleva

Heterogeneous treatment effects (HTEs) are commonly identified during randomized controlled trials (RCTs). Identifying subgroups of patients with similar treatment effects is of high interest in clinical research to advance precision…

Machine Learning · Computer Science 2022-12-06 Peniel N. Argaw , Elizabeth Healey , Isaac S. Kohane

Real-time monitoring in modern medical research introduces functional longitudinal data, characterized by continuous-time measurements of outcomes, treatments, and confounders. This complexity leads to uncountably infinite…

Methodology · Statistics 2025-08-15 Andrew Ying

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

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

Many practical decision-making problems in economics and healthcare seek to estimate the average treatment effect (ATE) from observational data. The Double/Debiased Machine Learning (DML) is one of the prevalent methods to estimate ATE in…

Econometrics · Economics 2022-12-07 Yiyan Huang , Cheuk Hang Leung , Xing Yan , Qi Wu , Shumin Ma , Zhiri Yuan , Dongdong Wang , Zhixiang Huang

Causal inference, estimating causal effects from observational data, is a fundamental tool in many disciplines. Of particular importance across a variety of domains is the continuous treatment setting, where the variable of intervention has…

Machine Learning · Computer Science 2026-05-15 Christopher Stith , Medha Barath , Vahid Balazadeh , Jesse C. Cresswell , Rahul G. Krishnan

The estimation of Conditional Average Treatment Effects (CATE) is crucial for understanding the heterogeneity of treatment effects in clinical trials. We evaluate the performance of common methods, including causal forests and various…

Methodology · Statistics 2024-07-12 Oshri Machluf , Tzviel Frostig , Gal Shoham , Tomer Milo , Elad Berkman , Raviv Pryluk

To effectively optimize and personalize treatments, it is necessary to investigate the heterogeneity of treatment effects. With the wide range of users being treated over many online controlled experiments, the typical approach of manually…

Methodology · Statistics 2022-11-07 John Cai , Weinan Wang

Estimating the conditional average treatment effect (CATE) from observational data is relevant for many applications such as personalized medicine. Here, we focus on the widespread setting where the observational data come from multiple…

Machine Learning · Computer Science 2024-06-05 Jonas Schweisthal , Dennis Frauen , Mihaela van der Schaar , Stefan Feuerriegel

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

A new meta-algorithm for estimating the conditional average treatment effects is proposed in the paper. The main idea underlying the algorithm is to consider a new dataset consisting of feature vectors produced by means of concatenation of…

Machine Learning · Statistics 2019-09-10 Lev V. Utkin , Mikhail V. Kots , Viacheslav S. Chukanov

When treating depression, clinicians are interested in determining the optimal treatment for a given patient, which is challenging given the amount of treatments available. To advance individualized treatment allocation, integrating data…

Federated Continual Learning (FCL) has emerged as a robust solution for collaborative model training in dynamic environments, where data samples are continuously generated and distributed across multiple devices. This survey provides a…

Machine Learning · Computer Science 2025-07-17 Parisa Hamedi , Roozbeh Razavi-Far , Ehsan Hallaji

For treatment effects - one of the core issues in modern econometric analysis - prediction and estimation are two sides of the same coin. As it turns out, machine learning methods are the tool for generalized prediction models. Combined…

Econometrics · Economics 2021-04-27 Daniel Jacob

Functional data play a pivotal role across science and engineering, yet their infinite-dimensional nature makes representation learning challenging. Conventional statistical models depend on pre-chosen basis expansions or kernels, limiting…

Machine Learning · Computer Science 2025-10-02 Yifei Gao , Yong Chen , Chen Zhang

In this position and problem pitch paper, we offer a solution to the reference class problem in causal inference. We apply the Reconcile algorithm for model multiplicity in machine learning to reconcile heterogeneous effects in causal…

Machine Learning · Computer Science 2024-06-12 Audrey Chang , Emily Diana , Alexander Williams Tolbert