Random Batch Method with Momentum Correction. - Yanshun Zhao; Jingrun Chen; Zhiwen Zhang (CoRR, 2024)