An EnKF-based scheme to optimize hyper-parameters and features for SVM classifier

被引:18
作者
Ji, Yingsheng [1 ,2 ,3 ]
Chen, Yushu [3 ]
Fu, Haohuan [1 ,2 ]
Yang, Guangwen [1 ,2 ,3 ]
机构
[1] Tsinghua Univ, Key Lab Earth Syst Modeling, Minist Educ, Beijing, Peoples R China
[2] Tsinghua Univ, Ctr Earth Syst Sci, Beijing, Peoples R China
[3] Tsinghua Univ, Dept Comp Sci & Technol, Beijing, Peoples R China
关键词
EnKF; SVM; Hyper-parameter optimization; Feature weighting; Feature selection; SUPPORT VECTOR MACHINE; PARTICLE SWARM OPTIMIZATION; FEATURE-SELECTION; MODEL SELECTION; SEARCH; SYSTEM;
D O I
10.1016/j.patcog.2016.08.014
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The quality of models built by machine learning algorithms mostly depends on the careful tuning of hyper-parameters and feature weights. This paper introduces a novel scheme to optimize hyper-parameters and features by using the Ensemble Kalman Filter (EnKF), which is an iterative optimization algorithm used for high-dimensional nonlinear systems. We build a framework for applying the EnKF method on parameter optimization problems. We propose ensemble evolution to converge to the global optimum. We also optimize the EnKF calculation for large datasets by using the computationally efficient UR decomposition. To demonstrate the performance of our proposed design, we apply our approach for the tuning problem of Support Vector Machines. Experimental results show that the better global optima can be identified by our approach with acceptable computation cost compared to three state-of-the-art Bayesian optimization methods (SMAC, TPE and SPEARMINT). (C) 2016 Elsevier Ltd. All rights reserved.
引用
收藏
页码:202 / 213
页数:12
相关论文
共 56 条
  • [1] Optimizing resources in model selection for support vector machine
    Adankon, Mathias M.
    Cheriet, Mohamed
    [J]. PATTERN RECOGNITION, 2007, 40 (03) : 953 - 963
  • [2] Parameter estimation in an intermediate complexity earth system model using an ensemble Kalman filter
    Annan, JD
    Hargreaves, JC
    Edwards, NR
    Marsh, R
    [J]. OCEAN MODELLING, 2005, 8 (1-2) : 135 - 154
  • [3] [Anonymous], 2003, Ocean Dynamics, DOI [10.1007/s10236-003-0036-9, DOI 10.1007/S10236-003-0036-9]
  • [4] [Anonymous], 2003, ATMOSPHERIC MODELING
  • [5] Automatic model selection for the optimization of SVM kernels
    Ayat, NE
    Cheriet, M
    Suen, CY
    [J]. PATTERN RECOGNITION, 2005, 38 (10) : 1733 - 1745
  • [6] A PSO and pattern search based memetic algorithm for SVMs parameters optimization
    Bao, Yukun
    Hu, Zhongyi
    Xiong, Tao
    [J]. NEUROCOMPUTING, 2013, 117 : 98 - 106
  • [7] Finding optimal model parameters by deterministic and annealed focused grid search
    Barbero Jimenez, Alvaro
    Lopez Lazaro, Jorge
    Dorronsoro, Jose R.
    [J]. NEUROCOMPUTING, 2009, 72 (13-15) : 2824 - 2832
  • [8] Bergstra J., 2011, Adv. Neural Inf. Process. Syst, V24
  • [9] Bergstra J, 2012, J MACH LEARN RES, V13, P281
  • [10] Application of machine learning algorithms to the study of noise artifacts in gravitational-wave data
    Biswas, Rahul
    Blackburn, Lindy
    Cao, Junwei
    Essick, Reed
    Hodge, Kari Alison
    Katsavounidis, Erotokritos
    Kim, Kyungmin
    Kim, Young-Min
    Le Bigot, Eric-Olivier
    Lee, Chang-Hwan
    Oh, John J.
    Oh, Sang Hoon
    Son, Edwin J.
    Tao, Ye
    Vaulin, Ruslan
    Wang, Xiaoge
    [J]. PHYSICAL REVIEW D, 2013, 88 (06)