Learning Queueing Policies for Organ Transplantation Allocation using Interpretable Counterfactual Survival Analysis
Jeroen Berrevoets, Ahmed M. Alaa, Zhaozhi Qian, James Jordon, Alexander E. S. Gimson, Mihaela van der Schaar
Abstract
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.
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 04f1d630-90fb-4527-b8a7-f14031e94c8bCited by top-tier papers5
- NAISR: A 3D Neural Additive Model for Interpretable Shape RepresentationYining Jiao, Carlton J. Zdanski, Julia S. Kimbell, Andrew Prince et al.ICLR 2024 · 7 citations
- Differentiable and Transportable Structure LearningJeroen Berrevoets, Nabeel Seedat, Fergus Imrie, Mihaela van der SchaarICML 2023 · 4 citations
- Near-Optimal Dynamic Matching via Coarsening with Application to Heart TransplantationItai Zilberstein, Ioannis Anagnostides, Zachary Sollie, Arman Kilic et al.ICML 2026 · 2 citations
- Causal Explanation-Guided Learning for Organ AllocationAlessandro Marchese, Jeroen Berrevoets, Sam VerbovenNeurIPS 2025 · 1 citation
- Operator Theory-Driven Autoformulation of MDPs for Control of Queueing SystemsVictor Baillet, Yuanzhang Xiao, Nicolás Astorga, Mihaela van der SchaarICLR 2026
Builds on3
- Estimating counterfactual treatment outcomes over time through adversarially balanced representationsIoana Bica, Ahmed M. Alaa, James Jordon, Mihaela van der SchaarICLR 2020 · 224 citations
- Learning Disentangled Representations for CounterFactual RegressionNegar Hassanpour, Russell GreinerICLR 2020 · 176 citations
- Estimating the Effects of Continuous-valued Interventions using Generative Adversarial NetworksIoana Bica, James Jordon, Mihaela van der SchaarNeurIPS 2020 · 137 citations
Related papers
- OrganITE: Optimal transplant donor organ offering using an individual treatment effectJeroen Berrevoets, James Jordon, Ioana Bica, Alexander Gimson et al.NeurIPS 2020 · 51 citations
- Closing the loop in medical decision support by understanding clinical decision-making: A case study on organ transplantationYuchao Qin, Fergus Imrie, Alihan Hüyük, Daniel Jarrett et al.NeurIPS 2021 · 7 citations
- Counterbalancing Learning and Strategic Incentives in Allocation MarketsJamie Kang, Faidra Monachou, Moran Koren, Itai AshlagiNeurIPS 2021 · 1 citation
- Causal Modeling of Policy Interventions From Treatment-Outcome SequencesCaglar Hizli, S. T. John, Anne Tuulikki Juuti, Tuure Tapani Saarinen et al.ICML 2023 · 7 citations
- Improving the Generation and Evaluation of Synthetic Data for Downstream Medical Causal InferenceHarry Amad, Zhaozhi Qian, Dennis Frauen, Julianna Piskorz et al.NeurIPS 2025 · 6 citations
