A Distributed Frank-Wolfe Framework for Learning Low-Rank Matrices with the Trace Norm - Inria - Institut national de recherche en sciences et technologies du numérique Accéder directement au contenu
Rapport (Rapport De Recherche) Année : 2017

A Distributed Frank-Wolfe Framework for Learning Low-Rank Matrices with the Trace Norm

Résumé

We consider the problem of learning a high-dimensional but low-rank matrix from a large-scale dataset distributed over several machines, where low-rankness is enforced by a convex trace norm constraint. We propose DFW-Trace, a distributed Frank-Wolfe algorithm which leverages the low-rank structure of its updates to achieve efficiency in time, memory and communication usage. The step at the heart of DFW-Trace is solved approximately using a distributed version of the power method. We provide a theoretical analysis of the convergence of DFW-Trace, showing that we can ensure sublinear convergence in expectation to an optimal solution with few power iterations per epoch. We implement DFW-Trace in the Apache Spark distributed programming framework and validate the usefulness of our approach on synthetic and real data, including the ImageNet dataset with high-dimensional features extracted from a deep neural network.
Fichier principal
Vignette du fichier
main.pdf (533.01 Ko) Télécharger le fichier
Origine : Fichiers produits par l'(les) auteur(s)
Loading...

Dates et versions

hal-01672066 , version 1 (23-12-2017)

Identifiants

Citer

Wenjie Zheng, Aurélien Bellet, Patrick Gallinari. A Distributed Frank-Wolfe Framework for Learning Low-Rank Matrices with the Trace Norm. [Research Report] Inria Lille. 2017, pp.1-19. ⟨hal-01672066⟩
461 Consultations
134 Téléchargements

Altmetric

Partager

Gmail Facebook X LinkedIn More