Hyperparameter Optimization Using a Genetic Algorithm Considering Verification Time in a Convolutional Neural Network

Hyperparameter optimization is a very difficult problem in developing deep learning algorithms. In this paper, a genetic algorithm was applied to solve this problem. The accuracy and the verification time were considered by conducting a fitness evaluation. The algorithm was evaluated by using a simp...

Ausführliche Beschreibung

Gespeichert in:
Bibliographische Detailangaben
Veröffentlicht in:Journal of electrical engineering & technology 2020-03, Vol.15 (2), p.721-726
Hauptverfasser: Han, Ji-Hoon, Choi, Dong-Jin, Park, Sang-Uk, Hong, Sun-Ki
Format: Artikel
Sprache:eng
Schlagworte:
Online-Zugang:Volltext
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
Beschreibung
Zusammenfassung:Hyperparameter optimization is a very difficult problem in developing deep learning algorithms. In this paper, a genetic algorithm was applied to solve this problem. The accuracy and the verification time were considered by conducting a fitness evaluation. The algorithm was evaluated by using a simple model that has a single convolution layer and a single fully connected layer. A model with three layers was used. The MNIST dataset and a motor fault diagnosis dataset were used to train the algorithm. The results show that the method is useful for reducing the training time.
ISSN:1975-0102
2093-7423
DOI:10.1007/s42835-020-00343-7