Fast and Guaranteed Tensor Decomposition via Sketching
Wang, Yining, Tung, Hsiao-Yu, Smola, Alexander J., Anandkumar, Anima
–Neural Information Processing Systems
Tensor CANDECOMP/PARAFAC (CP) decomposition has wide applications in statistical learning of latent variable models and in data mining. In this paper, we propose fast and randomized tensor CP decomposition algorithms based on sketching. We build on the idea of count sketches, but introduce many novel ideas which are unique to tensors. We develop novel methods for randomized com- putation of tensor contractions via FFTs, without explicitly forming the tensors. Such tensor contractions are encountered in decomposition methods such as ten- sor power iterations and alternating least squares.
Neural Information Processing Systems
Feb-14-2020, 07:58:50 GMT