Learning Queueing Policies for Organ Transplantation Allocation using Interpretable Counterfactual Survival Analysis

Organ transplantation is often the last resort for treating end-stage illnesses, but managing transplant wait-lists is challenging because of organ scarcity and the complexity of assessing donorrecipient compatibility. In this paper, we develop a data-driven model for (real-time) organ allocation using observational data for transplant outcomes. Our model integrates a queuing-theoretic framework with unsupervised learning to cluster the organs into “organ types”, and then construct priority queues (associated with each organ type) wherein incoming patients are assigned. To reason about organ allocations, the model uses synthetic controls to infer a patient’s survival outcomes under counterfactual allocations to the different organ types—the model is trained end-toend to optimize the trade-o↵ between patient waiting time and expected survival time. The usage of synthetic controls enable patient-level interpretations of allocation decisions that can be presented and understood by clinicians. We test our model on multiple data sets, and show that it outperforms other organ-allocation policies in terms of added life-years, and death count. Furthermore, we introduce a novel organ-allocation simulator to accurately test new policies.

[1]  H. Bismuth,et al.  What did the European Liver Transplant Registry bring to liver transplantation? , 2020, Transplant international : official journal of the European Society for Organ Transplantation.

[2]  George H. Chen.  Deep Kernel Survival Analysis and Subject-Specific Survival Time Prediction Intervals , 2020, MLHC.

[3]  Negar Hassanpour,et al.  Learning Disentangled Representations for CounterFactual Regression , 2020, ICLR.

[4]  M. Schaar,et al.  Estimating the Effects of Continuous-valued Interventions using Generative Adversarial Networks , 2020, NeurIPS.

[5]  A. Gimson.  Development of a UK liver transplantation selection and allocation scheme. , 2020, Current opinion in organ transplantation.

[6]  Mihaela van der Schaar,et al.  Estimating Counterfactual Treatment Outcomes over Time Through Adversarially Balanced Representations , 2020, ICLR.

[7]  Fredrik D. Johansson,et al.  Generalization Bounds and Representation Learning for Estimation of Potential Outcomes and Causal Effects , 2020, J. Mach. Learn. Res..

[8]  M. Schaar,et al.  Learning Overlapping Representations for the Estimation of Individualized Treatment Effects , 2020, AISTATS.

[9]  Jeroen Berrevoets,et al.  OrganITE: Optimal transplant donor organ offering using an individual treatment effect , 2020, NeurIPS.

[10]  Negar Hassanpour,et al.  CounterFactual Regression with Importance Sampling Weights , 2019, IJCAI.

[11]  Michèle Sebag,et al.  Multi-Domain Adversarial Learning , 2019, ICLR.

[12]  Cynthia Rudin,et al.  Stop explaining black box machine learning models for high stakes decisions and use interpretable models instead , 2018, Nature Machine Intelligence.

[13]  D. Tao,et al.  Deep Domain Generalization via Conditional Invariant Adversarial Networks , 2018, ECCV.

[14]  Ankur Teredesai,et al.  Interpretable Machine Learning in Healthcare , 2018, 2018 IEEE International Conference on Healthcare Informatics (ICHI).

[15]  Mihaela van der Schaar,et al.  GANITE: Estimation of Individualized Treatment Effects using Generative Adversarial Nets , 2018, ICLR.

[16]  Sanjay Mehrotra,et al.  LivSim: An Open-Source Simulation Software Platform for Community Research and Development for Liver Allocation Policies. , 2017, Transplantation.

[17]  Aidong Zhang,et al.  Representation Learning for Treatment Effect Estimation from Observational Data , 2018, NeurIPS.

[18]  Mihaela van der Schaar,et al.  Deep Counterfactual Networks with Propensity-Dropout , 2017, ArXiv.

[19]  Mihaela van der Schaar,et al.  Bayesian Inference of Individualized Treatment Effects using Multi-task Gaussian Processes , 2017, NIPS.

[20]  Been Kim,et al.  Towards A Rigorous Science of Interpretable Machine Learning , 2017, 1702.08608.

[21]  Mihaela van der Schaar,et al.  Personalized Donor-Recipient Matching for Organ Transplantation , 2016, AAAI.

[22]  Uri Shalit,et al.  Estimating individual treatment effect: generalization bounds and algorithms , 2016, ICML.

[23]  Ying Daisy Zhuo,et al.  Personalized Diabetes Management Using Electronic Medical Records , 2016, Diabetes Care.

[24]  Issa J Dahabreh,et al.  Using group data to treat individuals: understanding heterogeneous treatment effects in the age of precision medicine and patient-centred evidence. , 2016, International journal of epidemiology.

[25]  Uri Shalit,et al.  Learning Representations for Counterfactual Inference , 2016, ICML.

[26]  François Laviolette,et al.  Domain-Adversarial Training of Neural Networks , 2015, J. Mach. Learn. Res..

[27]  Susan Athey,et al.  Recursive partitioning for heterogeneous causal effects , 2015, Proceedings of the National Academy of Sciences.

[28]  Jens Hainmueller,et al.  Comparative Politics and the Synthetic Control Method , 2014 .

[29]  Rachel J. Johnson,et al.  Organ Donation and Transplantation in the UK—The Last Decade: A Report From the UK National Transplant Registry , 2014, Transplantation.

[30]  T. Therneau,et al.  Hyponatremia and mortality among patients on the liver-transplant waiting list. , 2008, The New England journal of medicine.

[31]  A. Burroughs,et al.  Selection of patients for liver transplantation and allocation of donated livers in the UK , 2007, Gut.

[32]  Keying Ye,et al.  Applied Bayesian Modeling and Causal Inference From Incomplete-Data Perspectives , 2005, Technometrics.

[33]  G. Imbens,et al.  The Propensity Score with Continuous Treatments , 2005 .

[34]  Robert M Merion,et al.  Simulating the Allocation of Organs for Transplantation , 2004, Health care management science.

[35]  G. Imbens,et al.  Estimation of Causal Effects using Propensity Score Weighting: An Application to Data on Right Heart Catheterization , 2001, Health Services and Outcomes Research Methodology.

[36]  Alberto Abadie,et al.  The Economic Costs of Conflict: A Case Study of the , 2003 .

[37]  P. Kamath,et al.  A model to predict poor survival in patients undergoing transjugular intrahepatic portosystemic shunts , 2000, Hepatology.

[38]  J. Cecka.  The UNOS Scientific Renal Transplant Registry--2000. , 2000, Clinical transplants.

[39]  R. Tibshirani.  Regression Shrinkage and Selection via the Lasso , 1996 .

[40]  J. Little.  A Proof for the Queuing Formula: L = λW , 1961 .