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
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