[2D4-GS-2] Machine learning

[2D4-GS-2-03] Analysis of Weight Balancing on Long-Tailed Recognition Problem

〇Naoya Hasegawa¹, Issei Sato¹ (1. Graduate School of Information and Technology, The University of Tokyo)

Keywords:Long-Tailed Recognition, Imbalanced Data, Image Classification, Weight Decay, Regularization

Long-tailed recognition, where the per-class sample size is highly skewed, has recently gained in importance and is challenging because the accuracy of data belonging to classes with a few samples deteriorates in naive training. Two-stage training with weight decay and weight clipping has been proposed to improve the accuracy of long-tailed data.
However, this method requires tuning many hyperparameters in the second stage of training, and why it is effective for long-tailed data is unknown. We analyzed the algorithm and found that it can be decomposed into the increase in the FDR of the feature extractor by weight decay and logit adjustment by weight decay and weight clipping. On the basis of this analysis, we propose a training algorithm without the second stage that results in both improved accuracy and simplification.