Automatic Hyperparameter Tuning in Deep Convolutional Neural Networks Using Asynchronous Reinforcement Learning

被引:39
作者
Neary, Patrick L. [1 ]
机构
[1] Utah State Univ, Dept Comp Sci, Logan, UT 84322 USA
来源
2018 IEEE INTERNATIONAL CONFERENCE ON COGNITIVE COMPUTING (ICCC) | 2018年
关键词
image recognition; neural networks; machine learning; convolutional neural networks; artificial intelligence; hyperparameter tuning; deep learning;
D O I
10.1109/ICCC.2018.00017
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Major gains have been made in recent years in object recognition due to advances in deep neural networks. One struggle with deep learning, however, revolves around the fact that currently it is unknown what network architecture is best for a given problem. Consequently, different configurations are tried until one is identified that gives acceptable results. This paper proposes an asynchronous reinforcement learning algorithm that finds an optimal network configuration by automatically adjusting parameters for a given problem. It is shown that asynchronous reinforcement learning is able to converge on an optimal solution for the MNIST data set.
引用
收藏
页码:73 / 77
页数:5
相关论文
共 13 条
[1]  
Allen Martin., 2008, P 23 NATL C ARTIFICI, P1440
[2]  
[Anonymous], 2016, ARXIV160201783CS
[3]  
[Anonymous], 2016, DESIGNING NEURAL NET
[4]  
[Anonymous], 2017, J ADV TRANSPORT, DOI DOI 10.1155/2017/8608032
[5]  
[Anonymous], PLOS COMPUTATIONAL B
[6]  
Chen X.-Y., 2016, Journal of Information Hiding and Multimedia Signal Processing, V7, P1345
[7]   Deep learning [J].
LeCun, Yann ;
Bengio, Yoshua ;
Hinton, Geoffrey .
NATURE, 2015, 521 (7553) :436-444
[8]  
Pu L, 2016, 2016 CIE INTERNATIONAL CONFERENCE ON RADAR (RADAR)
[9]  
Schaffer J. D., 1992, COGANN-92. International Workshop on Combinations of Genetic Algorithms and Neural Networks (Cat. No.92TH0435-8), P1, DOI 10.1109/COGANN.1992.273950
[10]   VERY DEEP LEARNING FOR SHIP DISCRIMINATION IN SYNTHETIC APERTURE RADAR IMAGERY [J].
Schwegmann, C. P. ;
Kleynhans, W. ;
Salmon, B. P. ;
Mdakane, L. W. ;
Meyer, R. G. V. .
2016 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS), 2016, :104-107