Machine learning-based MRI texture analysis to predict occult lymph node metastasis in early-stage oral tongue squamous cell carcinoma

Abstract

Objectives: To develop and compare several machine learning models to predict occult cervical lymph node (LN) metastasis in early-stage oral tongue squamous cell cancer (OTSCC) from preoperative MRI texture features.

Materials and methods: We retrospectively enrolled 116 patients with early-stage OTSCC (cT1-2N0) who had been surgically treated by tumor excision and elective neck dissection (END). For each patient, we extracted 86 texture features from T2-weighted imaging (T2WI) and contrast-enhanced T1-weighted imaging (ceT1WI), respectively. Dimension reduction was performed in three consecutive steps: reproducibility analysis, collinearity analysis, and information gain algorithm. Models were created using six machine learning methods, including logistic regression (LR), random forest (RF), naïve Bayes (NB), support vector machine (SVM), AdaBoost, and neural network (NN). Their performance was assessed using tenfold cross-validation.

Results: Occult LN metastasis was pathologically detected in 42.2% (49/116) of the patients. No significant association was identified between node status and patients' gender, age, or clinical T stage. Dimension reduction steps selected 6 texture features. The NB model gave the best overall performance, which correctly classified the nodal status in 74.1% (86/116) of the carcinomas, with an AUC of 0.802.

Conclusion: Machine learning-based MRI texture analysis offers a feasible tool for preoperative prediction of occult cervical node metastasis in early-stage OTSCC.

Key points: • A machine learning-based MRI texture analysis approach was adopted to predict occult cervical node metastasis in early-stage OTSCC with no evidence of node involvement on conventional images. • Six texture features from T2WI and ceT1WI of preoperative MRI were selected to construct the predictive model. • After comparing six machine learning methods, naïve Bayes (NB) achieved the best performance by correctly identifying the node status in 74.1% of the patients, using tenfold cross-validation.

Keywords: Computer-assisted diagnosis; Lymphatic metastasis; Machine learning; Magnetic resonance imaging; Squamous cell carcinoma of head and neck.