Radiomics-based prediction of recurrence for head and neck cancer patients using data imbalanced correction

To propose a radiomics-based prediction model for head and neck squamous cell carcinoma (HSNCC) recurrence after radiation therapy using a novel data imbalance correction method known as Gaussian noise upsampling (GNUS).

The dataset includes 97 HNSCC patients treated with definitive radiotherapy alone or concurrent chemoradiotherapy at two institutions. We performed radiomics analysis using nine segmentations created on pretreatment positron emission tomography and computed tomography images. Feature selection was performed by the least absolute shrinkage and selection operator analysis via five-fold cross-validation. The proposed GNUS was compared with seven conventional data-imbalance correction methods. Classification models of HNSCC recurrence were constructed on oversampled features using the machine learning algorithms of linear regression. Their predictive performance was evaluated based on accuracy, sensitivity, specificity, and the area under the curve (AUC) of the receiver operating performance characteristic curve via five-fold cross-validation using the same combinations as for feature selection.

The prediction model without data imbalance correction shows sensitivity, specificity, accuracy, and AUC values of 83 %, 96 %, 92 %, and 0.96, respectively. The conventional model with the best performance is the random over-sampler model, which shows sensitivity, specificity, accuracy, and AUC values of 93 %, 91 %, 92 %, 0.97, respectively, whereas the GNUS model shows values of 93 %, 94 %, 94 %, 0.98, respectively.

Oversampling methods can reduce sensitivity and specificity bias. The proposed GNUS can improve accuracy as well as reduce sensitivity and specificity bias.