Machine learning model for predicting excessive muscle loss during neoadjuvant chemoradiotherapy in oesophageal cancer.

BACKGROUND

Excessive skeletal muscle loss during neoadjuvant concurrent chemoradiotherapy (NACRT) is significantly related to survival outcomes of oesophageal cancer. However, the conventional method for measuring skeletal muscle mass requires computed tomography (CT) images, and the calculation process is labour-intensive. In this study, we built machine-learning models to predict excessive skeletal muscle loss, using only body mass index data and blood laboratory test results.

METHODS

We randomly split the data of 232 male patients treated with NACRT for oesophageal cancer into the training (70%) and test (30%) sets for 1000 iterations. The naive random over sampling method was applied to each training set to adjust for class imbalance, and we used seven different machine-learning algorithms to predict excessive skeletal muscle loss. We used five input variables, namely, relative change percentage in body mass index, albumin, prognostic nutritional index, neutrophil-to-lymphocyte ratio, and platelet-to-lymphocyte ratio over 50 days. According to our previous study results, which used the maximal χ2 method, 10.0% decrease of skeletal muscle index over 50 days was determined as the cut-off value to define the excessive skeletal muscle loss.

RESULTS

The five input variables were significantly different between the excessive and the non-excessive muscle loss group (all P < 0.001). None of the clinicopathologic variables differed significantly between the two groups. The ensemble model of logistic regression and support vector classifier showed the highest area under the curve value among all the other models [area under the curve = 0.808, 95% confidence interval (CI): 0.708-0.894]. The sensitivity and specificity of the ensemble model were 73.7% (95% CI: 52.6%-89.5%) and 74.5% (95% CI: 62.7%-86.3%), respectively.

CONCLUSIONS

Machine learning model using the ensemble of logistic regression and support vector classifier most effectively predicted the excessive muscle loss following NACRT in patients with oesophageal cancer. This model can easily screen the patients with excessive muscle loss who need an active intervention or timely care following NACRT.