Improving Probabilistic Diffusion Models With Optimal Diagonal Covariance Matching - Ou, Zijing; Zhang, Mingtian; Zhang, Andi; Xiao, Tim; Li, Yingzhen; Barber, David (International Conference on Learning Representations, 2025)