mlsanityData to 14 Oct 2026

Multi-Treatment Multi-Task Uplift Modeling for Enhancing User Growth

arXiv 2408.12803 ↗First posted 23 Aug 2024cs.LG cs.AI cs.IRcited by 5

Not on the map. Why?

Abstract (arXiv)

As a key component in boosting online user growth, uplift modeling aims to measure individual user responses (e.g., whether to play the game) to various treatments, such as gaming bonuses, thereby enhancing business outcomes. However, previous research typically considers a single-task, single-treatment setting, where only one treatment exists and the overall treatment effect is measured by a single type of user response. In this paper, we propose a Multi-Treatment Multi-Task (MTMT) uplift network to estimate treatment effects in a multi-task scenario. We identify the multi-treatment problem as a causal inference problem with a tiered response, comprising a base effect (from offering a treatment) and an incremental effect (from offering a specific type of treatment), where the base effect can be numerically much larger than the incremental effect. Specifically, MTMT separately encodes user features and treatments. The user feature encoder uses a multi-gate mixture of experts (MMOE) network to encode relevant user features, explicitly learning inter-task relations. The resultant embeddings are used to measure natural responses per task. Furthermore, we introduce a treatment-user feature interaction module to model correlations between each treatment and user feature. Consequently, we separately measure the base and incremental treatment effect for each task based on the produced treatment-aware representations. Experimental results based on an offline public dataset and an online proprietary dataset demonstrate the effectiveness of MTMT in single/multi-treatment and single/multi-task settings. Additionally, MTMT has been deployed in our gaming platform to improve user experience.

Key facts. Cited by 5 papers in the map (data through 14 Oct 2026). Builds on 22 of its 38 references that are in the map. First posted on arXiv 23 Aug 2024.

Builds on

22 of 38 references.

References · 22

  1. 2017Attention Is All You Needcited by 60,361
  2. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  3. 2014Adam: A Method for Stochastic Optimizationcited by 37,240
  4. 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 5,045
  5. 2017An Overview of Multi-Task Learning in Deep Neural Networkscited by 803
  6. 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 600
  7. 2017Quasi-Oracle Estimation of Heterogeneous Treatment Effectscited by 332
  8. 2019Adapting Neural Networks for the Estimation of Treatment Effectscited by 287
  9. 2013Learning Factored Representations in a Deep Mixture of Expertscited by 278
  10. 2015Recursive Partitioning for Heterogeneous Causal Effectscited by 255
  11. 2020A Survey on Causal Inferencecited by 202
  12. 2020Learning to Branch for Multi-Task Learningcited by 115
10 more
  1. 2019Learning Counterfactual Representations for Estimating Individual Dose-Response Curvescited by 90
  2. 2021On Inductive Biases for Heterogeneous Treatment Effect Estimationcited by 75
  3. 2017Uplift Modeling with Multiple Treatments and General Response Typescited by 41
  4. 2020A unified survey of treatment effect heterogeneity modeling and uplift modelingcited by 36
  5. 2019Uplift Modeling for Multiple Treatments with Cost Optimizationcited by 25
  6. 2023Explicit Feature Interaction-aware Uplift Network for Online Marketingcited by 25
  7. 2024Entire Chain Uplift Modeling with Context-Enhanced Learning for Intelligent Marketingcited by 7
  8. 2024M3TN: Multi-gate Mixture-of-Experts based Multi-valued Treatment Network for Uplift Modelingcited by 6
  9. 2023Advances and Challenges of Multi-task Learning Method in Recommender System: A Surveycited by 4
  10. 2017A Practically Competitive and Provably Consistent Algorithm for Uplift Modelingcited by 3

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 2; 2026: 3.

5 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2023KDSM: An uplift modeling framework based on knowledge distillation and sample matching
  2. 2025The Estimation of Continual Causal Effect for Dataset Shifting Streams
  3. 2025Guardrailed Uplift Targeting: A Causal Optimization Playbook for Marketing Strategy
  4. 2024Progress or Regress? Self-Improvement Reversal in Post-training
  5. 2025Deep Disentangled Representation Network for Treatment Effect Estimation
  6. 2023Towards Causal Foundation Model: on Duality between Causal Inference and Attention
  7. 2025Technical Report: Facilitating the Adoption of Causal Inference Methods Through LLM-Empowered Co-Pilot
  8. 2026Causal Effect Estimation with Latent Textual Treatments