Sharpness-Aware Minimization Enhances Feature Quality via Balanced Learning - Springer, Jacob; Nagarajan, Vaishnavh; Raghunathan, Aditi (International Conference on Learning Representations, 2024)