Hierarchical Ensemble Learning for Alzheimer's Disease Classification

In this paper, we propose to tackle the problem of Alzheimer's Disease (AD) classification by a novel Hierarchical Ensemble Learning (HEL) framework. Given an MRI image of a subject, our method will divide it into multiple slices, and generate the classification result in a coarse-to-fine way: First, for each slice, multiple pre-trained deep neural networks are adopted to extract features, and classiflers trained with each type of these features are used to generate the coarse predictions; Second, we employ ensemble learning on the coarse results to generate a refined result for each slice; At last, the given subject is classified based on the refined results aggregated from all slices. Using pre-trained networks for feature extraction can reduce the computational costs of training significantly, and the ensemble of multiple features and predicted results from slices can increase the classification accuracy effectively. Hence, our method can achieve the balance between efficiency and effectiveness. Experimental results show that the HEL framework can obtain notable performance gains with respect to various features and classifiers.

[1]  Andrew Zisserman,et al.  Very Deep Convolutional Networks for Large-Scale Image Recognition , 2014, ICLR.

[2]  Antonio Criminisi,et al.  Decision Forests with Long-Range Spatial Context for Organ Localization in CT Volumes , 2009 .

[3]  Martin A. Lindquist,et al.  Detecting functional connectivity change points for single-subject fMRI data , 2013, Front. Comput. Neurosci..

[4]  Seong-Whan Lee,et al.  Deep sparse multitask learning for feature selection in Alzheimer ’ s disease diagnosis , 2015 .

[5]  Vladimir Fonov,et al.  Prediction of Alzheimer's disease in subjects with mild cognitive impairment from the ADNI cohort using patterns of cortical thinning , 2013, NeuroImage.

[6]  Peter A. Bandettini,et al.  Does feature selection improve classification accuracy? Impact of sample size and feature selection on classification using anatomical magnetic resonance images , 2012, NeuroImage.

[7]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[8]  Marie Chupin,et al.  Automatic classi fi cation of patients with Alzheimer ' s disease from structural MRI : A comparison of ten methods using the ADNI database , 2010 .

[9]  Heikki Huttunen,et al.  Machine learning framework for early MRI-based Alzheimer's conversion prediction in MCI subjects , 2015, NeuroImage.

[10]  Seong-Whan Lee,et al.  Hierarchical feature representation and multimodal fusion with deep learning for AD/MCI diagnosis , 2014, NeuroImage.

[11]  Sung Yong Shin,et al.  Individual subject classification for Alzheimer's disease based on incremental learning using a spatial frequency representation of cortical thickness data , 2012, NeuroImage.

[12]  Daoqiang Zhang,et al.  Tree-Guided Sparse Coding for Brain Disease Classification , 2012, MICCAI.

[13]  R. Polikar,et al.  Multimodal EEG, MRI and PET data fusion for Alzheimer's disease diagnosis , 2010, 2010 Annual International Conference of the IEEE Engineering in Medicine and Biology.

[14]  Geoffrey E. Hinton,et al.  ImageNet classification with deep convolutional neural networks , 2012, Commun. ACM.

[15]  Daoqiang Zhang,et al.  Hierarchical fusion of features and classifier decisions for Alzheimer's disease diagnosis , 2014, Human brain mapping.

[16]  Chih-Jen Lin,et al.  LIBLINEAR: A Library for Large Linear Classification , 2008, J. Mach. Learn. Res..

[17]  Vince D. Calhoun,et al.  Deep learning for neuroimaging: a validation study , 2013, Front. Neurosci..

[18]  Dumitru Erhan,et al.  Going deeper with convolutions , 2014, 2015 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[19]  Paul M. Thompson,et al.  Sparse reduced-rank regression detects genetic associations with voxel-wise longitudinal phenotypes in Alzheimer's disease , 2012, NeuroImage.

[20]  Giovanni Montana,et al.  Predicting Alzheimer's disease: a neuroimaging study with 3D convolutional neural networks , 2015, ICPRAM 2015.

[21]  Kathryn Ziegler-Graham,et al.  Forecasting the global burden of Alzheimer’s disease , 2007, Alzheimer's & Dementia.

[22]  A. V.DavidSánchez,et al.  Advanced support vector machines and kernel methods , 2003, Neurocomputing.

[23]  Dinggang Shen,et al.  A Robust Deep Model for Improved Classification of AD/MCI Patients , 2015, IEEE Journal of Biomedical and Health Informatics.

[24]  Hongming Zhou,et al.  Extreme Learning Machine for Regression and Multiclass Classification , 2012, IEEE Transactions on Systems, Man, and Cybernetics, Part B (Cybernetics).

[25]  Dinggang Shen,et al.  Deep Learning-Based Feature Representation for AD/MCI Classification , 2013, MICCAI.