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

Nov-19-2025, 18:34:04 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.