A Learning-Rate Schedule for Stochastic Gradient Methods to Matrix Factorization
Abstract
Stochastic gradient methods are effective to solve matrix factorization problems. However, it is well known that the performance of stochastic gradient method highly depends on the learning rate schedule used; a good schedule can significantly boost the training process. In this paper, motivated from past works on convex optimization which assign a learning rate for each variable, we propose a new schedule for matrix factorization. The experiments demonstrate that the proposed schedule leads to faster convergence than existing ones. Our schedule uses the same parameter on all data sets included in our experiments; that is, the time spent on learning rate selection can be significantly reduced. By applying this schedule to a state-of-the-art matrix factorization package, the resulting implementation outperforms available parallel matrix factorization packages.
Document Details
- Document Type
- Technical Report
- Publication Date
- Apr 17, 2015
- Accession Number
- AD1015881
Entities
People
- Chih-jen Lin
- Wei-sheng Chin
- Yong Zhuang
- Yu-chin Juan
Organizations
- National Taiwan University