Doubly Robust Fusion of Many Treatments for Policy Learning
Not on the map. Why?
Abstract (arXiv)
Individualized treatment rules/recommendations (ITRs) aim to improve patient outcomes by tailoring treatments to the characteristics of each individual. However, when there are many treatment groups, existing methods face significant challenges due to data sparsity within treatment groups and highly unbalanced covariate distributions across groups. To address these challenges, we propose a novel calibration-weighted treatment fusion procedure that robustly balances covariates across treatment groups and fuses similar treatments using a penalized working model. The fusion procedure ensures the recovery of latent treatment group structures when either the calibration model or the outcome model is correctly specified. In the fused treatment space, practitioners can seamlessly apply state-of-the-art ITR learning methods with the flexibility to utilize a subset of covariates, thereby achieving robustness while addressing practical concerns such as fairness. We establish theoretical guarantees, including consistency, the oracle property of treatment fusion, and regret bounds when integrated with multi-armed ITR learning methods such as policy trees. Simulation studies show superior group recovery and policy value compared to existing approaches. We illustrate the practical utility of our method using a nationwide electronic health record-derived de-identified database containing data from patients with Chronic Lymphocytic Leukemia and Small Lymphocytic Lymphoma.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 34 of its 60 references that are in the map. First posted on arXiv 12 May 2025.
Builds on
34 of 60 references.
References · 34
- 2018Offline Multi-Action Policy Learning: Generalization and Optimizationcited by 64
- 2022Off-Policy Evaluation for Large Action Spaces via Embeddingscited by 53
- 2021Causal Effect Inference for Structured Treatmentscited by 38
- 2017The Impact of Random Models on Clustering Similaritycited by 34
- 2023Off-Policy Evaluation for Large Action Spaces via Conjunct Effect Modelingcited by 34
- 2020Learning Optimal Distributionally Robust Individualized Treatment Rulescited by 28
- 2019Semi-Parametric Efficient Policy Learning with Continuous Actionscited by 26
- 2022Conformal Off-Policy Prediction in Contextual Banditscited by 25
- 2023Augmented balancing weights as linear regressioncited by 21
- 2023Zero-shot causal learningcited by 17
- 2023Reliable Off-Policy Learning for Dosage Combinationscited by 15
- 2015A concave pairwise fusion approach to subgroup analysiscited by 14
22 more
- 2020Deep Jump Learning for Off-Policy Evaluation in Continuous Treatment Settingscited by 14
- 2022Conformal Off-policy Predictioncited by 14
- 2023Off-Policy Evaluation for Large Action Spaces via Policy Convolutioncited by 14
- 2023Synthetic Combinations: A Causal Inference Framework for Combinatorial Interventionscited by 12
- 2015Using Decision Lists to Construct Interpretable and Parsimonious Treatment Regimescited by 11
- 2020Nonparametric inverse probability weighted estimators based on the highly adaptive lassocited by 8
- 2020Improving trial generalizability using observational studiescited by 7
- 2022Estimating Heterogeneous Causal Effects of High-Dimensional Treatments: Application to Conjoint Analysiscited by 7
- 2024Bayesian Off-Policy Evaluation and Learning for Large Action Spacescited by 5
- 2024Minimax Regret Estimation for Generalizing Heterogeneous Treatment Effects with Multisite Datacited by 5
- 2020Model-assisted cohort selection with bias analysis for generating large-scale cohorts from the EHR for oncology researchcited by 4
- 2020Estimation and inference on high-dimensional individualized treatment rule in observational data using split-and-pooled de-correlated scorecited by 4
- 2021Transfer learning of individualized treatment rules from experimental to real-world datacited by 4
- 2023Optimal Individualized Treatment Rule for Combination Treatments Under Budget Constraintscited by 4
- 2019High dimensional precision medicine from patient-derived xenograftscited by 3
- 2020Learning Robust Decision Policies from Observational Datacited by 3
- 2024Robust Offline Policy Learning with Observational Data from Multiple Sourcescited by 3
- 2022Targeted Optimal Treatment Regime Learning Using Summary Statisticscited by 2
- 2024Causal Customer Churn Analysis with Low-rank Tensor Block Hazard Modelcited by 2
- 2021Flexible Inference of Optimal Individualized Treatment Strategy in Covariate Adjusted Randomization with Multiple Covariatescited by 1
- 2023Multi-Label Residual Weighted Learning for Individualized Combination Treatment Rulecited by 1
- 2024Inference for Optimal Linear Treatment Regimes in Personalized Decision-makingcited by 1
Built on by
Citing papers per year: 2026: 1.
1 paper.