Accurate prediction of postoperative recurrence is important for optimizing the treatment strategies for non-small cell lung cancer (NSCLC). Previous studies identified the PD-L1 expression in NSCLC as a risk factor for postoperative recurrence. This study aimed to examine the contribution of PD-L1 expression to predicting postoperative recurrence using machine learning. The clinical data of 647 patients with NSCLC who underwent surgical resection were collected and stratified into training (80%), validation (10%), and testing (10%) datasets. Machine learning models were trained on the training data using clinical parameters including PD-L1 expression. The top-performing model was assessed on the test data using the SHAP analysis and partial dependence plots to quantify the contribution of the PD-L1 expression. Multivariate Cox proportional hazards model was used to validate the association between PD-L1 expression and postoperative recurrence. The random forest model demonstrated the highest predictive performance with the SHAP analysis, highlighting PD-L1 expression as an important feature, and the multivariate Cox analysis indicated a significant increase in the risk of postoperative recurrence with each increment in PD-L1 expression. These findings suggest that variations in PD-L1 expression may provide valuable information for clinical decision-making regarding lung cancer treatment strategies.
Keywords: Machine learning; Multivariate Cox proportional hazard model; Non-small cell lung cancer; Random forest; Recurrence-free survival.
© 2024. The Author(s).