Generating Efficient DNN-Ensembles with Evolutionary Computation

In this work, we leverage ensemble learning as a tool for the creation of faster, smaller, and more accurate deep learning models. We demonstrate that we can jointly optimize for accuracy, inference time, and the number of parameters by combining DNN classifiers. To achieve this, we combine multiple ensemble strategies: bagging, boosting, and an ordered chain of classifiers. To reduce the number of DNN ensemble evaluations during the search, we propose EARN, an evolutionary approach that optimizes the ensemble according to three objectives regarding the constraints specified by the user. We run EARN on 10 image classification datasets with an initial pool of 32 state-of-the-art DCNN on both CPU and GPU platforms, and we generate models with speedups up to $7.60\times$, reductions of parameters by $10\times$, or increases in accuracy up to $6.01\%$ regarding the best DNN in the pool. In addition, our method generates models that are $5.6\times$ faster than the state-of-the-art methods for automatic model generation.

[1]  Hiroshi Inoue Adaptive Ensemble Prediction for Deep Neural Networks based on Confidence Level , 2019, AISTATS.

[2]  Kalyanmoy Deb,et al.  A fast and elitist multiobjective genetic algorithm: NSGA-II , 2002, IEEE Trans. Evol. Comput..

[3]  Marco Laumanns,et al.  SPEA2: Improving the strength pareto evolutionary algorithm , 2001 .

[4]  Wei Tang,et al.  Corrigendum to "Ensembling neural networks: Many could be better than all" [Artificial Intelligence 137 (1-2) (2002) 239-263] , 2010, Artif. Intell..

[5]  Maya R. Gupta,et al.  Quit When You Can: Efficient Evaluation of Ensembles with Ordering Optimization , 2018, ArXiv.

[6]  Yoav Freund,et al.  Experiments with a New Boosting Algorithm , 1996, ICML.

[7]  Jing Lu,et al.  Creating ensembles of classifiers via fuzzy clustering and deflection , 2010, Fuzzy Sets Syst..

[8]  Xin Yao,et al.  DIVACE: Diverse and Accurate Ensemble Learning Algorithm , 2004, IDEAL.

[9]  Hisao Ishibuchi,et al.  Designing Fuzzy Ensemble Classifiers by Evolutionary Multiobjective Optimization with an Entropy-Based Diversity Criterion , 2006, 2006 Sixth International Conference on Hybrid Intelligent Systems (HIS'06).

[10]  J JonesMichael,et al.  Robust Real-Time Face Detection , 2004 .

[11]  Oleksandr Makeyev,et al.  Neural network with ensembles , 2010, The 2010 International Joint Conference on Neural Networks (IJCNN).

[12]  Leo Breiman,et al.  Random Forests , 2001, Machine Learning.

[13]  Evgenii Sopov,et al.  Self-Configuring Ensemble of Neural Network Classifiers for Emotion Recognition in the Intelligent Human-Machine Interaction , 2015, 2015 IEEE Symposium Series on Computational Intelligence.

[14]  Yang Yu,et al.  Pareto Ensemble Pruning , 2015, AAAI.

[15]  V. B. Surya Prasath,et al.  Choosing Mutation and Crossover Ratios for Genetic Algorithms - A Review with a New Dynamic Approach , 2019, Inf..

[16]  Yee Whye Teh,et al.  Neural Ensemble Search for Performant and Calibrated Predictions , 2020, ArXiv.

[17]  N. Chawla,et al.  Evolutionary Ensembles : Combining Learning Agents using Genetic Algorithms , 2005 .

[18]  Trevor Hastie,et al.  Multi-class AdaBoost ∗ , 2009 .

[19]  Bernhard Sendhoff,et al.  Generalization Improvement in Multi-Objective Learning , 2006, The 2006 IEEE International Joint Conference on Neural Network Proceedings.

[20]  Luca Benini,et al.  Constrained deep neural network architecture search for IoT devices accounting hardware calibration , 2019, NeurIPS.

[21]  Dongxiao Zhang,et al.  Efficient Ensemble-Based Closed-Loop Production Optimization , 2009 .

[22]  Huanhuan Chen,et al.  When does Diversity Help Generalization in Classification Ensembles? , 2019, ArXiv.

[23]  Wei Tang,et al.  Ensembling neural networks: Many could be better than all , 2002, Artif. Intell..

[24]  Yoav Freund,et al.  An Adaptive Version of the Boost by Majority Algorithm , 1999, COLT '99.

[25]  Yang Yu,et al.  Subset Selection by Pareto Optimization , 2015, NIPS.

[26]  William Nick Street,et al.  Ensemble Pruning Via Semi-definite Programming , 2006, J. Mach. Learn. Res..

[27]  Yang Yu,et al.  Diversity Regularized Ensemble Pruning , 2012, ECML/PKDD.