Spotlight
Single Pass Entrywise-Transformed Low Rank Approximation
Yifei Jiang · Yi Li · Yiming Sun · Jiaxin Wang · David Woodruff
Abstract:
In applications such as natural language processing or computer vision, one is given a large n×nn×n matrix A=(ai,j)A=(ai,j) and would like to compute a matrix decomposition, e.g., a low rank approximation, of a function f(A)=(f(ai,j))f(A)=(f(ai,j)) applied entrywise to AA. A very important special case is the likelihood function f(A)=log(|aij|+1)f(A)=log(∣∣aij∣∣+1). A natural way to do this would be to simply apply ff to each entry of AA, and then compute the matrix decomposition, but this requires storing all of AA as well as multiple passes over its entries. Recent work of Liang et al. shows how to find a rank-kk factorization to f(A)f(A) using only n⋅\poly(\eps−1klogn)n⋅\poly(\eps−1klogn) words of memory, with overall error 10‖f(A)−[f(A)]k‖2F+\poly(ϵ/k)‖f(A)‖21,210∥f(A)−[f(A)]k∥2F+\poly(ϵ/k)∥f(A)∥21,2, where [f(A)]k[f(A)]k is the best rank-kk approximation to f(A)f(A) and ‖f(A)‖21,2∥f(A)∥21,2 is the square of the sum of Euclidean lengths of rows of f(A)f(A). Their algorithm uses 33 passes over the entries of AA. The authors pose the open question of obtaining an algorithm with n⋅\poly(\eps−1klogn)n⋅\poly(\eps−1klogn) words of memory using only a single pass over the entries of AA.
In this paper we resolve this open question, obtaining the first single-pass algorithm for this problem and for the same class of functions ff studied by Liang et al. Moreover, our error is ‖f(A)−[f(A)]k‖2F+\poly(ϵ/k)‖f(A)‖2F∥f(A)−[f(A)]k∥2F+\poly(ϵ/k)∥f(A)∥2F, where ‖f(A)‖2F∥f(A)∥2F is the sum of squares of Euclidean lengths of rows of f(A)f(A). Thus our error is significantly smaller, as it removes the factor of 1010 and also ‖f(A)‖2F≤‖f(A)‖21,2∥f(A)∥2F≤∥f(A)∥21,2.
Chat is not available.