Improved Stochastic Optimization of LogSumExp