Improving Individual Brain Age Prediction Using an Ensemble Deep Learning Framework.

Autor: Kuo CY; Aging and Health Research Center, National Yang Ming Chiao Tung University, Taipei, Taiwan., Tai TM; NVIDIA AI Technology Center, NVIDIA, Taipei, Taiwan., Lee PL; Institute of Neuroscience, National Yang Ming Chiao Tung University, Taipei, Taiwan., Tseng CW; NVIDIA AI Technology Center, NVIDIA, Taipei, Taiwan., Chen CY; NVIDIA AI Technology Center, NVIDIA, Taipei, Taiwan., Chen LK; Aging and Health Research Center, National Yang Ming Chiao Tung University, Taipei, Taiwan.; Center for Geriatrics and Gerontology, Taipei Veterans General Hospital, Taipei, Taiwan., Lee CK; NVIDIA AI Technology Center, NVIDIA, Taipei, Taiwan., Chou KH; Institute of Neuroscience, National Yang Ming Chiao Tung University, Taipei, Taiwan.; Brain Research Center, National Yang Ming Chiao Tung University, Taipei, Taiwan., See S; NVIDIA AI Technology Center, NVIDIA, Taipei, Taiwan., Lin CP; Aging and Health Research Center, National Yang Ming Chiao Tung University, Taipei, Taiwan.; Institute of Neuroscience, National Yang Ming Chiao Tung University, Taipei, Taiwan.; Brain Research Center, National Yang Ming Chiao Tung University, Taipei, Taiwan.
Jazyk: angličtina
Zdroj: Frontiers in psychiatry [Front Psychiatry] 2021 Mar 23; Vol. 12, pp. 626677. Date of Electronic Publication: 2021 Mar 23 (Print Publication: 2021).
DOI: 10.3389/fpsyt.2021.626677
Abstrakt: Brain age is an imaging-based biomarker with excellent feasibility for characterizing individual brain health and may serve as a single quantitative index for clinical and domain-specific usage. Brain age has been successfully estimated using extensive neuroimaging data from healthy participants with various feature extraction and conventional machine learning (ML) approaches. Recently, several end-to-end deep learning (DL) analytical frameworks have been proposed as alternative approaches to predict individual brain age with higher accuracy. However, the optimal approach to select and assemble appropriate input feature sets for DL analytical frameworks remains to be determined. In the Predictive Analytics Competition 2019, we proposed a hierarchical analytical framework which first used ML algorithms to investigate the potential contribution of different input features for predicting individual brain age. The obtained information then served as a priori knowledge for determining the input feature sets of the final ensemble DL prediction model. Systematic evaluation revealed that ML approaches with multiple concurrent input features, including tissue volume and density, achieved higher prediction accuracy when compared with approaches with a single input feature set [Ridge regression: mean absolute error (MAE) = 4.51 years, R 2 = 0.88; support vector regression, MAE = 4.42 years, R 2 = 0.88]. Based on this evaluation, a final ensemble DL brain age prediction model integrating multiple feature sets was constructed with reasonable computation capacity and achieved higher prediction accuracy when compared with ML approaches in the training dataset (MAE = 3.77 years; R 2 = 0.90). Furthermore, the proposed ensemble DL brain age prediction model also demonstrated sufficient generalizability in the testing dataset (MAE = 3.33 years). In summary, this study provides initial evidence of how-to efficiency for integrating ML and advanced DL approaches into a unified analytical framework for predicting individual brain age with higher accuracy. With the increase in large open multiple-modality neuroimaging datasets, ensemble DL strategies with appropriate input feature sets serve as a candidate approach for predicting individual brain age in the future.
Competing Interests: The authors declare that the research was conducted in the absence of any commercial or financial relationships that could be construed as a potential conflict of interest.
(Copyright © 2021 Kuo, Tai, Lee, Tseng, Chen, Chen, Lee, Chou, See and Lin.)
Databáze: MEDLINE