DFacTo: Distributed Factorization of Tensors

By Joon Hee Choi, S. V. N. Vishwanathan. 2014

161 citations recorded by the source corpus.

We present a technique for significantly speeding up Alternating Least Squares (ALS) and Gradient Descent (GD), two widely used algorithms for tensor factorization. By exploiting properties of the Khatri-Rao product, we show how to efficiently address a computationally challenging sub-step of both algorithms. Our algorithm, DFacTo, only requires two sparse matrix-vector products and is easy to parallelize. DFacTo is not only scalable but also on average 4 to 10 times faster than competing algorithms on a variety of datasets. For instance, DFacTo only takes 480 seconds on 4 machines to perform one iteration of the ALS algorithm and 1,143 seconds to perform one iteration of the GD algorithm on a 6.5 million x 2.5 million x 1.5 million dimensional tensor with 1.2 billion non-zero entries.

Subfields: Computational Mathematics, Computer Vision and Pattern Recognition, Hardware and Architecture

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Computational Mathematics

  1. Andrzej S. Cichocki
  2. Qibin Zhao
  3. Anh Huy Phan
  4. Danilo P. Mandic
  5. Xiaoqian Jiang
  6. Namgil Lee
  7. Hwanjo Yu
  8. Yejin Kim

In Computer Vision and Pattern Recognition

  1. Qibin Zhao
  2. Jian Lou
  3. Andrzej S. Cichocki
  4. Joyce Ho
  5. Hwanjo Yu
  6. Ivan V. Oseledets
  7. Jimeng Sun
  8. R. Vuduc

In Hardware and Architecture

  1. Andrzej S. Cichocki
  2. Anh Huy Phan
  3. Qibin Zhao
  4. Joyce Ho
  5. Ivan V. Oseledets
  6. Jimeng Sun
  7. R. Vuduc
  8. Xiaoqian Jiang

Citations in the corpus