Improve ROI with Causal Learning and Conformal Prediction
Meng Ai, Zhuo Chen, Jibin Wang, Jing Shang, Tao Tao, Zhen Li
Abstract
In the commercial sphere, such as operations and maintenance, advertising, and marketing recommendations, intelligent decision-making utilizing data mining and neural network technologies is crucial, especially in resource allocation to optimize ROI. This study delves into the Cost-aware Binary Treatment Assignment Problem (C-BTAP) across different industries, with a focus on the state-of-the-art Direct ROI Prediction (DRP) method. However, the DRP model confronts issues like covariate shift and insufficient training data, hindering its real-world effectiveness. Addressing these challenges is essential for ensuring dependable and robust predictions in varied operational contexts.
This paper presents a robust Direct ROI Prediction (rDRP) method, designed to address challenges in real-world deployment of neural network-based uplift models, particularly under conditions of covariate shift and insufficient training data. The rDRP method, enhancing the standard DRP model, does not alter the model's structure or require retraining. It utilizes conformal prediction and Monte Carlo dropout for interval estimation, adapting to model uncertainty and data distribution shifts. A heuristic calibration method, inspired by a Kaggle competition, combines point and interval estimates. The effectiveness of these approaches is validated through offline tests and online A/B tests in various settings, demonstrating significant improvements in target rewards compared to the state-of-the-art method.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext ace05cfa-739b-4de1-8731-cdcd464c60d4Builds on9
- Image-to-Image Regression with Distribution-Free Uncertainty Quantification and Applications in ImagingAnastasios N. Angelopoulos, Amit Pal Singh Kohli, Stephen Bates, Michael I. Jordan et al.ICML 2022 · 115 citations
- On Inductive Biases for Heterogeneous Treatment Effect EstimationAlicia Curth, Mihaela van der SchaarNeurIPS 2021 · 114 citations
- Decision-Focused Learning without Decision-Making: Learning Locally Optimized Decision LossesSanket Shah, Kai Wang, Bryan Wilder, Andrew Perrault et al.NeurIPS 2022 · 79 citations
- Decision-Focused Learning: Through the Lens of Learning to RankJayanta Mandi, Víctor Bucarey, Maxime Mulamba Ke Tchomba, Tias GunsICML 2022 · 73 citations
- Direct Heterogeneous Causal Learning for Resource Allocation Problems in MarketingHao Zhou, Shaoming Li, Guibin Jiang, Jiaqi Zheng et al.AAAI 2023 · 35 citations
Related papers
- Learning to Predict IR Drop with Effective Training for ReRAM-based Neural Network HardwareSugil Lee, Giju Jung, Mohammed E. Fouda, Jongeun Lee et al.DAC 2020 · 47 citations
- DDPO: Direct Dual Propensity Optimization for Post-Click Conversion Rate EstimationHongzu Su, Lichao Meng, Lei Zhu, Ke Lu et al.SIGIR 2024 · 6 citations
- Identifying Causal-Effect Inference Failure with Uncertainty-Aware ModelsAndrew Jesson, Sören Mindermann, Uri Shalit, Yarin GalNeurIPS 2020 · 85 citations
- Estimating Model Performance Under Covariate Shift Without LabelsJakub Bialek, Juhani Kivimäki, Wojtek Kuberski, Nikolaos PerrakisNeurIPS 2025 · 10 citations
- Bi-Level Decision-Focused Causal Learning for Large-Scale Marketing Optimization: Bridging Observational and Experimental DataShuli Zhang, Hao Zhou, Jiaqi Zheng, Guibin Jiang et al.NeurIPS 2025 · 2 citations
