A Prediction Model for Tacrolimus Daily Dose in Kidney Transplant Recipients With Machine Learning and Deep Learning Techniques

Abstract

Tacrolimus is a major immunosuppressor against post-transplant rejection in kidney transplant recipients. However, the narrow therapeutic index of tacrolimus and considerable variability among individuals are challenges for therapeutic outcomes. The aim of this study was to compare different machine learning and deep learning algorithms and establish individualized dose prediction models by using the best performing algorithm. Therefore, among the 10 commonly used algorithms we compared, the TabNet algorithm outperformed other algorithms with the highest R 2 (0.824), the lowest prediction error [mean absolute error (MAE) 0.468, mean square error (MSE) 0.558, and root mean square error (RMSE) 0.745], and good performance of overestimated (5.29%) or underestimated dose percentage (8.52%). In the final prediction model, the last tacrolimus daily dose, the last tacrolimus therapeutic drug monitoring value, time after transplantation, hematocrit, serum creatinine, aspartate aminotransferase, weight, CYP3A5 , body mass index, and uric acid were the most influential variables on tacrolimus daily dose. Our study provides a reference for the application of deep learning technique in tacrolimus dose estimation, and the TabNet model with desirable predictive performance is expected to be expanded and applied in future clinical practice.