On regularization of gradient descent, layer imbalance and flat minima

Re-assign community