1. Brain Age Estimation From MRI Using Cascade Networks With Ranking Loss
- Author
-
Jian Cheng, Dacheng Tao, Hao Guan, Jiyang Jiang, Haogang Zhu, Wei Wen, Tao Liu, Ziyang Liu, and Zhenzhou Wu
- Subjects
FOS: Computer and information sciences ,Mean squared error ,Correlation coefficient ,Computer science ,Computer Vision and Pattern Recognition (cs.CV) ,Computer Science - Computer Vision and Pattern Recognition ,Neuroimaging ,Alzheimer Disease ,FOS: Electrical engineering, electronic engineering, information engineering ,medicine ,Humans ,Dementia ,Cognitive Dysfunction ,Electrical and Electronic Engineering ,Radiological and Ultrasound Technology ,business.industry ,Image and Video Processing (eess.IV) ,Brain ,Pattern recognition ,Electrical Engineering and Systems Science - Image and Video Processing ,medicine.disease ,Magnetic Resonance Imaging ,Computer Science Applications ,Support vector machine ,Ranking ,Feature (computer vision) ,Biomarker (medicine) ,Artificial intelligence ,business ,Software - Abstract
Chronological age of healthy people is able to be predicted accurately using deep neural networks from neuroimaging data, and the predicted brain age could serve as a biomarker for detecting aging-related diseases. In this paper, a novel 3D convolutional network, called two-stage-age-network (TSAN), is proposed to estimate brain age from T1-weighted MRI data. Compared with existing methods, TSAN has the following improvements. First, TSAN uses a two-stage cascade network architecture, where the first-stage network estimates a rough brain age, then the second-stage network estimates the brain age more accurately from the discretized brain age by the first-stage network. Second, to our knowledge, TSAN is the first work to apply novel ranking losses in brain age estimation, together with the traditional mean square error (MSE) loss. Third, densely connected paths are used to combine feature maps with different scales. The experiments with $6586$ MRIs showed that TSAN could provide accurate brain age estimation, yielding mean absolute error (MAE) of $2.428$ and Pearson's correlation coefficient (PCC) of $0.985$, between the estimated and chronological ages. Furthermore, using the brain age gap between brain age and chronological age as a biomarker, Alzheimer's disease (AD) and Mild Cognitive Impairment (MCI) can be distinguished from healthy control (HC) subjects by support vector machine (SVM). Classification AUC in AD/HC and MCI/HC was $0.904$ and $0.823$, respectively. It showed that brain age gap is an effective biomarker associated with risk of dementia, and has potential for early-stage dementia risk screening. The codes and trained models have been released on GitHub: https://github.com/Milan-BUAA/TSAN-brain-age-estimation., Comment: Accepted by IEEE transactions on Medical Imaging, 13 pages, 6 figures
- Published
- 2021
- Full Text
- View/download PDF