Predicting graft and patient outcomes following kidney transplantation using interpretable machine learning models

Abstract

AbstractThe decision to accept a deceased donor organ offer for transplant, or wait for something potentially better in the future, can be challenging. Clinical decision support tools predicting transplant outcomes are lacking. This project uses interpretable methods to predict both graft failure and patient death using data from previously accepted kidney transplant offers. Using more than 25 years of transplant outcome data, we train and compare several survival analysis models in single risk settings. In addition, we use post hoc interpretability techniques to clinically validate these models. Neural networks show comparable performance to the Cox proportional hazard model, with concordance of 0.63 and 0.79 for prediction of graft failure and patient death, respectively. Donor and recipient ages, the number of mismatches at DR locus, dialysis type, and primary renal disease appear to be important features for transplant outcome prediction. Owing to their good predictive performance and the clinical relevance of their post hoc interpretation, neural networks represent a promising core component in the construction of future decision support systems for transplant offering.