KMS Chongqing Institute of Green and Intelligent Technology, CAS
| Hyperparameter Learning for Deep Learning-Based Recommender Systems | |
Wu, Di1,2 ; Sun, Bo3,4; Shang, Mingsheng3,4
| |
| 2023-07-01 | |
| 摘要 | Deep learning (DL)-based recommender system (RS), particularly for its advances in the recent five years, has been startling. It reshapes the architectures of traditional RSs by lifting their limitations in dealing with data sparsity and cold-start issues. Yet, the performance of DL-based RS, like many other DL-based intelligent systems, heavily relies on selecting hyperparameters. Unfortunately, the most common selection approach is still Grid Search that requires numerous computational resources and human efforts. Motivated by this, this paper proposes a general hyperparameter optimization framework, named DE-Opt, which can be applied to most existing DL-based RSs seamlessly. The main idea of DE-Opt is to incorporate differential evolution (DE) into the model training process of a DL-based RS at a layer-wise granularity to simultaneously auto-learn its two key hyperparameters, namely, the learning rate ? and the regularization coefficient ?. As a result, the system performance in terms of recommendation accuracy and computational efficiency is uplifted. Experimental results on three benchmark datasets substantiate that: 1) DE-Opt is compatible with the most recent advancements of DL-based RS to automate their hyperparameter-tuning processes, and 2) DE-Opt excels among its state-of-the-art hyperparameter-optimization competitors in terms of both higher learning performance and lower runtime. |
| 关键词 | Terms-Deep learning differential evolution grid search hyperparameter tuning online services recommender systems |
| DOI | 10.1109/TSC.2023.3234623 |
| 发表期刊 | IEEE TRANSACTIONS ON SERVICES COMPUTING
 |
| ISSN | 1939-1374 |
| 卷号 | 16期号:4页码:2699-2712 |
| 通讯作者 | Shang, Mingsheng(msshang@cigit.ac.cn) |
| 收录类别 | SCI |
| WOS记录号 | WOS:001045785600029 |
| 语种 | 英语 |
