Skip to Main content Skip to Navigation
New interface
Conference papers

Uplift Modeling with Generalization Guarantees

Abstract : 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.
Document type :
Conference papers
Complete list of metadata
Contributor : Massih AMINI Connect in order to contact the contributor
Submitted on : Wednesday, June 23, 2021 - 10:45:44 AM
Last modification on : Wednesday, July 6, 2022 - 4:24:57 AM
Long-term archiving on: : Friday, September 24, 2021 - 6:22:41 PM


Files produced by the author(s)


  • 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⟩



Record views


Files downloads