Multi-Treatment Multi-Task Uplift Modeling for Enhancing User Growth
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
- 2017Attention Is All You Needcited by 60,361
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 5,045
- 2017An Overview of Multi-Task Learning in Deep Neural Networkscited by 803
- 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 600
- 2017Quasi-Oracle Estimation of Heterogeneous Treatment Effectscited by 332
- 2019Adapting Neural Networks for the Estimation of Treatment Effectscited by 287
- 2013Learning Factored Representations in a Deep Mixture of Expertscited by 278
- 2015Recursive Partitioning for Heterogeneous Causal Effectscited by 255
- 2020A Survey on Causal Inferencecited by 202
- 2020Learning to Branch for Multi-Task Learningcited by 115
10 more
- 2019Learning Counterfactual Representations for Estimating Individual Dose-Response Curvescited by 90
- 2021On Inductive Biases for Heterogeneous Treatment Effect Estimationcited by 75
- 2017Uplift Modeling with Multiple Treatments and General Response Typescited by 41
- 2020A unified survey of treatment effect heterogeneity modeling and uplift modelingcited by 36
- 2019Uplift Modeling for Multiple Treatments with Cost Optimizationcited by 25
- 2023Explicit Feature Interaction-aware Uplift Network for Online Marketingcited by 25
- 2024Entire Chain Uplift Modeling with Context-Enhanced Learning for Intelligent Marketingcited by 7
- 2024M3TN: Multi-gate Mixture-of-Experts based Multi-valued Treatment Network for Uplift Modelingcited by 6
- 2023Advances and Challenges of Multi-task Learning Method in Recommender System: A Surveycited by 4
- 2017A Practically Competitive and Provably Consistent Algorithm for Uplift Modelingcited by 3
Built on by
Citing papers per year: 2025: 2; 2026: 3.
5 papers.
- 2026Decoding ML Decision: An Agentic Reasoning Framework for Large-Scale Ranking Systemcited by 1
- 2025TSCAN: Context-Aware Uplift Modeling via Two-Stage Training for Online Merchant Business Diagnosiscited by 0
- 2025KLAN: Kuaishou Landing-page Adaptive Navigatorcited by 0
- 2026Advancing Edge Classification through High-Dimensional Causal Modeling of Node-Edge Interplaycited by 0
- 2026FunnelCausalNet: Funnel-aware Joint Conversion-Revenue Uplift for Multi-tier Coupon Allocationcited by 0
- 2026FunnelCausalNet: Funnel-aware Joint Conversion-Revenue Uplift for Multi-tier Coupon Allocation12 Aug 2026
- 2026Advancing Edge Classification through High-Dimensional Causal Modeling of Node-Edge Interplay1 May 2026
- 2026Decoding ML Decision: An Agentic Reasoning Framework for Large-Scale Ranking System20 Feb 2026
- 2025KLAN: Kuaishou Landing-page Adaptive Navigator31 Jul 2025
- 2025TSCAN: Context-Aware Uplift Modeling via Two-Stage Training for Online Merchant Business Diagnosis26 Apr 2025
Similar, not cited
Close in wording, no citation either way
- 2023KDSM: An uplift modeling framework based on knowledge distillation and sample matching
- 2025The Estimation of Continual Causal Effect for Dataset Shifting Streams
- 2025Guardrailed Uplift Targeting: A Causal Optimization Playbook for Marketing Strategy
- 2024Progress or Regress? Self-Improvement Reversal in Post-training
- 2025Deep Disentangled Representation Network for Treatment Effect Estimation
- 2023Towards Causal Foundation Model: on Duality between Causal Inference and Attention
- 2025Technical Report: Facilitating the Adoption of Causal Inference Methods Through LLM-Empowered Co-Pilot
- 2026Causal Effect Estimation with Latent Textual Treatments