An Effective Hyperparameter Optimization Algorithm for DNN to Predict Passengers at a Metro Station

被引:11
作者
Tsai, Chun-Wei [1 ]
Fang, Zhi-Yan [1 ]
机构
[1] Natl Sun Yat Sen Univ, Dept Comp Sci & Engn, 70 Lienhai Rd, Kaohsiung, Taiwan
关键词
Deep learning; and hyperparameter optimization; SEARCH;
D O I
10.1145/3410156
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
As one of the public transportation systems, metro is certainly an indispensable part in urban areas of a metropolis today. Several successful results have shown that deep learning technologies might provide an effective way to predict the number of passengers at a metro station. However, most information systems based on deep learning technologies are usually designed and tuned manually by using domain knowledge and trial-and-error; thus, how to find out a set of suitable hyperparameters for a deep neural network (DNN) has become a critical research issue. To deal with the problem of hyperparameter setting for a DNN in solving the prediction of passengers at a metro station, a novel metaheuristic algorithm called search economics for hyperparameter optimization is presented to improve the accuracy rate of such a prediction system. The basic idea of the proposed algorithm is to divide the solution space into a set of subspaces first and then assign a different number of search agents to each subspace based on the "potential of each subspace." The potential is estimated based on the objective values of the searched solutions, the objective values of the probe solutions, and the computation time invested in each subspace. The proposed method is compared with Bayesian, random forest, support vector regression, DNN, and DNN with different hyperparameter search algorithms, namely, grid search, simulated annealing, and particle swarm optimization. The simulation results using the data provided by the government of Taipei city, Taiwan, indicate that the proposed method outperforms all the other forecasting methods compared in this article in terms of the mean absolute percentage error.
引用
收藏
页数:24
相关论文
共 48 条
[41]   Optimizing hyperparameters of deep learning in predicting bus passengers based on simulated annealing [J].
Tsai, Chun-Wei ;
Hsia, Chien-Hui ;
Yang, Shuang-Jie ;
Liu, Shih-Jui ;
Fang, Zhi-Yan .
APPLIED SOFT COMPUTING, 2020, 88
[42]   An Effective IoT Service-to-Interface Assignment Algorithm via Search Economics [J].
Tsai, Chun-Wei ;
Liu, Shi-Jui .
IEEE INTERNET OF THINGS JOURNAL, 2018, 5 (03) :1708-1718
[43]   An effective WSN deployment algorithm via search economics [J].
Tsai, Chun-Wei .
COMPUTER NETWORKS, 2016, 101 :178-191
[44]   Metaheuristic Scheduling for Cloud: A Survey [J].
Tsai, Chun-Wei ;
Rodrigues, Joel J. P. C. .
IEEE SYSTEMS JOURNAL, 2014, 8 (01) :279-291
[45]  
Voelcker J., 2014, 1 2 BILLION VEHICLES
[46]  
Wang J., 2018, Combination of hyperband and bayesian optimization for hyperparameter optimization in deep learning
[47]  
Yao Q., 2018, ARXIV181013306
[48]   Particle swarm optimization-based automatic parameter selection for deep neural networks and its applications in large-scale and high-dimensional data [J].
Ye, Fei .
PLOS ONE, 2017, 12 (12)