jzallen07 / SpringBoard-Capstone-2--Classifying-Chest-X-rays

Using a CNN to classify medical imaging

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Springboard-Capstone-2-X-Ray Classification

using machine learning to screen x-rays for pathology

Files:

  • The initial data exploration can be found[here]
  • The file sorting script can be found [here]
  • The final model can be found [here]
  • The project write ups can be found [here] and the final [here]

Background

example images

Many thousands of x-rays are performed daily. Within the medical community the assessment of an x-ray especially a chest x-ray (CXR), at least in the rapid screening since is a highly common if not ubiquitous task amounts many medical specialties. Official review is done by a Radiologist (MD or DO). To them this task routine in most cases due to the fact that they have evaluated so many. However, to get to that point they have had to undergo many years of school and residency. In addition to this, despite being a minor task, the sheer number of them requries significant amounts of time which can be taxing on the system.

Data

From the NIH X-ray dataset found here. This dataset contains some 112,120 thoracic scans from 30 or so tousand patients. They have labeled using NLP techniques based on findings in the relevant health records. It should be noted that the accuracy of these labels has been brought into question by many individuals.

Methodology

Use a transitional model neural network (deep learning model) to classify images based upon the provided label. Specifically the ResNet50 base with several custom layers on top will be used. Keras was used with a TensorFlow backend.

Conclusion

performance

The resulting model performed with similar performance to those described here. Although, it should be noted that 9 categories were used there and 15 were used in this exercise.

prediction results

About

Using a CNN to classify medical imaging


Languages

Language:Jupyter Notebook 100.0%