English
Related papers

Related papers: Causal Representation Learning with Optimal Compre…

200 papers

Real-world multi-agent systems are often dynamic and continuous, where the agents co-evolve and undergo changes in their trajectories and interactions over time. For example, the COVID-19 transmission in the U.S. can be viewed as a…

Machine Learning · Computer Science 2024-03-04 Zijie Huang , Jeehyun Hwang , Junkai Zhang , Jinwoo Baik , Weitong Zhang , Dominik Wodarz , Yizhou Sun , Quanquan Gu , Wei Wang

In precision medicine, one of the most important problems is estimating the optimal individualized treatment rules (ITR), which typically involves recommending treatment decisions based on fully observed individual characteristics of…

Methodology · Statistics 2025-10-15 Yue Zhang , Shanshan Luo , Zhi Geng , Yangbo He

Heterogeneous treatment effects, which vary according to individual covariates, are crucial in fields such as personalized medicine and tailored treatment strategies. In many applications, rather than considering the heterogeneity induced…

Methodology · Statistics 2025-08-26 Peng Wu , Pengtao Zeng , Zhaoqing Tian , Shaojie Wei

The conditional average treatment effect (CATE) is widely used in personalized medicine to inform therapeutic decisions. However, state-of-the-art methods for CATE estimation (so-called meta-learners) often perform poorly in the presence of…

Machine Learning · Computer Science 2026-03-12 Valentyn Melnychuk , Dennis Frauen , Jonas Schweisthal , Stefan Feuerriegel

Causal inference from observational datasets often relies on measuring and adjusting for covariates. In practice, measurements of the covariates can often be noisy and/or biased, or only measurements of their proxies may be available.…

Machine Learning · Computer Science 2022-02-23 Wenshuo Guo , Mingzhang Yin , Yixin Wang , Michael I. Jordan

Unmeasured confounding presents a significant challenge in causal inference from observational studies. Classical approaches often rely on collecting proxy variables, such as instrumental variables. However, in applications where the…

Methodology · Statistics 2025-01-16 Xiaochuan Shi , Dehan Kong , Linbo Wang

In recent years, there has been a growing interest in using machine learning techniques for the estimation of treatment effects. Most of the best-performing methods rely on representation learning strategies that encourage shared behavior…

Machine Learning · Computer Science 2024-04-19 Roger Pros , Jordi Vitrià

We study Federated Causal Inference, an approach to estimate treatment effects from decentralized data across centers. We compare three classes of Average Treatment Effect (ATE) estimators derived from the Plug-in G-Formula, ranging from…

Machine Learning · Statistics 2025-03-26 Rémi Khellaf , Aurélien Bellet , Julie Josse

Estimating heterogeneous treatment effect (HTE) for survival outcomes has gained increasing attention, as it captures the variation in treatment efficacy across patients or subgroups in delaying disease progression. However, most existing…

Methodology · Statistics 2025-11-27 Na Bo , Ying Ding

For observational studies, we study the sensitivity of causal inference when treatment assignments may depend on unobserved confounders. We develop a loss minimization approach for estimating bounds on the conditional average treatment…

Methodology · Statistics 2022-03-11 Steve Yadlowsky , Hongseok Namkoong , Sanjay Basu , John Duchi , Lu Tian

We propose a new modeling and estimation approach to select the optimal treatment regime from different options through constructing a robust estimating equation. The method is protected against misspecification of the propensity score…

Methodology · Statistics 2022-11-15 Trinetri Ghosh , Yanyuan Ma , Wensheng Zhu , Yuanjia Wang

Selecting data for training machine learning models is crucial since large, web-scraped, real datasets contain noisy artifacts that affect the quality and relevance of individual data points. These noisy artifacts will impact model…

Machine Learning · Computer Science 2025-03-20 Samuel Kessler , Tam Le , Vu Nguyen

It is increasingly common to encounter prediction tasks in the biomedical sciences for which multiple datasets are available for model training. Common approaches such as pooling datasets and applying standard statistical learning methods…

Machine Learning · Statistics 2021-10-05 Gabriel Loewinger , Rolando Acosta Nunez , Rahul Mazumder , Giovanni Parmigiani

Clinical study populations often differ meaningfully from the broader populations to which results are intended to generalize. Weighting methods such as inverse probability of sampling weights (IPSW) reweight study participants to resemble…

Methodology · Statistics 2025-12-02 William Stewart , Carly L. Brantner , Elizabeth A. Stuart , Laine Thomas

We develop a mathematical framework to define an optimal individualized treatment rule (ITR) within the context of prioritized outcomes in a randomized controlled trial. Our optimality criterion is based on the framework of generalized…

Methodology · Statistics 2025-06-17 François Petit , Gérard Biau , Raphaël Porcher

Predicting which action (treatment) will lead to a better outcome is a central task in decision support systems. To build a prediction model in real situations, learning from biased observational data is a critical issue due to the lack of…

Machine Learning · Statistics 2020-06-11 Akira Tanimoto , Tomoya Sakai , Takashi Takenouchi , Hisashi Kashima

Unsupervised graph alignment finds the node correspondence between a pair of attributed graphs by only exploiting graph structure and node features. One category of recent studies first computes the node representation and then matches…

Machine Learning · Computer Science 2025-05-07 Songyang Chen , Yu Liu , Lei Zou , Zexuan Wang , Youfang Lin

There is growing interest in estimating and analyzing heterogeneous treatment effects in experimental and observational studies. We describe a number of meta-algorithms that can take advantage of any supervised learning or regression method…

Statistics Theory · Mathematics 2019-06-18 Sören R. Künzel , Jasjeet S. Sekhon , Peter J. Bickel , Bin Yu

Uplift modeling is crucial in various applications ranging from marketing and policy-making to personalized recommendations. The main objective is to learn optimal treatment allocations for a heterogeneous population. A primary line of…

Methodology · Statistics 2023-12-20 Preetam Nandy , Xiufan Yu , Wanjun Liu , Ye Tu , Kinjal Basu , Shaunak Chatterjee

As an important problem in causal inference, we discuss the estimation of treatment effects (TEs). Representing the confounder as a latent variable, we propose Intact-VAE, a new variant of variational autoencoder (VAE), motivated by the…

Machine Learning · Statistics 2022-04-22 Pengzhou Wu , Kenji Fukumizu