KMS Chongqing Institute of Green and Intelligent Technology, CAS
An adaptive latent factor model via particle swarm optimization | |
Wang, Qingxian1; Chen, Sili2; Luo, Xin3,4 | |
2019-12-05 | |
摘要 | Latent factor (LF) models are greatly efficient in extracting valuable knowledge from High-Dimensional and Sparse (HiDS) matrices which are usually seen in many industrial applications. Stochastic gradient descent (SGD) is an effective algorithm to build an LF model, yet its convergence rate depends vastly on the learning rate which should be tuned with care. Therefore, automatic selection of an optimal learning rate for an SGD-based LF model is a meaningful issue. To address it, this study incorporates the principle of particle swarm optimization (PSO) into an SGD-based LF model for searching an optimal learning rate automatically. With it, we further propose an adaptive Latent Factor (ALF) model. Empirical studies on four HiDS matrices from real industrial applications indicate that an ALF model obvious outperforms an LF model according to convergence rate, and maintain competitive prediction accuracy for missing data. (C) 2019 Elsevier B.V. All rights reserved. |
关键词 | Latent factor analysis Particle swarm optimization High-dimensional and sparse matrix Stochastic gradient descent Self-adaptive model |
DOI | 10.1016/j.neucom.2019.08.052 |
发表期刊 | NEUROCOMPUTING |
ISSN | 0925-2312 |
卷号 | 369页码:176-184 |
通讯作者 | Luo, Xin(luoxin21@cigit.ac.cn) |
收录类别 | SCI |
WOS记录号 | WOS:000492298500016 |
语种 | 英语 |