Deep ensemble learning of sparse regression models for brain disease diagnosis.
Recent studies on brain imaging analysis witnessed the core roles of machine learning techniques in computer-assisted intervention for brain disease diagnosis. Of various machine-learning techniques, sparse regression models have proved their effectiveness in handling high-dimensional data but with a small number of training samples, especially in medical problems. In the meantime, deep learning methods have been making great successes by outperforming the state-of-the-art performances in various applications. In this paper, we propose a novel framework that combines the two conceptually different methods of sparse regression and deep learning for Alzheimer's disease/mild cognitive impairment diagnosis and prognosis. Specifically, we first train multiple sparse regression models, each of which is trained with different values of a regularization control parameter. Thus, our multiple sparse regression models potentially select different feature subsets from the original feature set; thereby they have different powers to predict the response values, i.e., clinical label and clinical scores in our work. By regarding the response values from our sparse regression models as target-level representations, we then build a deep convolutional neural network for clinical decision making, which thus we call 'Deep Ensemble Sparse Regression Network.' To our best knowledge, this is the first work that combines sparse regression models with deep neural network. In our experiments with the ADNI cohort, we validated the effectiveness of the proposed method by achieving the highest diagnostic accuracies in three classification tasks. We also rigorously analyzed our results and compared with the previous studies on the ADNI cohort in the literature.
Department of Brain and Cognitive Engineering, Korea University, Seoul 02841, Republic of Korea. Electronic address: hisuk@korea.ac.kr.
Heung-Il
HI
Suk
Department of Brain and Cognitive Engineering, Korea University, Seoul 02841, Republic of Korea.
Seong-Whan
SW
Lee
Department of Brain and Cognitive Engineering, Korea University, Seoul 02841, Republic of Korea; Biomedical Research Imaging Center and Department of Radiology, University of North Carolina at Chapel Hill, NC 27599, USA.
Dinggang
D
Shen
D000368:Aged; D000544:Alzheimer Disease / Q000000981:diagnostic imaging*; D001921:Brain / Q000000981:diagnostic imaging*; D000066491:Clinical Decision-Making; D060825:Cognitive Dysfunction / Q000000981:diagnostic imaging*; D005260:Female; D006801:Humans; D000069550:Machine Learning; D008279:Magnetic Resonance Imaging / Q000379:methods*; D008297:Male; D011379:Prognosis; D012044:Regression Analysis; D015203:Reproducibility of Results; D012680:Sensitivity and Specificity
Alzheimer’s disease; Convolutional neural network; Deep ensemble learning; Sparse regression model