A novel computer-aided diagnosis system for breast MRI based on feature selection and ensemble learning

• Published in: Computers in biology and medicine Vol. 83; pp. 157 - 165
• Main Authors: Lu, Wei, Li, Zhe, Chu, Jinghui
• Format: Journal Article
• Language: English
• Published: United States Elsevier Ltd 01.04.2017; Elsevier Limited
• Subjects: Accuracy; Algorithms; Artificial intelligence; Benign; Breast cancer; Breast Neoplasms - diagnostic imaging; Breast Neoplasms - pathology; Breasts; Classification; Classification systems; Computed tomography; Computer applications; Computer-aided diagnosis; Data mining; Data processing; Diagnosis; Ensemble learning; Feature extraction; Feature selection; Female; Humans; Image Interpretation, Computer-Assisted - methods; Image processing; Imaging techniques; Internal Medicine; Learning algorithms; Machine Learning; Magnetic resonance imaging; Magnetic Resonance Imaging - methods; Mammography; Medical diagnosis; Medical imaging; Medical science; Morphology; Mortality; MRI; Neural networks; NMR; Nuclear magnetic resonance; Other; Pattern recognition; Pattern Recognition, Automated - methods; Positron emission tomography; Reproducibility of Results; Researchers; Segmentation; Sensitivity and Specificity; Subspace methods; Subspaces; Texture; Tumors; Ultrasound; Feature selection; Breast cancer; Ensemble learning; MRI; Computer-aided diagnosis; Classification
• ISSN: 0010-4825; 1879-0534; 1879-0534
• DOI: 10.1016/j.compbiomed.2017.03.002

Breast cancer is a common cancer among women. With the development of modern medical science and information technology, medical imaging techniques have an increasingly important role in the early detection and diagnosis of breast cancer. In this paper, we propose an automated computer-aided diagnosis (CADx) framework for magnetic resonance imaging (MRI). The scheme consists of an ensemble of several machine learning-based techniques, including ensemble under-sampling (EUS) for imbalanced data processing, the Relief algorithm for feature selection, the subspace method for providing data diversity, and Adaboost for improving the performance of base classifiers. We extracted morphological, various texture, and Gabor features. To clarify the feature subsets' physical meaning, subspaces are built by combining morphological features with each kind of texture or Gabor feature. We tested our proposal using a manually segmented Region of Interest (ROI) data set, which contains 438 images of malignant tumors and 1898 images of normal tissues or benign tumors. Our proposal achieves an area under the ROC curve (AUC) value of 0.9617, which outperforms most other state-of-the-art breast MRI CADx systems. Compared with other methods, our proposal significantly reduces the false-positive classification rate. •The dimensionality of the features we use is larger than most state-of-the-art methods. Various features including morphological, gabor, and several types of texture features are extracted to make a comprehensive characterization of breast masses.•We select the optimal feature subset from the original feature set with Relief according to their type, which helps diminish the redundant and irrelevant features as well as take the physical meaning of features into consideration.•We propose a novel ensemble learning framework based on the combination of EUS, subspace and Adaboost, which helps alleviate the data imbalance problem and improve the overall classification accuracy of the whole CADx system.