Deep Jump Learning for Off-Policy Evaluation in Continuous Treatment SettingsDownload PDF

21 May 2021, 20:44 (edited 05 Nov 2021)NeurIPS 2021 PosterReaders: Everyone
  • Keywords: Statistical Learning, Off-Policy Evaluation, Deep Learning, Continuous Treatments, Multi-Scale Change Point Detection, Precision medicine
  • TL;DR: Develop deep jump learning for off-policy evaluation in continuous treatment settings, by adaptively discretizing the treatment space.
  • Abstract: We consider off-policy evaluation (OPE) in continuous treatment settings, such as personalized dose-finding. In OPE, one aims to estimate the mean outcome under a new treatment decision rule using historical data generated by a different decision rule. Most existing works on OPE focus on discrete treatment settings. To handle continuous treatments, we develop a novel estimation method for OPE using deep jump learning. The key ingredient of our method lies in adaptively discretizing the treatment space using deep discretization, by leveraging deep learning and multi-scale change point detection. This allows us to apply existing OPE methods in discrete treatments to handle continuous treatments. Our method is further justified by theoretical results, simulations, and a real application to Warfarin Dosing.
  • Supplementary Material: pdf
  • Code Of Conduct: I certify that all co-authors of this work have read and commit to adhering to the NeurIPS Statement on Ethics, Fairness, Inclusivity, and Code of Conduct.
  • Code: https://github.com/HengruiCai/DJL
15 Replies

Loading