Obálka

Automatic Neural Architecture Search Based on an Estimation of Distribution Algorithm for Binary Classification of Image Databases

Uloženo v:
Vydáno v:Mathematics vol. 13, no. 4 (2025), p. 605
Hlavní autor: Franco-Gaona, Erick
Další autoři: Avila-Garcia, Maria Susana, Cruz-Aceves, Ivan
Vydáno:
MDPI AG
Témata:
Marginal Distribution
Machine learning
Artificial intelligence
Image databases
Databases
Optimization techniques
Artificial neural networks
Genetic algorithms
Neural networks
Searching
Classification
Image classification
Algorithms
Methods
Automation
Computing time
Heuristic methods
On-line přístup:Citation/Abstract
Full Text + Graphics
Full Text - PDF
Tagy: Přidat tag
Žádné tagy, Buďte první, kdo vytvoří štítek k tomuto záznamu!
Abstrakt:Convolutional neural networks (CNNs) are widely used for image classification; however, setting the appropriate hyperparameters before training is subjective and time consuming, and the search space is not properly explored. This paper presents a novel method for the automatic neural architecture search based on an estimation of distribution algorithm (EDA) for binary classification problems. The hyperparameters were coded in binary form due to the nature of the metaheuristics used in the automatic search stage of CNN architectures which was performed using the Boltzmann Univariate Marginal Distribution algorithm (BUMDA) chosen by statistical comparison between four metaheuristics to explore the search space, whose computational complexity is O(<inline-formula>229</inline-formula>). Moreover, the proposed method is compared with multiple state-of-the-art methods on five databases, testing its efficiency in terms of accuracy and F1-score. In the experimental results, the proposed method achieved an F1-score of 97.2%, 98.73%, 97.23%, 98.36%, and 98.7% in its best evaluation, better results than the literature. Finally, the computational time of the proposed method for the test set was ≈0.6 s, 1 s, 0.7 s, 0.5 s, and 0.1 s, respectively.
ISSN:2227-7390
DOI:10.3390/math13040605
Zdroj:Engineering Database