Search papers, labs, and topics across Lattice.
This paper introduces the Cross-Head Attention Uplift Network (CHAUN) and the Robust Adversarial Inverse Propensity Score (RA-IPS) method to improve uplift modeling by addressing inter-group similarity and unobserved confounding. CHAUN utilizes shared feature embeddings and cross-head attention to enhance the modeling of treatment-specific and control-specific representations, while RA-IPS optimizes propensity weights to reduce bias in the absence of true propensity scores. Experiments show that CHAUN outperforms existing uplift models by up to 25.6% in QINI scores, with RA-IPS providing an additional 5.4% improvement under confounding conditions.
CHAUN achieves a remarkable 25.6% improvement in QINI scores, redefining the benchmarks for uplift modeling in the presence of unobserved confounding.
Uplift modeling, crucial for estimating individual treatment effects (ITE), faces dual challenges: flexibly leveraging inter-group similarity to enhance discriminative power and debiasing under unobserved confounding scenarios. In this paper, we propose the Cross-Head Attention Uplift Network (CHAUN) and Robust Adversarial Inverse Propensity Score (RA-IPS) method to address these limitations. CHAUN employs shared feature embeddings and cross-head attention mechanisms to dynamically integrate treatment-specific and control-specific representations, enhancing inter-group correlation modeling. Theoretically, we prove that access to the true propensity scores ensures ITE identifiability even with unobserved confounders. For practical scenarios lacking true propensity scores, RA-IPS adversarially optimizes propensity weights within constrained uncertainty sets to mitigate bias from unobserved variables. Experiments on public datasets (CRITEO-UPLIFT, LAZADA) and a production e-commerce dataset demonstrate CHAUN's superiority over state-of-the-art uplift models, achieving relative improvements of up to 25.6% in QINI scores. RA-IPS further enhances robustness, outperforming standard IPS by 5.4% under unobserved confounding. The results validate the effectiveness of our proposed methods in real-world causal inference tasks.
DAMO