Comparison with the vanilla SGD baseline