Variational Schr\"odinger Momentum Diffusion