UC CS 21 Senior Design Project: "Machine Learning and Deep Learning: What Happens After Training?"
This is a research project investigating how machine learning systems behave once they have been deployed. For now, we are researching how do these neural networks trained for classification behave in response to augmentations.
Project done by Prudhviraj Naidu.
Advised by Dr. Nan Niu.
- Project Description
- User Interface Specification
- Test Plan and Results
- User Manual
- Spring Final PPT Presentation
- Final Expo Poster
- Assessments
- Summary of Hours and Justification
Q1: How to download the dataset?
Ans: Download the entire ImageNet dataset and use the specific wnids provided in the text files in the data directory.
Q2: How long does it take to train?
Ans: It took 8 hours to train for 120 epochs on Kaggle GPU
Q3: Can we use TPU?
Ans: No, the keras training pipeline uses data augmentation operations which are currently not supported by TPU.