CSpace
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
DOI10.1016/j.neucom.2019.08.052
发表期刊NEUROCOMPUTING
ISSN0925-2312
卷号369页码:176-184
通讯作者Luo, Xin(luoxin21@cigit.ac.cn)
收录类别SCI
WOS记录号WOS:000492298500016
语种英语