arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2412.09232cs.LG
收藏

连续处理下的增益建模:一种先预测后优化的方法

Uplift modeling with continuous treatments: A predict-then-optimize approach

KU Leuven(库勒文大学);Humboldt University of Berlin(柏林洪堡大学)

Simon De Vos, Christopher Bockel-Rickermann, Stefan Lessmann, Wouter Verbeke

AI总结:

针对增益建模多聚焦二元处理、难以适配现实中连续剂量处理的问题,提出先预测后优化框架,用因果机器学习估计条件平均剂量响应,再以整数线性规划求解连续剂量分配,支持公平约束与效用最大化,在多领域展现灵活性。

摘要:

增益建模的目标是通过确定哪些实体应接受处理,推荐能优化特定结果的行动。一种常见方法包含两个步骤:首先是推理步骤,估计条件平均处理效应(CATEs);其次是优化步骤,根据实体的CATE值排序,并在给定预算内将处理分配给排名前k的实体。尽管增益建模通常聚焦于二元处理,但许多现实应用的特征是连续值处理,即处理剂量。\n本文提出一种先预测后优化的框架,使增益建模可支持连续处理。首先在推理步骤中,使用因果机器学习技术从数据中估计条件平均剂量响应(CADRs)。其次在优化步骤中,我们将连续处理的分配任务构建为剂量分配问题,并使用整数线性规划(ILP)求解。该方法使决策者能够高效且有效地分配处理剂量,同时平衡资源可用性,还可添加公平性考量等额外约束,或调整目标函数以考虑依赖于实例的成本和收益,从而最大化效用。\n实验对比了多种CADR估计器,阐明了策略价值与公平性之间的权衡,以及调整后目标函数的影响。这展示了该框架在医疗保健、信贷发放和人力资源管理等多样应用中的优势与灵活性。所有代码可在github.com/SimonDeVos/UMCT获取。

英文摘要:

The goal of uplift modeling is to recommend actions that optimize specific outcomes by determining which entities should receive treatment. One common approach involves two steps: first, an inference step that estimates conditional average treatment effects (CATEs), and second, an optimization step that ranks entities based on their CATE values and assigns treatment to the top k within a given budget. While uplift modeling typically focuses on binary treatments, many real-world applications are characterized by continuous-valued treatments, i.e., a treatment dose. This paper presents a predict-then-optimize framework to allow for continuous treatments in uplift modeling. First, in the inference step, conditional average dose responses (CADRs) are estimated from data using causal machine learning techniques. Second, in the optimization step, we frame the assignment task of continuous treatments as a dose-allocation problem and solve it using integer linear programming (ILP). This approach allows decision-makers to efficiently and effectively allocate treatment doses while balancing resource availability, with the possibility of adding extra constraints like fairness considerations or adapting the objective function to take into account instance-dependent costs and benefits to maximize utility. The experiments compare several CADR estimators and illustrate the trade-offs between policy value and fairness, as well as the impact of an adapted objective function. This showcases the framework's advantages and flexibility across diverse applications in healthcare, lending, and human resource management. All code is available on github.com/SimonDeVos/UMCT.

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

↑