arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2505.08092stat.MEstat.ML

用于策略学习的多种治疗的双重稳健融合

Doubly Robust Fusion of Many Treatments for Policy Learning

Ke Zhu, Jianing Chu, Ilya Lipkovich, Wenyu Ye, Shu Yang

AI总结:

针对多治疗组中数据稀疏和协变量不平衡问题,提出校准加权治疗融合程序,双重稳健恢复潜在结构,理论保证且模拟和真实数据验证优于现有方法。

摘要:

个体化治疗规则/推荐(ITRs)旨在通过根据每个个体的特征定制治疗来改善患者预后。然而,当存在许多治疗组时,现有方法面临重大挑战,原因在于治疗组内的数据稀疏性以及跨组协变量分布的高度不平衡。为应对这些挑战,我们提出了一种新颖的校准加权治疗融合程序,该程序稳健地平衡跨治疗组的协变量,并通过惩罚工作模型融合相似的治疗。该融合程序确保在校准模型或结果模型被正确指定时,能够恢复潜在的治疗组结构。在融合的治疗空间中,从业者可以无缝应用最先进的ITR学习方法,并灵活利用协变量的子集,从而在解决公平性等实际问题的同时实现稳健性。我们建立了理论保证,包括一致性、治疗融合的oracle性质,以及与多臂ITR学习方法(如策略树)集成时的遗憾界。模拟研究表明,与现有方法相比,我们的方法在组恢复和策略价值方面表现更优。我们使用一个全国性的电子健康记录衍生的去标识化数据库(包含慢性淋巴细胞白血病和小淋巴细胞淋巴瘤患者的数据)来展示我们方法的实际效用。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑