Stochastic trace estimation for positive trace-class operators

arXiv:2608.28135 2026 Optimization 1 ideas extracted · analyzed Sep 2, 2026

What the math gives to ML

The paper develops matrix-free trace estimators for positive trace-class operators, extending Hutchinson and Hutch++ to infinite-dimensional settings. The transferable asset is variance reduction from projecting out a low-rank subspace discovered through operator-vector products, reducing the number of expensive products needed for reliable trace estimation from roughly O(epsilon^-2) to O(epsilon^-1) in the positive-semidefinite case. A neural-network use is to estimate the trace of a damped Hessian or empirical Fisher through Hessian-vector products, enabling curvature-controlled learning-rate schedules or regularization without materializing the matrix. The most defensible first transfer is a Hutch++ curvature monitor whose predictions can be tested against ordinary Hutchinson at equal Hessian-vector-product cost.

Ideas from this paper

Unverified 2026

Hutch++ Curvature Controller

Replace the noisy Hutchinson estimate of a neural-network Hessian trace with a variance-reduced Hutch++ estimate computed only from Hessian-vector products. Use the estimated normalized curvature to cap or rescale the optimizer step, so learning-rate reductions occur when the loss landscape becomes globally sharp rather than when an individual minibatch gradient happens to be large.

Useful6/10
Difficulty5/10
Novelty6/10
Paper: Stochastic trace estimation for positive trace-class operators arXiv:2608.28135