Optimal Stochastic Trace Estimation in Generative Modeling

Xinyang Liu, Hengrong Du, Wei Deng +1

Hutchinson estimators are widely employed in training divergence-based likelihoods for diffusion models to ensure optimal transport (OT) properties. However, this estimator often s…