learning rate update needs to be more stable