Back

Machine learning analysis of the UNOS database fails to predict lung transplant outcomes

Luo, L.; Mozejko, M.; Markov, N.; Peltekian, A.; Mohsin, S.; Carns, M.; Cooper, P.; Lysne, J.; Joudi, A.; Betensley, A.; Bemiss, B. C.; Myers, C.; Bharat, A.; Tomic, R.; Arunachalam, A.; Szczurek, E.; Budinger, G. S.; Misharin, A. V.; Subramani, M. V.

2024-10-21 transplantation
10.1101/2024.10.19.24315817 medRxiv
Show abstract

Lung transplantation is a life-saving therapy for end-stage lung disease but has the poorest survival among solid organ transplants. We analyzed standardized electronic health record (EHR) data from the United Network for Organ Sharing (UNOS) to predict one-, three-, and five-year survival and favorable long-term outcomes post-lung transplant. We applied two multivariate machine learning approaches, XGBoost or a tabular BERT model called EHRFormer, to data from 43,869 first-time lung transplant recipients (1987-2022). XGBoost and EHRFormer identified features that align closely with established risk factors for worse outcomes such as length of index stay, recipient age, and creatinine at the time of transplant. We developed a simple perturbation method with EHRFormer to probe in silico multivariate interactions between features that influence model prediction. Despite their attention to known risk factors, machine learning applied to EHR data collected by UNOS poorly predict one-, three-, and five-year mortality after lung transplant.

Matching journals

The top 6 journals account for 50% of the predicted probability mass.

50% of probability mass above

"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.