Improved SOGD algorithm and it does reduce the loss function now