top of page

Diagnosis with Convolutional Nets

This project uses Python and TensorFlow to analyze 120,000 chest x-rays, each labeled with zero or more of 14 health issues, and identify which diagnoses apply to individual x-rays.

​

Hyperparameter Tuning - Grid search is used to find the best learning rate.

Transfer Learning - Inception Net is used as a base, and the last half of the layers are tuned on the training set. Results - Validation AUC sits at .79 right now. It's the best result on Kaggle at the moment (as of 2/22), but you still wouldn't want it to replace your doctor.

​

Click through to the GitHub repo here.

​

As a next step, I'd love to rebalance the dataset and try a mixture-of-experts model to increase precision and recall.

bottom of page