cs.LGJul 24, 2026

Generalised Balanced Softmax: A Finite-Data Perspective on Logit Adjustment for Long-Tailed Recognition

Authors: Yi-Hang ZhuRajeev RamanShiqi SuJianyuan SunXinyu YangNan XingHuiyu Zhou

Organizations: School of Computing and Mathematical Sciences, University of Leicester, Leicester, LE1 7RH, UK · Scientific Computing, Rutherford Appleton Laboratory, Science and Technology Facilities Council, Didcot, OX11 0QX, UK · School of Computer Science and Informatics, De Montfort University, Leicester, LE1 9BH, UK · School of Automation and Information Engineering, Xi’an University of Technology, Xi’an, 710048, China

Abstract

Models trained on long-tailed data using standard softmax tend to exhibit higher training error and a larger generalisation gap for classes with fewer training samples. We characterise this class-wise disparity as the preference issue and quantify it using a new metric, the model imbalance level II. To understand this issue, we analyse how imbalanced training data adversely affects class-wise gradients under standard softmax training. This paper then develops a finite-data Generalised Balanced Softmax (GBS) framework for analysing and mitigating the preference issue. The framework uses the training-time logit adjustment znc+βlogNcz_{nc}+β\log|N_c|, which is algebraically identical to the training-time logit-adjusted loss of Menon et al. (2021) when τ=βτ=β. The case β=1β=1 also coincides with Balanced Softmax and with the unit adjustment supported by the Fisher-consistency argument under the true data distribution, corresponding to an idealised infinite-data setting. Building on this existing loss family, this paper uses a heuristic power-law assumption to motivate the adjustable coefficient and studies how ββ affects trained models. Across the evaluated long-tailed benchmarks, β=1β=1 does not attain the highest average testing recall on most datasets, showing that a different coefficient can be preferable when training on finite data. The selected values of ββ reduce II and improve average testing recall relative to the β=1β=1 reference, while retaining negligible computational overhead and compatibility with existing representation-learning frameworks.

Explore similar work

CardsList
  1. A Strong Balanced-Softmax Classifier-Retraining Baseline for Long-Tailed Recognition

    Jul 10, 2026Juan Terven, Diana Margarita Córdova Esparza, Julio Alejandro Romero Gonzalez +4Few-Shot LearningExtreme Class Imbalance