A high-performance implementation of Bayesian matrix factorization with limited communication

Tom Vander Aa*, Xiangju Qin, Paul Blomstedt, Roel Wuyts, Wilfried Verachtert, Samuel Kaski

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference article in proceedingsScientificpeer-review

1 Citation (Scopus)

Abstract

Matrix factorization is a very common machine learning technique in recommender systems. Bayesian Matrix Factorization (BMF) algorithms would be attractive because of their ability to quantify uncertainty in their predictions and avoid over-fitting, combined with high prediction accuracy. However, they have not been widely used on large-scale data because of their prohibitive computational cost. In recent work, efforts have been made to reduce the cost, both by improving the scalability of the BMF algorithm as well as its implementation, but so far mainly separately. In this paper we show that the state-of-the-art of both approaches to scalability can be combined. We combine the recent highly-scalable Posterior Propagation algorithm for BMF, which parallelizes computation of blocks of the matrix, with a distributed BMF implementation that users asynchronous communication within each block. We show that the combination of the two methods gives substantial improvements in the scalability of BMF on web-scale datasets, when the goal is to reduce the wall-clock time.

Original languageEnglish
Title of host publicationComputational Science – ICCS 2020 - 20th International Conference, Proceedings
EditorsValeria V. Krzhizhanovskaya, Gábor Závodszky, Michael H. Lees, Peter M.A. Sloot, Peter M.A. Sloot, Peter M.A. Sloot, Jack J. Dongarra, Sérgio Brissos, João Teixeira
PublisherSpringer
Pages3-16
Number of pages14
ISBN (Print)9783030504328
DOIs
Publication statusPublished - 1 Jan 2020
MoE publication typeA4 Conference publication
EventInternational Conference on Computational Science - Amsterdam, Netherlands
Duration: 3 Jun 20205 Jun 2020
Conference number: 20

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume12142 LNCS
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

ConferenceInternational Conference on Computational Science
Abbreviated titleICCS
Country/TerritoryNetherlands
CityAmsterdam
Period03/06/202005/06/2020

Funding

The research leading to these results has received funding from the European Union?s Horizon2020 research and innovation programme under the EPEEC project, grant agreement No 801051. The work was also supported by the Academy of Finland (Flagship programme: Finnish Center for Artificial Intelligence, FCAI; grants 319264, 292334). We acknowledge PRACE for awarding us access to Hazel Hen at GCS@HLRS, Germany.

Fingerprint

Dive into the research topics of 'A high-performance implementation of Bayesian matrix factorization with limited communication'. Together they form a unique fingerprint.
  • Interactive machine learning from multiple biodata sources

    Kaski, S. (Principal investigator), Bhat, A. (Project Member), Trinh, T. (Project Member), Scherting, B. (Project Member), Siren, J. (Project Member), Gadd, C. (Project Member), Hegde, P. (Project Member), Chauhan, R. (Project Member), Jain, A. (Project Member), Jälkö, J. (Project Member), Hämäläinen, A. (Project Member), Tran, A. (Project Member) & Shen, Z. (Project Member)

    01/01/201931/08/2021

    Project: Academy of Finland: Other research funding

  • Interactive machine learning from multiple biodata sources

    Kaski, S. (Principal investigator) & Filstroff, L. (Project Member)

    01/01/201631/08/2021

    Project: Academy of Finland: Other research funding

Cite this