arXiv · 2503.05289
An Analytical Model for Overparameterized Learning Under Class Imbalance
Abstract
We study class-imbalanced linear classification in a high-dimensional Gaussian mixture model. We develop a tight, closed form approximation for the test error of several practical learning methods, including logit adjustment and class dependent temperature. Our approximation allows us to analytically tune and compare these methods, highlighting how and when they overcome the pitfalls of standard cross-entropy minimization. We test our theoretical findings on simulated data and imbalanced CIFAR10, MNIST and FashionMNIST datasets.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Eliav Mor, Yair Carmon. 2025-03-07. An Analytical Model for Overparameterized Learning Under Class Imbalance. https://arxiv.org/abs/2503.05289
Cite the original work for its findings. Save a collection to share your selection of sources.