Brain MR image classification using two-dimensional discrete wavelet transform and AdaBoost with random forests

This paper presents an automated and accurate computer-aided diagnosis (CAD) system for brain magnetic resonance (MR) image classification. The system first utilizes two-dimensional discrete wavelet transform (2D DWT) for extracting features from the images. After feature vector normalization, probabilistic principal component analysis (PPCA) is employed to reduce the dimensionality of the feature vector. The reduced features are applied to the classifier to categorize MR images into normal and abnormal. This scheme uses an AdaBoost algorithm with random forests as its base classifier. Three benchmark MR image datasets, Dataset-66, Dataset-160, and Dataset-255, have been used to validate the proposed system. A 5×5-fold stratified cross validation scheme is used to enhance the generalization capability of the proposed scheme. Simulation results are compared with the existing schemes and it is observed that the proposed scheme outperforms others in all the three datasets.

[1]  Yudong Zhang,et al.  A Novel Method for Magnetic Resonance Brain Image Classification Based on Adaptive Chaotic PSO , 2010 .

[2]  David G. Stork,et al.  Pattern Classification , 1973 .

[3]  Yudong Zhang,et al.  An MR Brain Images Classifier System via Particle Swarm Optimization and Kernel Support Vector Machine , 2013, TheScientificWorldJournal.

[4]  Yudong Zhang,et al.  MAGNETIC RESONANCE BRAIN IMAGE CLASSIFICATION BY AN IMPROVED ARTIFICIAL BEE COLONY ALGORITHM , 2011 .

[5]  Amitava Chatterjee,et al.  A Slantlet transform based intelligent system for magnetic resonance brain image classification , 2006, Biomed. Signal Process. Control..

[6]  Catherine Westbrook,et al.  Handbook of MRI technique , 1994 .

[7]  Radford M. Neal.  Pattern Recognition and Machine Learning , 2007, Technometrics.

[8]  Michael E. Tipping,et al.  Probabilistic Principal Component Analysis , 1999 .

[9]  Emma Pickwell-MacPherson,et al.  Frequency-Wavelet Domain Deconvolution for terahertz reflection imaging and spectroscopy. , 2010, Optics express.

[10]  Martin Vetterli,et al.  Wavelets and filter banks: theory and design , 1992, IEEE Trans. Signal Process..

[11]  Paul K. Joseph,et al.  Classification of MRI brain images using combined wavelet entropy based spider web plots and probabilistic neural network , 2013, Pattern Recognit. Lett..

[12]  Yudong Zhang,et al.  Preclinical Diagnosis of Magnetic Resonance (MR) Brain Images via Discrete Wavelet Packet Transform with Tsallis Entropy and Generalized Eigenvalue Proximal Support Vector Machine (GEPSVM) , 2015, Entropy.

[13]  S. Mallat.  A wavelet tour of signal processing , 1998 .

[14]  Kenneth Revett,et al.  Computer-aided diagnosis of human brain tumor through MRI: A survey and a new algorithm , 2014, Expert Syst. Appl..

[15]  Leo Breiman,et al.  Random Forests , 2001, Machine Learning.

[16]  Yoav Freund,et al.  Experiments with a New Boosting Algorithm , 1996, ICML.

[17]  Sudeb Das,et al.  Brain Mr Image Classification Using Multiscale Geometric Analysis of Ripplet , 2013 .

[18]  Christopher M. Bishop,et al.  Pattern Recognition and Machine Learning (Information Science and Statistics) , 2006 .

[19]  Abdel-Badeeh M. Salem,et al.  Hybrid intelligent techniques for MRI brain images classification , 2010, Digit. Signal Process..

[20]  Yudong Zhang,et al.  Automated classification of brain images using wavelet-energy and biogeography-based optimization , 2016, Multimedia Tools and Applications.

[21]  Robert E. Schapire,et al.  A Brief Introduction to Boosting , 1999, IJCAI.

[22]  Lalit M. Patnaik,et al.  Classification of magnetic resonance brain images using wavelets as input to support vector machine and neural network , 2006, Biomed. Signal Process. Control..

[23]  Yudong Zhang,et al.  Feed‐forward neural network optimized by hybridization of PSO and ABC for abnormal brain detection , 2015, Int. J. Imaging Syst. Technol..

[24]  David G. Stork,et al.  Pattern Classification (2nd ed.) , 1999 .

[25]  Yudong Zhang,et al.  A hybrid method for MRI brain image classification , 2011, Expert Syst. Appl..

[26]  B. Cui,et al.  Wetland Degradation and Ecological Restoration , 2013, The Scientific World Journal.

[27]  Yudong Zhang,et al.  Pathological Brain Detection in Magnetic Resonance Imaging Scanning by Wavelet Entropy and Hybridization of Biogeography-based Optimization and Particle Swarm Optimization , 2015 .

[28]  Mahrokh G. Shayesteh,et al.  Robust algorithm for brain magnetic resonance image (MRI) classification based on GARCH variances series , 2013, Biomed. Signal Process. Control..

[30]  Huazhong Shu,et al.  Artifact Suppressed Dictionary Learning for Low-Dose CT Image Processing , 2014, IEEE Transactions on Medical Imaging.

[31]  Yudong Zhang,et al.  AN MR BRAIN IMAGES CLASSIFIER VIA PRINCIPAL COMPONENT ANALYSIS AND KERNEL SUPPORT , 2012 .