ABSTRACT
Objective To explore the establishment of a prognostic model based on machine learning algorithm to predict primary graft dysfunction (PGD) in patients with idiopathic pulmonary fibrosis (IPF) after lung transplantation. Methods Clinical data of 226 IPF patients who underwent lung transplantation were retrospectively analyzed. All patients were randomly divided into the training and test sets at a ratio of 7:3. Using regularized logistic regression, random forest, support vector machine and artificial neural network, the prognostic model was established through variable screening, model establishment and model optimization. The performance of this prognostic model was assessed by the area under the receiver operating characteristic curve (AUC), positive predictive value, negative predictive value and accuracy. Results Sixteen key features were selected for model establishment. The AUC of the four prognostic models all exceeded 0.7. DeLong and McNemar tests found no significant difference in the performance among different models (both P>0.05). Conclusions Based on four machine learning algorithms, the prognostic model for grade 3 PGD after lung transplantation is preliminarily established. The overall prediction performance of each model is similar, which may predict the risk of grade 3 PGD in IPF patients after lung transplantation.