Hutchinson's Estimator is Bad at Kronecker-Trace-Estimation
Not on the map. Why?
Abstract (arXiv)
We study the problem of estimating the trace of a matrix A that can only be accessed through Kronecker-matrix-vector products. That is, for any Kronecker-structured vector x = otimesi=1k xi, we can compute Ax. We focus on the natural generalization of Hutchinson's Estimator to this setting, proving tight rates for the number of matrix-vector products this estimator needs to find a (1±varepsilon) approximation to the trace of A. We find an exact equation for the variance of the estimator when using a Kronecker of Gaussian vectors, revealing an intimate relationship between Hutchinson's Estimator, the partial trace operator, and the partial transpose operator. Using this equation, we show that when using real vectors, in the worst case, this estimator needs O(frac3kvarepsilon2) products to recover a (1±varepsilon) approximation of the trace of any PSD A, and a matching lower bound for certain PSD A. However, when using complex vectors, this can be exponentially improved to Theta(frac2kvarepsilon2). Further, if the xi vectors are low-dimensional and if we instead build x as the Kronecker product of (scaled) random unit vectors on the complex sphere, then as few as frac1.33kvarepsilon2 samples suffice. We show that Hutchinson's Estimator converges slowest when A itself also has Kronecker structure. We conclude with some theoretical evidence suggesting that, by combining Hutchinson's Estimator with other techniques, it may be possible to avoid the exponential dependence on k.
Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). First posted on arXiv 10 Sep 2023.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026Accurate Trace Estimation with Fewer Random Bits via Recursive TensorSketch16 Sep 2026
- 2026Improving TensorSketch Using Complex Random Variables11 Aug 2026
- 2026Stochastic trace estimation with tensor train random vectors14 Jun 2026
- 2026Efficient DP-SGD for LLMs with Randomized Clipping24 May 2026