Pre-training 3D convolutional neural networks for prodromal alzheimer's disease classification

• Main Authors: Jiang, Hongchao, Miao, Chunyan
• Other Authors: School of Computer Science and Engineering
• Format: Conference or Workshop Item
• Language: English
• Published: 2022
• Subjects: Engineering::Computer science and engineering; Machine Learning; Alzheimer's Disease
• Online Access: https://hdl.handle.net/10356/159700
• Institution: Nanyang Technological University

Alzheimer's disease (AD) is a chronic neurodegenerative disease that causes cognitive deficits, which severely interfere with daily life. Convolutional Neural Networks (CNNs) have been used to analyze Medical Resonance Imaging (MRI) scans for the early detection of AD. Prior works have explored supervised pre-training, unsupervised pre-training, and joint training to improve the diagnostic accuracy of CNNs. However, there is no consensus on the best approach. We compare the different pre-training methods in a standardized setting. Our experiments find that supervised pre-training and joint training outperform unsupervised pre-training when data is extremely limited. With more data, unsupervised pre-training closes the performance gap and, in some cases, outperforms supervised pre-training and joint training. In addition, we propose a simple hybrid approach of unsupervised pre-training followed by joint training that achieves the best performance.