glioAI

🧠 Automatic Brain Tumor Detection System Using DCNN

View the Project on GitHub ferasbg/glioAI

GlioAI: Automatic Brain Tumor Detection System

Automatic Brain Tumor Detection Using 2D Deep Convolutional Neural Network for Diffusion-Weighted MRI

Contents

Part I: Summary

Part II: Results

Part III: Conclusion and Future Work

Additional Documentation

Overview

GlioAI is an automatic brain cancer detection system that detects tumors in Head MRI scans.

Context

Objectives

Reduce Mortality Rates

Controlling Treatment Output

Scalability

Cost-Effective

Usability + Accessibility

Workflow

Synopsis

Back-End Design: Implement Convolutional Neural Network

We will be using a deep convolutional neural network, which is a neural network with a set of layers that will perform convolutions, pooling the set of regions of the image to extract features, along with with a softmax function that translates the last layer into a probability distribution.

ConvNet

Training Method

Dataset

The Brain MRI Images for Brain Tumor Detection was used to train the model which had 253 brain MRI scans.

The model was trained on 239 images belonging to two classes, and tested on 14.

Experiment and Results

Model and Training

The model consists of:

Model is trained on 25 epochs.

Models Accuracy Loss  
  Transfer Learning 95% 13%
  No Transfer Learning 76% 49%

Transfer Learning: Model Accuracy

Accuracy Graph

Transfer Learning: Loss Curve

Loss Curve Graph

No Transfer Learning: Loss Curve

Loss Curve Graph

No Transfer Learning: Model Accuracy

Accuracy Graph

Comparison of the Models

Visual Comparison

Evaluation

When comparing the results of the different models that were trained, it is clear that the transfer-learning based model is the most accurate deep learning model to deploy for the web app.

Conclusion

Feature Roadmap

App

Neural Network Architecture

Web Platform Engineering

Reflection

Given the current state that the model itself has been trained on a limited set(s) of patient MRI images with great accuracy, there is alot of area for improvement in terms of deploying extensive data augmentation (diversity of input image data for training), feature design, and overall application engineering and usability.

Future of GlioAI

Takeaway

The future of GlioAI will be a web platform that will allow doctors to recieve feedback from other verified doctors in order to make a far more efficient and accurate diagnosis in less than half the time.

GlioAI, the collaborative encyclopedic medical platform for doctors, built for the 21st century.

Bottleneck

Phase I: Build Crowdsourcing Protocols for Doctors in Need of Diagnostic Feedback

Phase II: Working With Tangible Atoms to Deploy Network for Shipping Treatments

Project OKRS

Dependencies

Deep Learning

Web Application

Links for Other Viewing Formats

References

Attribution

Icon by Srinivas Agra from thenounproject

Contributing

Contributions are always welcome! For bug reports or requests please submit an issue.

License

MIT