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 条
[1]  
[Anonymous], 2008, P ICML, DOI DOI 10.1145/1390156.1390177
[2]  
[Anonymous], 1993, HDB METAHEURISTICS
[3]  
Bergstra J, 2012, J MACH LEARN RES, V13, P281
[4]  
Bergstra James, 2011, Advances in Neural Information Processing Systems, V24, P2546, DOI DOI 10.5555/2986459.2986743
[5]   Metaheuristics in combinatorial optimization: Overview and conceptual comparison [J].
Blum, C ;
Roli, A .
ACM COMPUTING SURVEYS, 2003, 35 (03) :268-308
[6]   Evolutionary Deep Learning for Car Park Occupancy Prediction in Smart Cities [J].
Camero, Andres ;
Toutouh, Jamal ;
Stolfi, Daniel H. ;
Alba, Enrique .
LEARNING AND INTELLIGENT OPTIMIZATION, LION 12, 2019, 11353 :386-401
[7]  
Ceapa I., 2012, Proc. of the ACM SIGKDD Intl. Workshop on Urban Computing, Beijing, P134
[8]  
Elsken T, 2019, J MACH LEARN RES, V20
[9]  
Feurer M, 2019, SPRING SER CHALLENGE, P3, DOI 10.1007/978-3-030-05318-5_1
[10]  
Feurer M, 2015, AAAI CONF ARTIF INTE, P1128