Metabolite selection for machine learning in childhood brain tumour classification

Abstract

MRS can provide high accuracy in the diagnosis of childhood brain tumours when combined with machine learning. A feature selection method such as principal component analysis is commonly used to reduce the dimensionality of metabolite profiles prior to classification. However, an alternative approach of identifying the optimal set of metabolites has not been fully evaluated, possibly due to the challenges of defining this for a multi-class problem. This study aims to investigate metabolite selection from in vivo MRS for childhood brain tumour classification. Multi-site 1.5 T and 3 T cohorts of patients with a brain tumour and histological diagnosis of ependymoma, medulloblastoma and pilocytic astrocytoma were retrospectively evaluated. Dimensionality reduction was undertaken by selecting metabolite concentrations through multi-class receiver operating characteristics and compared with principal component analysis. Classification accuracy was determined through leave-one-out and k-fold cross-validation. Metabolites identified as crucial in tumour classification include myo-inositol (P 1H-MRS through support vector machine and 75% for 3 T 1H-MRS through linear discriminant analysis after oversampling the minority. The study suggests that a group of crucial metabolites helps to achieve better discrimination between childhood brain tumours.Copyright © 2022 The Authors. NMR in Biomedicine published by John Wiley & Sons Ltd.