An explainable AI approach to surgical and radiotherapy interventions for optimized treatment decision-making in early-stage non-small cell lung cancer.

Abstract

Background: For individual patients with early-stage non-small cell lung cancer (NSCLC), robust evidence to guide treatment selection between surgery and stereotactic body radiotherapy (SBRT) remains limited. This study aimed to develop machine learning-driven predictive models using the Surveillance, Epidemiology, and End Results (SEER) database to evaluate the efficacy of these treatments, thereby providing a data-driven foundation for personalized treatment decisions.

Methods: Stage I or IIA NSCLC patients diagnosed between 2012 and 2018 were identified from the SEER database. Six machine learning models, spanning from classical to advanced approaches, were employed to predict 1-, 3-, and 5-year survival, with their performance assessed using seven metrics. The SHAP (SHapley Additive exPlanations) interpretability method was employed to explain the optimal predictive model, focusing on analyzing the differences between surgical and radiotherapy treatments under various factors, providing valuable insights to optimizing treatment strategies. Patients diagnosed between 2019 and 2021 were selected as an external validation cohort to assess the generalizability and robustness of the previously developed models.

Results: A total of 26,566 patients were included in the training and internal testing cohort of the study. LightGBM (light gradient boosting machine) outperformed other models across most metrics for survival predictions. The SHAP interpretability analysis revealed that tumor location, tumor size, pathology, and treatment type were significant factors for 3- and 5-year predictions. Furthermore, at 3- and 5-year intervals, the efficacy of radiotherapy was comparable to surgery for left upper lobe tumors, while radiotherapy appeared slightly inferior to surgery for right lower lobe tumors. Meanwhile, for tumors <1.5 cm or 3.5-5 cm, lobectomy exhibited the best efficacy, while for tumors measuring 1.5-3.5 cm, the efficacy of lobectomy seemed to be slightly inferior to radiotherapy and sublobar resection. For adenocarcinoma and squamous cell carcinoma, radiotherapy and lobectomy could be regarded as the preferred treatment methods, respectively. Besides, for patients <45 or >75 years old, sublobar resection showed the best efficacy at the 5-year interval. The external validation cohort of 11,927 patients further confirmed the effectiveness of the models in predicting 1-, 3-, and 5-year survival outcomes, reinforcing their reliability and applicability in clinical decision-making.

Conclusions: This study provides valuable insights into treatment decision-making for stages I and IIA NSCLC. The LightGBM model is a reliable tool for survival prediction for early-stage NSCLC. By utilizing this model, it can be concluded that tumor location, tumor size, pathological type and age are vital factors significantly influencing the choice of treatment methods.