keyboard_arrow_up
Deep Learning and Augmentation Architectures for Image Classification in Alzheimer's Diagnosis

Authors

Jiawei Zhang 1 , Xin Zhang 1 and Xinyin Miao 2 , 1 PRA Group (Nasdaq: PRAA), USA, 2 American Airlines Group Inc (Nasdaq: AAL), USA

Abstract

This paper utilizes four cutting edge deep learning architectures, namely VGG19, Xception, InceptionV3, and ResNet50, with transfer learning, image augmentation and two layers of regularization to be able to accurately predict the Alzheimer's Disease classes under 33,982 MRI images with a 0.9563 accuracy, 0.9972 roc_auc, and 0.9559 F1 score in the testing scenario. By investigating the internal neural network structures and comparing the prediction performance, it provides the insight of how various deep learning architectures work differently with corresponding deep learning structure layouts. The main contribution of this article also includes the study of image augmentation and deep learning regularization methodologies to enhance architecture performance and reduce transfer learning over-fitting in the Alzheimer's disease image-based classification.

Keywords

Deep learning, Transfer Learning, Image Classification, Neural Network Architecture, Regularization, Augmentation

Full Text  Volume 16, Number 6