KDSM: An uplift modeling framework based on knowledge distillation and sample matching
On the map: Knowledge distillation → General knowledge distillation methods
Abstract (arXiv)
Uplift modeling aims to estimate the treatment effect on individuals, widely applied in the e-commerce platform to target persuadable customers and maximize the return of marketing activities. Among the existing uplift modeling methods, tree-based methods are adept at fitting increment and generalization, while neural-network-based models excel at predicting absolute value and precision, and these advantages have not been fully explored and combined. Also, the lack of counterfactual sample pairs is the root challenge in uplift modeling. In this paper, we proposed an uplift modeling framework based on Knowledge Distillation and Sample Matching (KDSM). The teacher model is the uplift decision tree (UpliftDT), whose structure is exploited to construct counterfactual sample pairs, and the pairwise incremental prediction is treated as another objective for the student model. Under the idea of multitask learning, the student model can achieve better performance on generalization and even surpass the teacher. Extensive offline experiments validate the universality of different combinations of teachers and student models and the superiority of KDSM measured against the baselines. In online A/B testing, the cost of each incremental room night is reduced by 6.5%.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in General knowledge distillation methods, one of 12 directions in Knowledge distillation. Builds on 16 of its 42 references that are in the map. First posted on arXiv 6 Mar 2023.
Builds on
16 of 42 references.
References · 16
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2016Learning without Forgettingcited by 2,496
- 2016Wide & Deep Learning for Recommender Systemscited by 747
- 2017DeepFM: A Factorization-Machine based Neural Network for CTR Predictioncited by 676
- 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 600
- 2017Deep Interest Network for Click-Through Rate Predictioncited by 544
- 2017Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learningcited by 478
- 2017Deep & Cross Network for Ad Click Predictionscited by 393
- 2017Quasi-Oracle Estimation of Heterogeneous Treatment Effectscited by 332
- 2018AutoInt: Automatic Feature Interaction Learning via Self-Attentive Neural Networkscited by 320
- 2018xDeepFM: Combining Explicit and Implicit Feature Interactions for Recommender Systemscited by 278
4 more
- 2020CausalML: Python Package for Causal Machine Learningcited by 74
- 2020A unified survey of treatment effect heterogeneity modeling and uplift modelingcited by 36
- 2021A Large Scale Benchmark for Individual Treatment Effect Prediction and Uplift Modelingcited by 12
- 2020A Practical Incremental Method to Train Deep CTR Modelscited by 10
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025The Estimation of Continual Causal Effect for Dataset Shifting Streams
- 2024Multi-Treatment Multi-Task Uplift Modeling for Enhancing User Growth
- 2025Guardrailed Uplift Targeting: A Causal Optimization Playbook for Marketing Strategy
- 2022Respecting Transfer Gap in Knowledge Distillation
- 2024M3TN: Multi-gate Mixture-of-Experts based Multi-valued Treatment Network for Uplift Modeling
- 2022Using Knowledge Distillation to improve interpretable models in a retail banking context
- 2022Parameter-Efficient and Student-Friendly Knowledge Distillation
- 2025Rethinking Decoupled Knowledge Distillation: A Predictive Distribution Perspective