rootstrap-org/

Alzheimer-Classifier-Demo

A machine learning model for waste classification

The aim is to build a model for MRI classification that identifies among the different classes:

This machine learning model will help medical staff get a second opinion on whether a pacient MRI indicates the presence of Alzheimer's desease

The model was built using Monai MONAI is a freely available, community-supported, PyTorch-based framework for deep learning in healthcare imaging.

It has two main design goals:

To be approachable and rapidly productive To be also configurable.

This model was created in the spirit of combining the interesting worlds of neuroscience and machine learning. It can be used to quickly detect Alzheimer's or a Mild Cognitive Impairment in a pacients MRI and hence help medical staff.

This model has an 86% of accuracy. Although it is an outstanding result, his means that 13% of times the prediction is wrong. This is why its important to understand that this tool is not a real medical opinion and can not be used as a final diagnosis by any means. This project does not aim to replace medical staff in diagnosing Alzheimer's desease, instead it is a tool to help them to get a quick and accurate second opinion.

The MRI data was gathered from the Alzheimer’s Desease Neuroimaging Initiative (ADNI) Database. Afterwards, the data was splitted into one folder for each class.

The full dataset used consisted of 1614 NIfTI files and the model has been trained to classify MRI into 3 classes:

Split into train/test The data was then splitted using the 60% for training, 20% for validation and 20% for testing

You can find the code for training at train.ipynb Training the model using DenseNet121 adapted for 3D images.

After 100 epochs, the model reached an accuracy of 85.76%