Pruning deep neural networks for encoding and decoding the human connectome

The main focus of this project is to identify biomarkers of neurodegenerative disorders such as Alzheimer’s Disease (AD) and Parkinson’s Disease (PD) in functional Magnetic Resonance Imaging (fMRI) scans. Deep learning models can be used to encode the human functional connectome and classify between...

全面介紹

Saved in:
書目詳細資料
主要作者: Tang, Sean
其他作者: Jagath C Rajapakse
格式: Final Year Project
語言:English
出版: Nanyang Technological University 2021
主題:
在線閱讀:https://hdl.handle.net/10356/149048
標簽: 添加標簽
沒有標簽, 成為第一個標記此記錄!
機構: Nanyang Technological University
語言: English
實物特徵
總結:The main focus of this project is to identify biomarkers of neurodegenerative disorders such as Alzheimer’s Disease (AD) and Parkinson’s Disease (PD) in functional Magnetic Resonance Imaging (fMRI) scans. Deep learning models can be used to encode the human functional connectome and classify between healthy subjects and patients with diseases, followed by a decoding process to identify salient features used in the classification. However, fMRI datasets have much more features than data samples, causing models to overfit easily. Existing solutions involving pruning the neural network range from recursive feature elimination which is too slow to a one-shot pruning approach which prunes too harshly. Thus, this project will explore the viability of improved pruning methodologies to attain an improved, sparser architecture. This project also goes beyond existing work on pruning multi-layer perceptron (MLP) to propose pruning approach for convolutional neural network (CNN), which can take in dynamic functional connectivity (dFC) matrices, as well as graph convolutional network (GCN), which is a better fit for encoding functional connectomes. The pruning algorithms proposed can also generalise to non-neuroimaging datasets, which is demonstrated by applying them to datasets like MNIST, CIFAR-10 and the CORA dataset, suggesting applications beyond the initial scope defined by this project.