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 条
  • [31] Bayesian Hyperparameter Optimization for Machine Learning Based eQTL Analysis
    Quitadamo, Andrew
    Johnson, James
    Shi, Xinghua
    ACM-BCB' 2017: PROCEEDINGS OF THE 8TH ACM INTERNATIONAL CONFERENCE ON BIOINFORMATICS, COMPUTATIONAL BIOLOGY,AND HEALTH INFORMATICS, 2017, : 98 - 106
  • [32] Initializing Bayesian Hyperparameter Optimization via Meta-Learning
    Feurer, Matthias
    Springenberg, Jost Tobias
    Hutter, Frank
    PROCEEDINGS OF THE TWENTY-NINTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE, 2015, : 1128 - 1135
  • [33] Efficient hyperparameter tuning for kernel ridge regression with Bayesian optimization
    Stuke, Annika
    Rinke, Patrick
    Todorovic, Milica
    MACHINE LEARNING-SCIENCE AND TECHNOLOGY, 2021, 2 (03):
  • [34] Disease prediction via Bayesian hyperparameter optimization and ensemble learning
    Liyuan Gao
    Yongmei Ding
    BMC Research Notes, 13
  • [35] Fast hyperparameter tuning using Bayesian optimization with directional derivatives
    Joy, Tinu Theckel
    Rana, Santu
    Gupta, Sunil
    Venkatesh, Svetha
    KNOWLEDGE-BASED SYSTEMS, 2020, 205
  • [36] Optimization of Recommender Systems Based on Inventory
    Demirezen, Emre M.
    Kumar, Subodha
    PRODUCTION AND OPERATIONS MANAGEMENT, 2016, 25 (04) : 593 - 608
  • [37] Structure-based hyperparameter selection with Bayesian optimization in multidimensional scaling
    Thomas Rusch
    Patrick Mair
    Kurt Hornik
    Statistics and Computing, 2023, 33
  • [38] Structure-based hyperparameter selection with Bayesian optimization in multidimensional scaling
    Rusch, Thomas
    Mair, Patrick
    Hornik, Kurt
    STATISTICS AND COMPUTING, 2023, 33 (01)
  • [39] Fair and green hyperparameter optimization via multi-objective and multiple information source Bayesian optimization
    Antonio Candelieri
    Andrea Ponti
    Francesco Archetti
    Machine Learning, 2024, 113 : 2701 - 2731
  • [40] Reproducible Hyperparameter Optimization
    Hertel, Lars
    Baldi, Pierre
    Gillen, Daniel L.
    JOURNAL OF COMPUTATIONAL AND GRAPHICAL STATISTICS, 2022, 31 (01) : 84 - 99