Accelerated Convergence of Stochastic Heavy Ball Method under Anisotropic Gradient Noise - Pan, Rui; Liu, Yuxing; Wang, Xiaoyu; Zhang, Tong (International Conference on Learning Representations, 2024)