Hyperparameter optimization for recommender systems through Bayesian optimization

被引:17
|
作者
Galuzzi, B. G. [1 ]
Giordani, I [1 ]
Candelieri, A. [1 ]
Perego, R. [1 ]
Archetti, F. [1 ,2 ]
机构
[1] Univ Milano Bicocca, Dept Comp Sci Syst & Commun, Viale Sarca 336, I-20125 Milan, Italy
[2] Consorzio Milano Ric, Via Roberto Cozzi 53, I-20126 Milan, Italy
关键词
Bayesian optimization; Collaborative filtering; Hyperparameters optimization; Matrix factorization; Recommender system;
D O I
10.1007/s10287-020-00376-3
中图分类号
O1 [数学]; C [社会科学总论];
学科分类号
03 ; 0303 ; 0701 ; 070101 ;
摘要
Recommender systems represent one of the most successful applications of machine learning in B2C online services, to help the users in their choices in many web services. Recommender system aims to predict the user preferences from a huge amount of data, basically the past behaviour of the user, using an efficient prediction algorithm. One of the most used is the matrix-factorization algorithm. Like many machine learning algorithms, its effectiveness goes through the tuning of its hyper-parameters, and the associated optimization problem also called hyper-parameter optimization. This represents a noisy time-consuming black-box optimization problem. The related objective function maps any possible hyper-parameter configuration to a numeric score quantifying the algorithm performance. In this work, we show how Bayesian optimization can help the tuning of three hyper-parameters: the number of latent factors, the regularization parameter, and the learning rate. Numerical results are obtained on a benchmark problem and show that Bayesian optimization obtains a better result than the default setting of the hyper-parameters and the random search.
引用
收藏
页码:495 / 515
页数:21
相关论文
共 50 条
  • [21] Hyperparameter Optimization: Comparing Genetic Algorithm against Grid Search and Bayesian Optimization
    Alibrahim, Hussain
    Ludwig, Simone A.
    2021 IEEE CONGRESS ON EVOLUTIONARY COMPUTATION (CEC 2021), 2021, : 1551 - 1559
  • [22] Hyperparameter Optimization of Bayesian Neural Network Using Bayesian Optimization and Intelligent Feature Engineering for Load Forecasting
    Zulfiqar, M.
    Gamage, Kelum A. A.
    Kamran, M.
    Rasheed, M. B.
    SENSORS, 2022, 22 (12)
  • [23] Bayesian Hyperparameter Optimization of Deep Neural Network Algorithms Based on Ant Colony Optimization
    Jlassi, Sinda
    Jdey, Imen
    Ltifi, Hela
    DOCUMENT ANALYSIS AND RECOGNITION, ICDAR 2021, PT III, 2021, 12823 : 585 - 594
  • [24] Automatic Hyperparameter Optimization for Transfer Learning on Medical Image Datasets Using Bayesian Optimization
    Borgli, Rune Johan
    Stensland, Hakon Kvale
    Riegler, Michael Alexander
    Halvorsen, Pal
    2019 13TH INTERNATIONAL SYMPOSIUM ON MEDICAL INFORMATION AND COMMUNICATION TECHNOLOGY (ISMICT), 2019, : 175 - 180
  • [25] Efficient Hyperparameter Optimization by Using Bayesian Optimization for Drug-Target Interaction Prediction
    Ban, Tomohiro
    Ohue, Masahito
    Akiyama, Yutaka
    2017 IEEE 7TH INTERNATIONAL CONFERENCE ON COMPUTATIONAL ADVANCES IN BIO AND MEDICAL SCIENCES (ICCABS), 2017,
  • [26] Informing the Use of Hyperparameter Optimization Through Metalearning
    Sanders, Samantha
    Giraud-Carrier, Christophe
    2017 17TH IEEE INTERNATIONAL CONFERENCE ON DATA MINING (ICDM), 2017, : 1051 - 1056
  • [28] Practical Multi-fidelity Bayesian Optimization for Hyperparameter Tuning
    Wu, Jian
    Toscano-Palmerin, Saul
    Frazier, Peter, I
    Wilson, Andrew Gordon
    35TH UNCERTAINTY IN ARTIFICIAL INTELLIGENCE CONFERENCE (UAI 2019), 2020, 115 : 788 - 798
  • [29] Efficient hyperparameter tuning for predicting student performance with Bayesian optimization
    Saleh Albahli
    Multimedia Tools and Applications, 2024, 83 : 52711 - 52735
  • [30] Disease prediction via Bayesian hyperparameter optimization and ensemble learning
    Gao Liyuan
    Ding Yongmei
    BMC RESEARCH NOTES, 2020, 13 (01)