arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2412.08052cs.LGstat.ML

CANDOR: 反事实注释的双重稳健离策略评估

CANDOR: Counterfactual ANnotated DOubly Robust Off-Policy Evaluation

Stanford University(斯坦福大学);Emory University(埃默里大学);Columbia University(哥伦比亚大学);University of Michigan(密歇根大学);The Gladstone Institutes(加利福尼亚大学旧金山分校格罗斯曼研究所)

Aishwarya Mandyam, Shengpu Tang, Jiayu Yao, Jenna Wiens, Barbara E. Engelhardt

AI总结:

提出基于双重稳健框架的离策略评估方法,通过仅在奖励模型组件中融入反事实注释,在理论保证和实验上优于其他策略。

摘要:

离策略评估(OPE)对于将上下文赌博算法应用于高风险决策环境(如医疗保健)至关重要,因为新治疗策略在部署前必须进行评估。不幸的是,OPE技术本质上受到可用数据广度的限制,这可能不足以评估新策略的性能。最近的工作尝试通过添加专家注释的反事实样本来改善数据集覆盖。然而,此类注释通常不完美,可能导致比不使用任何注释更差的估计器性能。为了更好地利用不完美注释,我们提出了一族基于双重稳健(DR)框架的OPE估计器,该框架将重要性采样(IS)与奖励模型(直接方法,DM)相结合以获得更好的统计保证。我们研究了三种融入反事实注释的方式。在温和假设下,我们证明仅在DM组件中使用注释能产生最理想的理论结果。在多个医疗保健任务(包括真实世界电子健康记录(EHR)数据)上的实验表明,该策略在错误指定的奖励模型和不准确的注释下最为稳健。通过解决不完美注释带来的挑战,这项工作拓宽了OPE方法的适用性,并促进了医疗保健中决策策略的更安全部署。

英文摘要:

Off-policy evaluation (OPE) is critical for applying contextual bandit algorithms to high-stakes decision-making settings such as healthcare, where new treatment policies must be evaluated prior to deployment. Unfortunately, OPE techniques are inherently limited by the breadth of the available data, which may not be sufficient to evaluate the performance of a new policy. Recent work attempts to improve dataset coverage by adding expert-annotated counterfactual samples. However, such annotations are often imperfect and can lead to worse estimator performance than using no annotations at all. To better leverage imperfect annotations, we propose a family of OPE estimators grounded in the doubly robust (DR) framework, which combines importance sampling (IS) with a reward model (direct method, DM) for better statistical guarantees. We study three ways of incorporating counterfactual annotations. Under mild assumptions, we prove that using annotations within just the DM component yields the most desirable theoretical results. Experiments on multiple healthcare tasks, including real-world electronic health records (EHR) data, show that this strategy is most robust under misspecified reward models and inaccurate annotations. By addressing the challenges posed by imperfect annotations, this work broadens the applicability of OPE methods and facilitates safer deployment of decision-making policies in healthcare.

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

↑