Generalised Balanced Softmax: A Finite-Data Perspective on Logit Adjustment for Long-Tailed Recognition
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 . 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 , which is algebraically identical to the training-time logit-adjusted loss of Menon et al. (2021) when . The case 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, 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 and improve average testing recall relative to the reference, while retaining negligible computational overhead and compatibility with existing representation-learning frameworks.