UTBoost: Gradient Boosted Decision Trees for Uplift Modeling
Abstract
Uplift modeling comprises a collection of machine learning techniques designed for managers to predict the incremental impact of specific actions on customer outcomes. However, accurately estimating this incremental impact poses significant challenges due to the necessity of determining the difference between two mutually exclusive outcomes for each individual. In our study, we introduce two novel modifications to the established Gradient Boosting Decision Trees (GBDT) technique. These modifications sequentially learn the causal effect, addressing the counterfactual dilemma. Each modification innovates upon the existing technique in terms of the ensemble learning method and the learning objective, respectively. Experiments with large-scale datasets validate the effectiveness of our methods, consistently achieving substantial improvements over baseline models.
Keywords
Cite
@article{arxiv.2312.02573,
title = {UTBoost: Gradient Boosted Decision Trees for Uplift Modeling},
author = {Junjie Gao and Xiangyu Zheng and DongDong Wang and Zhixiang Huang and Bangqi Zheng and Kai Yang},
journal= {arXiv preprint arXiv:2312.02573},
year = {2025}
}
Comments
Accepted by PRICAI 2024