A Novel Convolutional Neural Network Model Based on Beetle Antennae Search Optimization Algorithm fo

A Novel Convolutional Neural Network Model Based on Beetle Antennae Search Optimization Algorithm fo

Abstract:

Convolutional neural networks (CNNs) are widely used in the field of medical imaging diagnosis but have the disadvantages of slow training speed and low diagnostic accuracy due to the initialization of parameters before training. In this article, a CNN optimization method based on the beetle antennae search (BAS) optimization algorithm is proposed. The method optimizes the initial parameters of the CNN through the BAS optimization algorithm. Based on this optimization approach, a novel CNN model with a pretrained BAS optimization algorithm was developed and applied to the analysis and diagnosis of medical imaging data for intracranial hemorrhage. Experimental results on 330 test images show that the proposed method has a better diagnostic performance than the traditional CNN. The proposed method achieves a diagnostic accuracy of 93.9394% and 100% recall, and the diagnosis of 66 human head computerized tomography image data only takes 0.1596 s. Moreover, the proposed method has more advantages than the three other optimization algorithms.