Why is noise scaled by Ntrain in RMSProp about psgld

[ChunyuanLI]

The scaling for the noise is for faster convergence in practice; Otherwise, we need to train the model for a long time according to the theory.

[jaak-s]

Is the choice of Ntrain for the scaling arbitrary? Or do you think it will work in general for almost any dataset?

[ChunyuanLI]

Ntrain is the number of data points in the training dataset.

[jaak-s]

Yes, but you could use other numbers for scaling like a constant number (100) or batch size etc. So my question is whether you expect that Ntrain is a good choice in practice and it will work well for almost any dataset. Or should we try several values for the scaling and choose the best?

[ChunyuanLI]

I expect that Ntrain is a good choice in practice.

The "grad" is mean of the gradients computed in the mini-batch. We should use opts.N*grad to approximate the true gradient of the full dataset.

Instead, we consider the scaling issue in the stepsize "lr", and come to the update as following:

grad = lr* grad ./ pcder + sqrt(2*lr./pcder/opts.N).*randn(size(grad)) ;

However, this would take a long time to converge. In practice, I recommend:

grad = lr* grad ./ pcder + sqrt(2*lr./pcder).*randn(size(grad))/opts.N ;

[jaak-s]

Thank you for the explanation. I saw that also SGLD.m uses the same scaling by opts.N. So in your experience the same slow convergence holds true for the SGLD method too?

[ChunyuanLI]

Yes, the convergence also holds for SGLD.