Title
A Distributed Coordinate Descent Algorithm for Learning Factorization Machine.
Abstract
Although much effort has been made to implement Factorization Machine (FM) on distributed frameworks, most of them achieve bad model performance or low efficiency. In this paper, we propose a new distributed block coordinate descent algorithm to learn FM. In addition, a distributed pre-computation mechanism incorporated with an optimized Parameter Server framework is designed to avoid the massive repetitive calculations and further reduce the communication cost. Systematically, we evaluate the proposed distributed algorithm on three different genres of datasets for prediction. The experimental results show that the proposed algorithm achieves significantly better performance (3.8%–6.0% RMSE) than the state-of-the-art baselines, and also achieves a 4.6–12.3\\(\\times \\) speedup when reaching a comparable performance.
Year
DOI
Venue
2020
10.1007/978-3-030-47436-2_66
PAKDD (2)
DocType
Citations 
PageRank 
Conference
0
0.34
References 
Authors
0
5
Name
Order
Citations
PageRank
Kankan Zhao1111.46
Jing Zhang2373101.39
Liangfu Zhang300.34
Cuiping Li4399.19
Hong Chen535938.55