English
Related papers

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

200 papers

In this paper, we consider a challenging but realistic continual learning (CL) problem, Few-Shot Continual Active Learning (FoCAL), where a CL agent is provided with unlabeled data for a new or a previously learned task in each increment…

Machine Learning · Computer Science 2022-10-14 Ali Ayub , Carter Fendley

Imbalanced datasets pose a considerable challenge in training deep learning (DL) models for medical diagnostics, particularly for segmentation tasks. Imbalance may be associated with annotation quality limited annotated datasets, rare…

Image and Video Processing · Electrical Eng. & Systems 2025-04-08 Bashir Alam , Masa Cirkovic , Mete Harun Akcay , Md Kaf Shahrier , Sebastien Lafond , Hergys Rexha , Kurt Benke , Sepinoud Azimi , Janan Arslan

One-shot Federated Learning (OFL) significantly reduces communication costs in FL by aggregating trained models only once. However, the performance of advanced OFL methods is far behind the normal FL. In this work, we provide a causal view…

Machine Learning · Computer Science 2024-10-29 Zhenheng Tang , Yonggang Zhang , Peijie Dong , Yiu-ming Cheung , Amelie Chi Zhou , Bo Han , Xiaowen Chu

Federated learning benefits from cross-training strategies, which enables models to train on data from distinct sources to improve generalization capability. However, due to inherent differences in data distributions, the optimization goals…

Artificial Intelligence · Computer Science 2025-09-17 Zhuang Qi , Lei Meng , Ruohan Zhang , Yu Wang , Xin Qi , Xiangxu Meng , Han Yu , Qiang Yang

Motivated by applications in personalized medicine and individualized policymaking, there is a growing interest in techniques for quantifying treatment effect heterogeneity in terms of the conditional average treatment effect (CATE). Some…

Methodology · Statistics 2024-06-04 Pawel Morzywolek , Johan Decruyenaere , Stijn Vansteelandt

It is generally believed that robust training of extremely large networks is critical to their success in real-world applications. However, when taken to the extreme, methods that promote robustness can hurt the model's sensitivity to rare…

Machine Learning · Computer Science 2021-05-21 Seungyeon Kim , Daniel Glasner , Srikumar Ramalingam , Cho-Jui Hsieh , Kishore Papineni , Sanjiv Kumar

In this paper we study the problems of estimating heterogeneity in causal effects in experimental or observational studies and conducting inference about the magnitude of the differences in treatment effects across subsets of the…

Machine Learning · Statistics 2022-06-08 Susan Athey , Guido Imbens

Federated learning (FL) is increasingly being recognized as a key approach to overcoming the data silos that so frequently obstruct the training and deployment of machine-learning models in clinical settings. This work contributes to a…

Machine Learning · Computer Science 2024-07-08 Fatemeh Tavakoli , D. B. Emerson , Sana Ayromlou , John Jewell , Amrit Krishnan , Yuchong Zhang , Amol Verma , Fahad Razak

The paper proposes an estimator to make inference of heterogeneous treatment effects sorted by impact groups (GATES) for non-randomised experiments. The groups can be understood as a broader aggregation of the conditional average treatment…

Econometrics · Economics 2020-03-30 Daniel Jacob

Recently, self-attention mechanisms have shown impressive performance in various NLP and CV tasks, which can help capture sequential characteristics and derive global information. In this work, we explore how to extend self-attention…

Computer Vision and Pattern Recognition · Computer Science 2022-05-05 Haowei Zhu , Wenjing Ke , Dong Li , Ji Liu , Lu Tian , Yi Shan

Current research in Visual Navigation reveals opportunities for improvement. First, the direct adoption of RNNs and Transformers often overlooks the specific differences between Embodied AI and traditional sequential data modelling,…

Robotics · Computer Science 2024-10-08 Ruoyu Wang , Yao Liu , Yuanjiang Cao , Lina Yao

We study the problem of model selection in causal inference, specifically for conditional average treatment effect (CATE) estimation. Unlike machine learning, there is no perfect analogue of cross-validation for model selection as we do not…

Machine Learning · Computer Science 2024-04-30 Divyat Mahajan , Ioannis Mitliagkas , Brady Neal , Vasilis Syrgkanis

Confounding bias is a key challenge in causal effect estimation from observational data. Double Machine Learning (DML) addresses this issue by estimating treatment and outcome nuisance functions, constructing treatment and outcome…

Machine Learning · Computer Science 2026-05-26 Guodu Xiang , Kui Yu , Yujie Wang , Richang Hong , Fuyuan Cao , Jiye Liang

The credibility revolution advances the use of research designs that permit identification and estimation of causal effects. However, understanding which mechanisms produce measured causal effects remains a challenge. The dominant current…

Econometrics · Economics 2026-04-08 Jiawei Fu , Tara Slough

Causal effect estimation (CEE) provides a crucial tool for predicting the unobserved counterfactual outcome for an entity. As CEE relaxes the requirement for ``perfect'' counterfactual samples (e.g., patients with identical attributes and…

Machine Learning · Computer Science 2024-11-19 Hechuan Wen , Tong Chen , Guanhua Ye , Li Kheng Chai , Shazia Sadiq , Hongzhi Yin

Federated Learning (FL) has emerged as a promising Machine Learning paradigm, enabling multiple users to collaboratively train a shared model while preserving their local data. To minimize computing and communication costs associated with…

Machine Learning · Computer Science 2023-11-14 Sofiane Bouaziz , Hadjer Benmeziane , Youcef Imine , Leila Hamdad , Smail Niar , Hamza Ouarnoughi

The conditional tail average treatment effect (CTATE) is defined as a difference between the conditional tail expectations of potential outcomes, which can capture heterogeneity and deliver aggregated local information on treatment effects…

Applications · Statistics 2024-05-21 Le-Yu Chen , Yu-Min Yen

Data-driven societal event forecasting methods exploit relevant historical information to predict future events. These methods rely on historical labeled data and cannot accurately predict events when data are limited or of poor quality.…

Machine Learning · Computer Science 2021-12-13 Songgaojun Deng , Huzefa Rangwala , Yue Ning

Discovering the causality from observational data is a crucial task in various scientific domains. With increasing awareness of privacy, data are not allowed to be exposed, and it is very hard to learn causal graphs from dispersed data,…

Machine Learning · Computer Science 2023-12-12 Dezhi Yang , Xintong He , Jun Wang , Guoxian Yu , Carlotta Domeniconi , Jinglin Zhang

While sample sizes in randomized clinical trials are large enough to estimate the average treatment effect well, they are often insufficient for estimation of treatment-covariate interactions critical to studying data-driven precision…

Machine Learning · Statistics 2020-04-22 Steve Yadlowsky , Fabio Pellegrini , Federica Lionetto , Stefan Braune , Lu Tian