High Probability Bounds for Non-Convex Stochastic Optimization with Momentum - Li, Shaojie; Tang, Pengwei; Zhu, Bowei; Liu, Yong (International Conference on Learning Representations, 2026)