Stochastic diagonal estimation: probabilistic bounds and an improved algorithm. - Robert A. Baston; Yuji Nakatsukasa (CoRR, 2022)