Package: deepaugment
DeepAugmentDeepAugment discovers augmentation strategies tailored for your images. It uses Bayesian Optimization for optimizing data augmentation hyperparameters. The tool:reduces error rate of CNN models (shown 60% decrease in error for CIFAR-10 on WRN-28-10 compared to no augmentation)saves time by automating the processResources:slidesInstallation/UsageTutorial:google-colab$pipinstalldeepaugmentSimple usage (with any dataset)fromdeepaugment.deepaugmentimportDeepAugmentdeepaug=DeepAugment(my_images,my_labels)best_policies=deepaug.optimize(300)Simple usage (with CIFAR-10 on keras)deepaug=DeepAugment("cifar10")best_policies=deepaug.optimize(300)Advanced usagefromkeras.datasetsimportfashion_mnist# my configurationmy_config={"model":"basiccnn","method":"bayesian_optimization","train_set_size":2000,"opt_samples":3,"opt_last_n_epochs":3,"opt_initial_points":10,"child_epochs":50,"child_first_train_epochs":0,"child_batch_size":64}(x_train,y_train),(x_test,y_test)=fashion_mnist.load_data()# X_train.shape -> (N, M, M, 3)# y_train.shape -> (N)deepaug=DeepAugment(iamges=x_train,labels=y_train,config=my_config)best_policies=deepaug.optimize(300)ResultsCIFAR-10 best policies tested on WRN-28-10Method: Wide-ResNet-28-10 trained with CIFAR-10 augmented images by best found policies, and with unaugmented images (everything else same).Result:60% reduction in error(8.5% accuracy increase) by DeepAugmentDesign goalsDeepAugment is designed as a scalable and modular partner to AutoAugment (Cubuk et al., 2018). AutoAugment was one of the most exciting publications in 2018. It was the first method using Reinforcement Learning for this problem. AutoAugmentation, however, has no complete open-sourced implementation (controller module not available) preventing users to run it for their own datasets, and takes 15,000 iterations to learn (according to paper) augmentation policies, which requires massive computational resources. Thus most people could not benefit from it even if its source code would be fully available.DeepAugment addresses these two problems. Its main design goals are:minimize the computational complexity of optimization while maintaining quality of resultsbe modular and user-friendlyFirst goal is achieved by following changes compared to AutoAugment:Bayesian Optimization instead of Reinforcement Learningwhich requires much less number of iterations (~100 times)Minimized Child Modeldecreasing computational complexity of each training (~20 times)Less stochastic augmentation search space designdecreasing number of iterations neededFor achieving the second goal, user interface is designed in a way that it gives user broad configuration possibilities and model selections (e.g. selecting the child model or inputting a self-designed child model).ImportancePractical importanceDeepAugment makes optimization of data augmentation scalable, and thus enables users to optimize augmentation policies without needing massive computational resources.
As an estimate of its computational cost, it takes4.2 hours(500 iterations) on CIFAR-10 dataset which costs around$13using AWS p3.x2large instance.Academic importanceTo our knowlege, DeepAugment is the first method which utilizes Bayesian Optimization for the problem of data augmentation hyperparameter optimization.How it worksThree major components of DeepAugment are controller, augmenter, and child model. Overal workflow is that controller samples new augmentation policies, augmenter transforms images by the new policy, and child model is trained from scratch by augmented images. Then, a reward is calculated from child model's training history. This reward is returned back to the controller, and it updates its surrogate model with this reward and associated augmentation policy. Then, controller samples new policies again and same steps repeats. This process cycles until user-determined maximum number of iterations reached.Controller can be set for using either Bayesian Optimization (default) or Random Search. If set to Bayesian Optimization, samples new policies by a Random Forest Estimator and Expected Improvement acquistion function.Why Bayesian Optimization?In hyperparameter optimization, main choices are random search, grid search, bayesian optimization (BO), and reinforcement learning (RL) (in the order of method complexity). Google'sAutoAugmentuses RL for data augmentation hyperparameter tuning, but it takes 15,000 iterations to learn policies (which means training the child CNN model 15,000 times). Thus, it requires massive computational resources. Bayesian Optimization on the other hand learns good polices in 100-300 iterations, making it +40X faster. Additionally, it is better than grid search and random search in terms of accuracy, cost, and computation time in hyperparameter tuning(ref) (we can think optimization of augmentation policies as a hyperparameter tuning problem where hyperparameters are concerning with augmentations instead of the deep learning architecture). This result is not surprising since despite Grid Search or Random Search BO selects new hyperparameter as informed with previous results for tried hyperparameters.How does Bayesian Optimization work?Aim of Bayesian Optimization (BO) is findingset of parameterswhich maximize the value of anobjective function. It builds a surrogate model for predicting value of objective function for unexplored parameters. Working cycle of BO can be summarized as:Build a surrogate model of the objective functionFind parameters that perform best on the surrogate (or pick random hyperparameters)Execute objective function with these parametersUpdate the surrogate model with these parameters and result (value) of objective functionRepeat steps 2-4 until maximimum number of iterations reachedFor more detailed explanation, readthis blogpostexplaining BO in high-level, or take a glance atthis review paperAugmentation policyA policy describes the augmentation will be applied on a dataset. Each policy consists variables for two augmentation types, their magnitude and the portion of the data to be augmented. An example policy is as following:There are currently 20 types of augmentation techniques (above, right) that each aug. type variable can take. All techniques are (this list might grow in later versions):AUG_TYPES=["crop","gaussian-blur","rotate","shear","translate-x","translate-y","sharpen","emboss","additive-gaussian-noise","dropout","coarse-dropout","gamma-contrast","brighten","invert","fog","clouds","add-to-hue-and-saturation","coarse-salt-pepper","horizontal-flip","vertical-flip"]Child modelsourceChild model is trained over and over from scratch during the optimization process. Its number of training depends on the number of iterations chosen by the user, which is expected to be around 100-300 for obtaining good results. Child model is therefore the computational bottleneck of the algorithm. With the current design, training time is ~30 seconds for 32x32 images on AWS instance p3.x2large using V100 GPU (112 TensorFLOPS). It has 1,250,858 trainable parameters for 32x32 images. Below is the diagram of child model:Other choices for child CNN modelStandard Child model is a basic CNN where its diagram and details given above. However, you are not limited with that model. You can use your own keras model by assigning it into config dictionary as:my_config={"model":my_keras_model_object}deepaug=DeepAugment(my_images,my_labels,my_config)Or use an implemented small model, such as WideResNet-40-2 (while it is bigger than Basic CNN):my_config={"model":"wrn_40_2"}# depth(40) and wideness-factor(2) can be changed. e.g. wrn_20_4Or use a big model (not recommended unless you have massive computational resources):my_config={"model":"InceptionV3"}my_config={"model":"MobileNetV2"}Reward functionsourceReward function is calculated as mean of K highest validation accuracies of the child model which is not smaller than corresponding training accuracy by 0.05. K can be determined by the user by updatingopt_last_n_epochskey in config as argument toDeepAugment()class (K is 3 by default).Configuration optionsDeepAugment can be given a config dictionary during initialization. It is expected to have following keys:model: child model type. Options: "basiccnn", "inceptionv3", "mobilenetv2", "wrn__", or keras.models.Model objectmethod:"bayesian_optimization" or "random" (for random search)train_set_size:size of the training set during optimization. It should be small enough that computation will not take too long.opt_samples:number of samples optimizer will run for each augmentation-policy. Training of the child model is stochastic and validation accuracy results might be slightly different from run to run. The tool trains child model three times by default and takes average, in order to have more robust accuracy results.opt_last_n_epochs:number of non-overfitting epochs whose validation accuracy average will be used as reward. For each training,opt_last_n_epochshighest validation accuracies (where its difference to training accuracy is not more than 10%) are averaged and taken as reward.opt_initial_points:number of random initial policies will be tried by Bayesian Optimizer. It will be then_initial_pointsargument for skopt Optimizer (see itsdocumentation)child_epochs:number of epochs for the child modelchild_first_train_epochs:if not 0, child model is pre-trained without any augmentation and its resulting weights are load for each training with augmentation. The purpose is training child model 10-20 epochs once and thereby saving 10-20 epochs for each training of optimizer iterations which is +100 times.child_batch_size:batch size for the child modelper_aug_weights_path:path for pre-augmented training weights. Unneccessary ifchild_first_train_epochs=0logging:logging object for getting news about the optimization.notebook_path:path for recording all trainings in all iterations. For each iteration, training history, trial-no, sample-no, calculated reward and mean recent validation accuracy is recorded. Records is updated at each trial for ensuring records are not lost in case optimization interrupted unintentionally. Records can be found at "/reports/experiments//notebook.csv"Default configurations are as following:DEFAULT_CONFIG={"model":"basiccnn",#"method":"bayesian_optimization","train_set_size":2000,"opt_samples":3,"opt_last_n_epochs":3,"opt_initial_points":10,"child_epochs":50,"child_first_train_epochs":0,"child_batch_size":64,"pre_aug_weights_path":"pre_aug_weights.h5","logging":logging,"notebook_path":f"{EXPERIMENT_FOLDER_PATH}/notebook.csv",}Versioning rulesA three-number system is used, like1.2.3. Each increment of version is one of the following types:minor: if bugs are fixed, or documentation changed significantly.1.2.3 -> 1.2.4major: if a feature implemented differently, or a new feature added.1.2.3 -> 1.3.0disruptive: if a feature is removed or renamed.1.2.3 -> 2.0.0(Backward compatibility is not guaranteed)Note: Versions from0.0.0to1.0.0are considered asalpha phaseand do not follow this strategy.Data pipelineClass diagramCreated bypyreversePackage diagramCreated bypyreverseReferences[1] Cubuk et al., 2018. AutoAugment: Learning Augmentation Policies from Data
(arxiv)[2] Zoph et al., 2016. Neural Architecture Search with Reinforcement Learning
(arxiv)[3] Shahriari et al., 2016. A review of Bayesian Optimization
(ieee)[4] Dewancker et al. Bayesian Optimization Primer (white-paper)[5] DeVries, Taylor 2017. Improved Regularization of CNN's with Cutout
(arxiv)Blogs:A conceptual explanation of Bayesian Optimization (towardsdatascience)Comparison experiment: Bayesian Opt. vs Grid Search vs Random Search (mlconf)Libraries:scikit-optimizemgaugAutoAugment-unofficialAutomold(Self-driving car image-augmentation library)ContactBaris Ozmen,hbaristr@gmail.com