Proceedings

Off-Policy Evaluation and Learning for Survival Outcomes Under Censoring

Download BibTeX Cite

Proceedings of the Thirty-Fifth International Joint Conference on Artificial Intelligence. Main Track. Pages 2273-2281.

https://doi.org/10.24963/ijcai.2026/253

Abstract

Optimizing survival outcomes, such as patient survival or customer retention, is a critical objective in data-driven decision-making. Off-Policy Evaluation (OPE) provides a powerful framework for assessing such decision-making policies using logged data alone, without the need for costly or risky online experiments in high-stakes applications. However, typical estimators are not designed to handle right-censored survival outcomes, as they ignore unobserved survival times beyond the censoring time, leading to systematic underestimation of the true policy performance. To address this issue, we propose a novel framework for OPE and Off-Policy Learning (OPL) tailored for survival outcomes under censoring. Specifically, we introduce IPCW-IPS and IPCW-DR, which employ the Inverse Probability of Censoring Weighting technique to explicitly deal with censoring bias. We theoretically establish that our estimators are unbiased and that IPCW-DR achieves double robustness, ensuring consistency if either the propensity score or the outcome model is correct. Furthermore, we extend this framework to constrained OPL to optimize policy value under budget constraints. We demonstrate the effectiveness of our proposed methods through simulation studies and illustrate their practical impacts using public real-world data for both evaluation and learning tasks.

  1. [Ai et al., 2022] Meng Ai, Biao Li, Heyang Gong, Qingwei Yu, Shengjie Xue, Yuan Zhang, Yunzhou Zhang, and Peng Jiang. LBCF: A large-scale budget-constrained causal for- est algorithm. In Proceedings of the ACM Web Conference 2022, WWW '22, page 2310-2319, New York, NY, USA, 2022. Association for Computing Machinery.
  2. [Ascarza et al., 2018] Eva Ascarza, Scott A. Neslin, Oded Netzer, et al. In pursuit of enhanced customer retention management: Review, key issues, and future directions. Cust. Need. and Solut., 5:65-81, 2018.
  3. [Beygelzimer and Langford, 2009] Alina Beygelzimer and John Langford. The offset tree for learning with partial labels. In Proceedings of the ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, KDD '09, page 129-138, New York, NY, USA, 2009. As- sociation for Computing Machinery.
  4. [Cai et al., 2023] Tianchi Cai, Jiyan Jiang, Wenpeng Zhang, Shiji Zhou, Xierui Song, Li Yu, Lihong Gu, Xiaodong Zeng, Jinjie Gu, and Guannan Zhang. Marketing bud- get allocation with offline constrained deep reinforcement learning. In Proceedings of the ACM International Con- ference on Web Search and Data Mining, WSDM '23, page 186-194, New York, NY, USA, 2023. Association for Computing Machinery.
  5. [Chu et al., 2023] Jianing Chu, Shu Yang, and Wenbin Lu. Multiply robust off-policy evaluation and learning under truncation by death. In Proceedings of the International Conference on Machine Learning, ICML'23. JMLR.org, 2023.
  6. [Cox, 1972] David R. Cox. Regression models and life- tables. Journal of the Royal Statistical Society, Series B, 34(2):187-220, 1972.
  7. [Du et al., 2019] Shuyang Du, James Lee, and Farzin Ghaf- farizadeh. Improve user retention with causal learning. In Proceedings of Machine Learning Research, volume 104 of Proceedings of Machine Learning Research, pages 34- 49. PMLR, 05 Aug 2019.
  8. [Dudík et al., 2011] Miroslav Dudík, John Langford, and Li- hong Li. Doubly robust policy evaluation and learning. In Proceedings of the International Conference on Ma- chine Learning, ICML'11, page 1097-1104, Madison, WI, USA, 2011. Omnipress.
  9. [Fleming and Harrington, 2013] Thomas R Fleming and David P Harrington. Counting processes and survival analysis. John Wiley & Sons, 2013.
  10. [Geng et al., 2015] Yuan Geng, Hao Helen Zhang, and Wen- bin Lu. On optimal treatment regimes selection for mean survival time. Statistics in medicine, 34(7):1169-1184, 2015.
  11. [Geoffrion, 1974] Arthur M. Geoffrion. Lagrangean relax- ation for integer programming, pages 82-114. Springer Berlin Heidelberg, Berlin, Heidelberg, 1974.
  12. [Glorot and Bengio, 2010] Xavier Glorot and Yoshua Ben- gio. Understanding the difficulty of training deep feedfor- ward neural networks. In Yee Whye Teh and Mike Titter- ington, editors, Proceedings of the International Confer- ence on Artificial Intelligence and Statistics, volume 9 of Proceedings of Machine Learning Research, pages 249- 256, Chia Laguna Resort, Sardinia, Italy, 13-15 May 2010. PMLR.
  13. [Goodfellow et al., 2016] Ian Goodfellow, Yoshua Bengio, and Aaron Courville. Deep learning, volume 1. MIT press Cambridge, 2016.
  14. [Hao et al., 2020] Xiaotian Hao, Zhaoqing Peng, Yi Ma, Guan Wang, Junqi Jin, Jianye Hao, Shan Chen, Rongquan Bai, Mingzhou Xie, Miao Xu, Zhenzhe Zheng, Chuan Yu, Han Li, Jian Xu, and Kun Gai. Dynamic knapsack opti- mization towards efficient multi-channel sequential adver- tising. In Proceedings of the International Conference on Machine Learning, ICML'20. JMLR.org, 2020.
  15. [Horvitz and Thompson, 1952] Daniel G. Horvitz and Donovan J. Thompson. A generalization of sampling without replacement from a finite universe. Journal of the American Statistical Association, 47(260):663-685, 1952.
  16. [Hosmer Jr et al., 2013] David W Hosmer Jr, Stanley Lemeshow, and Rodney X Sturdivant. Applied logistic regression. John Wiley & Sons, 2013.
  17. [Huang and Jiang, 2020] Jiawei Huang and Nan Jiang. From importance sampling to doubly robust policy gradient. In Proceedings of the International Conference on Machine Learning, pages 4434-4443. PMLR, 2020.
  18. [Ishwaran et al., 2008] Hemant Ishwaran, Udaya B. Kogalur, Eugene H. Blackstone, and Michael S. Lauer. Random survival forests. The Annals of Applied Statistics, 2(3):841 -860, 2008.
  19. [Jenkins, 2005] Stephen P Jenkins. Survival analysis. Un- published manuscript, Institute for Social and Economic Research, University of Essex, Colchester, UK, 42:54-56, 2005.
  20. [Jeunen and Goethals, 2021] Olivier Jeunen and Bart Goethals. Pessimistic reward models for off-policy learning in recommendation. In Proceedings of the 15th ACM Conference on Recommender Systems, pages 63-74, 2021.
  21. [Jiang et al., 2017] Runchao Jiang, Wenbin Lu, Rui Song, and Marie Davidian. On estimation of optimal treatment regimes for maximizing t-year survival probability. Jour- nal of the Royal Statistical Society, Series B, 79(4):1165- 1185, sep 2017.
  22. [Kaplan and Meier, 1958] Edward L. Kaplan and Paul Meier. Nonparametric estimation from incomplete obser- vations. Journal of the American Statistical Association, 53(282):457-481, 1958.
  23. [Katzman et al., 2018] Jared L Katzman, Uri Shaham, Alexander Cloninger, Jonathan Bates, Tingting Jiang, and Yuval Kluger. Deepsurv: personalized treatment recom- mender system using a cox proportional hazards deep neural network. BMC Medical Research Methodology, 18(1):24, 2018.
  24. [Kingma and Ba, 2017] Diederik P. Kingma and Jimmy Ba. Adam: A method for stochastic optimization, 2017. [Kiyohara et al., 2024]
  25. Haruka Kiyohara, Masahiro No- mura, and Yuta Saito. Off-policy evaluation of slate bandit policies via optimizing abstraction. In Proceedings of the ACM Web Conference 2024, WWW '24, page 3150-3161, New York, NY, USA, 2024. Association for Computing Machinery.
  26. [LeCun et al., 2015] Yann LeCun, Yoshua Bengio, and Ge- offrey Hinton. Deep learning. nature, 521(7553):436-444, 2015.
  27. [Lee et al., 2018] Changhee Lee, William Zame, Jinsung Yoon, and Mihaela van der Schaar. Deephit: A deep learn- ing approach to survival analysis with competing risks. In Proceedings of the AAAI Conference on Artificial Intelli- gence, volume 32, 2018.
  28. [Murphy, 2003] Susan A. Murphy. Optimal dynamic treat- ment regimes. Journal of the Royal Statistical Society, Se- ries B, 65(2):331-355, 04 2003.
  29. [Nair and Hinton, 2010] Vinod Nair and Geoffrey E. Hinton. Rectified linear units improve restricted boltzmann ma- chines. In Proceedings of the International Conference on Machine Learning, ICML'10, page 807-814, Madison, WI, USA, 2010. Omnipress.
  30. [Peng et al., 2023] Jie Peng, Hao Zou, Jiashuo Liu, Shaom- ing Li, Yibao Jiang, Jian Pei, and Peng Cui. Offline policy evaluation in large action spaces via outcome-oriented ac- tion grouping. In Proceedings of the ACM Web Conference 2023, WWW '23, page 1220-1230, New York, NY, USA, 2023. Association for Computing Machinery.
  31. [Robins and Finkelstein, 2000] James M. Robins and Di- anne M. Finkelstein. Correcting for noncompliance and dependent censoring in an aids clinical trial with inverse probability of censoring weighted (ipcw) log-rank tests. Biometrics, 56(3):779-788, 2000.
  32. [Robins and Rotnitzky, 1995] James M. Robins and Andrea Rotnitzky. Semiparametric efficiency in multivariate re- gression models with missing data. Journal of the Ameri- can Statistical Association, 90(429):122-129, 1995.
  33. [Robins et al., 1995] James M. Robins, Andrea Rotnitzky, and Lue Ping Zhao. Analysis of semiparametric regres- sion models for repeated outcomes in the presence of miss- ing data. Journal of the American Statistical Association, 90(429):106-121, 1995.
  34. [Rosenbaum and Rubin, 1983] Paul R. Rosenbaum and Donald B. Rubin. The central role of the propensity score in observational studies for causal effects. Biometrika, 70(1):41-55, 04 1983.
  35. [Sachdeva et al., 2020] Noveen Sachdeva, Yi Su, and Thorsten Joachims. Off-policy bandits with deficient sup- port. In Proceedings of the ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, pages 965-975, 2020.
  36. [Saito and Joachims, 2021] Yuta Saito and Thorsten Joachims. Counterfactual learning and evaluation for recommender systems: Foundations, implementations, and recent advances. In Proceedings of the 15th ACM Conference on Recommender Systems, RecSys '21, page 828-830, New York, NY, USA, 2021. Association for Computing Machinery.
  37. [Saito et al., 2024] Yuta Saito, Jihan Yao, and Thorsten Joachims. POTEC: Off-policy learning for large action spaces via two-stage policy decomposition. arXiv preprint arXiv:2402.06151, 2024.
  38. [Uno et al., 2014] Hajime Uno, Brian Claggett, Lu Tian, Eisuke Inoue, Paul Gallo, Toshio Miyata, Deborah Schrag, Masahiro Takeuchi, Yoshiaki Uyama, Lihui Zhao, Hicham Skali, Scott Solomon, Susanna Jacobus, Michael Hughes, Milton Packer, and Lee-Jen Wei. Moving beyond the hazard ratio in quantifying the between-group differ- ence in survival analysis. Journal of Clinical Oncology, 32(22):2380-2385, 2014. PMID: 24982461.
  39. [Wang et al., 2017] Yu-Xiang Wang, Alekh Agarwal, and Miroslav Dudík. Optimal and adaptive off-policy evalua- tion in contextual bandits. In Doina Precup and Yee Whye Teh, editors, Proceedings of the International Conference on Machine Learning, volume 70 of Proceedings of Ma- chine Learning Research, pages 3589-3597. PMLR, 06- 11 Aug 2017.
  40. [Wang et al., 2025] Han Wang, Yang Xu, Wenbin Lu, and Rui Song. Off-policy evaluation under nonignorable miss- ing data. In Aarti Singh, Maryam Fazel, Daniel Hsu, Si- mon Lacoste-Julien, Felix Berkenkamp, Tegan Maharaj, Kiri Wagstaff, and Jerry Zhu, editors, Proceedings of the International Conference on Machine Learning, volume 267 of Proceedings of Machine Learning Research, pages 65020-65058. PMLR, 13-19 Jul 2025.
  41. [Zhang et al., 2021] Yang Zhang, Bo Tang, Qingyu Yang, Dou An, Hongyin Tang, Chenyang Xi, Xueying Li, and Feiyu Xiong. BCORLE(ς): an offline reinforcement learning and evaluation framework for coupons allocation in e-commerce market. In Advances in Neural Informa- tion Processing Systems, NIPS '21, Red Hook, NY, USA, 2021.
  42. [Curran Associates Inc. Zhao et al., 2019] Kui Zhao, Junhao Hua, Ling Yan, Qi Zhang, Huan Xu, and Cheng Yang. A unified framework for marketing budget allocation. In Proceed- ings of the ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, KDD '19, page 1820-1830, New York, NY, USA, 2019. Association for Computing Machinery.
  43. [Zhou et al., 2023] Hao Zhou, Shaoming Li, Guibin Jiang, Jiaqi Zheng, and Dong Wang. Direct heterogeneous causal learning for resource allocation problems in marketing. In Proceedings of the AAAI Conference on Artificial Intelli- gence, AAAI'23/IAAI'23/EAAI'23. AAAI Press, 2023.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards