Massive Dimensions Reduction and Hybridization with Meta-heuristics in Deep Learning

TL;DR

This paper introduces Histogram-based Blocking Differential Evolution (HBDE), a hybrid optimization method that significantly reduces parameters in deep neural networks like ResNet-18, improving training efficiency and performance on CIFAR datasets.

Contribution

The study proposes a novel hybrid meta-heuristic algorithm combining gradient-based and gradient-free methods with blocking to optimize large-scale DNNs, reducing parameters drastically.

Findings

HBDE reduces ResNet-18 parameters from 11 million to 3,000.

HBDE outperforms baseline algorithms on CIFAR-10 and CIFAR-100.

The approach demonstrates effective large-scale optimization with lower computational costs.

Abstract

Deep learning is mainly based on utilizing gradient-based optimization for training Deep Neural Network (DNN) models. Although robust and widely used, gradient-based optimization algorithms are prone to getting stuck in local minima. In this modern deep learning era, the state-of-the-art DNN models have millions and billions of parameters, including weights and biases, making them huge-scale optimization problems in terms of search space. Tuning a huge number of parameters is a challenging task that causes vanishing/exploding gradients and overfitting; likewise, utilized loss functions do not exactly represent our targeted performance metrics. A practical solution to exploring large and complex solution space is meta-heuristic algorithms. Since DNNs exceed thousands and millions of parameters, even robust meta-heuristic algorithms, such as Differential Evolution, struggle to efficiently explore and converge in such huge-dimensional search spaces, leading to very slow convergence and high memory demand. To tackle the mentioned curse of dimensionality, the concept of blocking was recently proposed as a technique that reduces the search space dimensions by grouping them into blocks. In this study, we aim to introduce Histogram-based Blocking Differential Evolution (HBDE), a novel approach that hybridizes gradient-based and gradient-free algorithms to optimize parameters. Experimental results demonstrated that the HBDE could reduce the parameters in the ResNet-18 model from 11M to 3K during the training/optimizing phase by metaheuristics, namely, the proposed HBDE, which outperforms baseline gradient-based and parent gradient-free DE algorithms evaluated on CIFAR-10 and CIFAR-100 datasets showcasing its effectiveness with reduced computational demands for the very first time.