Update README.md

README.md

@@ -2,7 +2,7 @@
 license: apache-2.0
 ---
 
-# EDL(VGG) - Pakistan Flood Detection
+# EDL(VGG) - Pakistan Flood Detection Purpose
 
 This model was created to help detect flooding and classify areas as flooded or not flooded given aerial photos of a given location. First responders need to act quickly in a crisis, but don’t have the time to collect large datasets of damage at a location. They also would likely want to use a non-technical interface to quickly create and use these models. Our solution Easy Deep Learning allows first responders and many others with use cases like them to rapidly create and use their deep learning models. This model was made using Intel Developer Cloud.
 
@@ -13,3 +13,8 @@ Sourcing a large enough dataset for training can often be very time consuming an
 ## Training Process
 
 This classifier was trained on our platform with only 6 real examples of non-flooded areas and 6 examples of flooded areas. From there we automatically generated 200 additional data points for our model in about 5 minutes using ipex-optimized code on an xpu compute device. Without additional training data, the vision transformer suffered catastrophically from overfitting, even after tuning its hyperparameters. It simply chose to mark all regions as either being flooded or not flooded across 5 runs. Augmenting the data synthetically allowed us to go from random guessing to 75%-100% accuracy across 8 test points across 5 runs with random test points selected for each run. 
+
+## Optimization
+
+1. Generative model was optimized for Pytorch XPU compute on intel's developer cloud using intel extention for Pytorch.
+2. VGG model uses MKL compiled pytorch instance. We found that for smaller models like VGG benefit from optimized CPU compute on smaller sample sizes more than GPU/XPU compute.