English

UTBoost: Gradient Boosted Decision Trees for Uplift Modeling

Machine Learning 2025-02-10 v2 Artificial Intelligence

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

R2 v1 2026-06-28T13:41:23.168Z