Multiclass imbalance learning: Improving classification of pediatric brain tumors from magnetic resonance spectroscopy

Magn Reson Med. 2017 Jun;77(6):2114-2124. doi: 10.1002/mrm.26318. Epub 2016 Jul 12.

Abstract

Purpose: Classification of pediatric brain tumors from 1 H-magnetic resonance spectroscopy (MRS) can aid diagnosis and management of brain tumors. However, varied incidence of the different tumor types leads to imbalanced class sizes and introduces difficulties in classifying rare tumor groups. This study assessed different imbalanced multiclass learning techniques and compared the use of complete spectra and quantified metabolite profiles for classification of three main childhood brain tumor types.

Methods: Single-voxel, Short echo time MRS data were collected from 90 patients with pilocytic astrocytoma (n = 42), medulloblastoma (n = 38), or ependymoma (n = 10). Both spectra and metabolite profiles were used to develop the learning algorithms. The borderline synthetic minority oversampling technique and AdaboostM1 were used to correct for the skewed distribution. Classifiers were trained using five different pattern recognition algorithms.

Results: Use of imbalanced learning techniques improved the balanced accuracy rate (BAR) of all classification methods (average BAR over all classification methods for spectra: oversampled data = 0.81, original = 0.63, P < 0.001; metabolite concentration: oversampled-data = 0.91, original = 0.75, P < 0.0001). Performance of all classifiers in discriminating ependymomas increased when oversampled data were used compared with original data for both complete spectra (F-measure P < 0.01) and metabolite profile (F-measure P < 0.001).

Conclusion: Imbalanced learning techniques improve the classification accuracy of childhood brain tumors from MRS where group sizes differ and facilitate the inclusion of rarer tumor types into clinical decision support systems. Magn Reson Med 77:2114-2124, 2017. © 2016 The Authors Magnetic Resonance in Medicine published by Wiley Periodicals, Inc. on behalf of International Society for Magnetic Resonance in Medicine. This is an open access article under the terms of the Creative Commons Attribution License, which permits use, distribution and reproduction in any medium, provided the original work is properly cited.

Keywords: MR spectroscopy; brain tumors; classification; diagnosis; imbalanced learning.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Algorithms*
  • Biomarkers, Tumor / metabolism*
  • Brain Neoplasms / diagnosis*
  • Brain Neoplasms / metabolism*
  • Diagnosis, Computer-Assisted / methods*
  • Female
  • Humans
  • Machine Learning*
  • Male
  • Middle Aged
  • Pattern Recognition, Automated / methods
  • Proton Magnetic Resonance Spectroscopy / methods*
  • Reproducibility of Results
  • Sensitivity and Specificity

Substances

  • Biomarkers, Tumor