arXiv:2511.01641cs.LGstat.ML2025-11被引 1

提出XTNet模型,精准估算多类别多值处理的交叉效应。

Cross-Treatment Effect Estimation for Multi-Category, Multi-Valued Causal Inference via Dynamic Neural Masking

  • 用动态掩码捕捉不同处理间的复杂交互作用。
  • 在真实数据A/B测试中,效果估计准确率显著优于基线方法。
  • 适合需要精细评估多种干预组合的医疗或营销场景。

反事实因果推断在扩展至多类别、多值处理时面临重大挑战,因异质干预间的复杂交叉效应难以建模。现有方法仍局限于二元或单一类型处理,受限于严格假设、可扩展性差,且缺乏对复杂干预场景的有效评估框架。本文提出XTNet,一种新型网络架构,用于多类别、多值处理效应估计。该方法引入具有动态掩码机制的交叉效应估计模块,无需严格结构假设即可捕捉处理间交互。架构采用分解策略,将基础效应与跨处理交互分离,实现对组合处理空间的高效建模。同时提出MCMV-AUCC评估指标,考虑处理成本与交互效应。在合成与真实世界数据集上的大量实验表明,XTNet在排序准确率和效应估计质量上持续优于当前最优基线。真实世界A/B测试结果进一步验证其有效性。

原文摘要 · Abstract (English)

Counterfactual causal inference faces significant challenges when extended to multi-category, multi-valued treatments, where complex cross-effects between heterogeneous interventions are difficult to model. Existing methodologies remain constrained to binary or single-type treatments and suffer from restrictive assumptions, limited scalability, and inadequate evaluation frameworks for complex intervention scenarios. We present XTNet, a novel network architecture for multi-category, multi-valued treatment effect estimation. Our approach introduces a cross-effect estimation module with dynamic masking mechanisms to capture treatment interactions without restrictive structural assumptions. The architecture employs a decomposition strategy separating basic effects from cross-treatment interactions, enabling efficient modeling of combinatorial treatment spaces. We also propose MCMV-AUCC, a suitable evaluation metric that accounts for treatment costs and interaction effects. Extensive experiments on synthetic and real-world datasets demonstrate that XTNet consistently outperforms state-of-the-art baselines in both ranking accuracy and effect estimation quality. The results of the real-world A/B test further confirm its effectiveness.

因果推断多值处理交叉效应动态掩码

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。