Towards the Distributed Wound Treatment Optimization Method for Training CNN Models: Analysis on the MNIST Dataset

Convolutional neural network (CNN) is a prominent algorithm in Deep Learning methods. CNN architectures have been used successfully to solve various problems in image processing, for example, segmentation, classification, and enhancement task. However, automatic search for suitable architectures and training parameters remain an open area of research, where metaheuristic algorithms have been used to fine-tuning the hyperparameters and learning parameters. This work presents a bio-inspired distributed strategy based on Wound Treatment optimization (WTO) for training the learning parameters of a LenNet CNN model fast and accurate. The proposed method was evaluated over the popular benchmark dataset MNIST for handwritten digit recognition. Experimental results showed an improvement of 36.87% in training time using the distributed WTO method compared to the baseline with a single learning agent, and the accuracy increases 4.69% more using the proposed method in contrast with the baseline. As this is a preliminary study towards the distributed WTO method for training CNN models, we anticipate this approach can be used in robotics, multi-agent systems, federated learning, complex optimization problems, and many others, where an optimization task is required to be solved fast and accurate.