Changing the Training Data Distribution to Reduce Simplicity Bias Improves In-distribution Generalization

Oct-10-2025, 07:26:02 GMT–Neural Information Processing Systems 

By studying a two-layer CNN, we rigorously prove that SAM learns different features more uniformly, particularly in early epochs.