Uplift Modeling with Generalization Guarantees - Archive ouverte HAL Access content directly
Conference Papers Year : 2021

Uplift Modeling with Generalization Guarantees


In this paper, we consider the task of ranking individuals based on the potential benefit of being "treated" (e.g. by a drug or exposure to recommendations or ads), referred to as Uplift Modeling in the literature. This application has gained a surge of interest in recent years and it is found in many applications such as personalized medicine, recommender systems or targeted advertising. In real life scenarios the capacity of models to rank individuals by potential benefit is measured by the Area Under the Uplift Curve (AUUC), a ranking metric related to the well known Area Under ROC Curve. In the case where the objective function, for learning model parameters, is different from AUUC, the capacity of the resulting system to generalize on AUUC is limited. To tackle this issue, we propose to learn a model that directly optimizes an upper bound on AUUC. To find such a model we first develop a generalization bound on AUUC and then derive from it a learning objective called AUUC-max, usable with linear and deep models. We empirically study the tightness of this generalization bound, its effectiveness for hyperparameters tuning and show the efficiency of the proposed learning objective compared to a wide range of competitive baselines on two classical uplift modeling benchmarks using real-world datasets.
Fichier principal
Vignette du fichier
kdd_artem.pdf (844.73 Ko) Télécharger le fichier
Origin : Files produced by the author(s)

Dates and versions

hal-03268412 , version 1 (23-06-2021)


  • HAL Id : hal-03268412 , version 1


Artem Betlei, Eustache Diemert, Massih-Reza Amini. Uplift Modeling with Generalization Guarantees. Special Interest Group on Knowledge Discovery and Data Mining (SIGKDD), Aug 2021, Singapore, Singapore. ⟨hal-03268412⟩
240 View
1311 Download


Gmail Facebook X LinkedIn More