The code in this repository implements a spectral graph convolutional network to train on structural magnetic resonance imaging data to predict amyloid-beta positivity, a hallmark pathology in Alzheimer's disease.
Additions to the SCGN:
-
Models to accept 3D tensors as input or 2D tensor input with part of its data appended to the fully connected layer
-
Early Stopping
-
Sensitiviy, specificity, postive predictive value, and negative predictive value
-
FC layer tensor embedding for TensorBoard visualization
Other stuff:
-
ADNI data processing and aggregation routines
-
Functions to shuffle and generate training, test, and validation sets
-
Script for running models for training and evaluation
python mci.py --help
evaluate.py
Clinical imaging: Structural T1-weighted MR Images. Parcellated into 86 regions to obtain volume data. AV45-PET exam results served as labels for training the model.
Clinical and demographic data also included as covariates.
[schematic here]
Input: 86 volume regions
Input: 86 volume regions at the begining; covariates appended to the FC layer
Input: 86 volume regions and covariates fed at the beginning
python mci.py -m3
-
Clone this repository.
git clone https://github.com/cysmnl/amyloid_graph cd amyloid_graph -
Install the dependencies. The code should run with TensorFlow 1.0 and newer.
pip install -r requirements.txt