Machine learning classification of mediastinal lymph node metastasis in NSCLC: a multicentre study in a Western European patient population

BACKGROUND: [(18)F] FDG PET-CT has an important role in the initial staging of lung cancer; however, accurate differentiation between activity in malignant and benign intrathoracic lymph nodes on PET-CT scans can be challenging. The purpose of the current study was to investigate the effect of incorporating primary tumour data and clinical features to differentiate between [(18)F] FDG-avid malignant and benign intrathoracic lymph nodes. METHODS: We retrospectively selected lung cancer patients who underwent PET-CT for initial staging in two centres in the Netherlands. The primary tumour and suspected lymph node metastases were annotated and cross-referenced with pathology results. Lymph nodes were classified as malignant or benign. From the image data, we extracted radiomic features and trained the classifier model using the extreme gradient boost (XGB) algorithm. Various scenarios were defined by selecting different combinations of data input and clinical features. Data from centre 1 were used for training and validation of the models using the XGB algorithm. To determine the performance of the model in a different hospital, the XGB model was tested using data from centre 2. RESULTS: Adding primary tumour data resulted in a significant gain in the performance of the trained classifier model. Adding the clinical information about distant metastases did not lead to significant improvement. The performance of the model in the test set (centre 2) was slightly but statistically significantly lower than in the validation set (centre 1). CONCLUSIONS: Using the XGB algorithm potentially leads to an improved model for the classification of intrathoracic lymph nodes. The inclusion of primary tumour data improved the performance of the model, while additional knowledge of distant metastases did not. In patients in whom metastases are limited to lymph nodes in the thorax, this may reduce costly and invasive procedures such as endobronchial ultrasound or mediastinoscopy procedures.