Data imbalance is one of the most important issues in the process of training deep learning networks. It directly affects the performance of the proposed models. Most recent methods have focused on improving the model’s performance by proposing new rebalance strategies, using more networks, transfer learning, etc. In particular, Balanced Meta-Softmax (BLMS) is a simple method that focuses on adjusting the softmax function to better fit imbalanced data via pre-calculating the number of samples of each parameter. However, this adjustment is constant throughout training. This seems inappropriate when applying data augmentation. In this paper, we proposed a simple yet robust approach to overcome the above drawback by calculating the number of samples that incorrectly predicted the label at epoch \(t-1\) to adjust the balance softmax function at epoch t. The experiment results on various long-tailed datasets have shown that the proposed framework significantly outperforms different methods.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Smooth Balance Softmax for Long-Tailed Image Classification

  • Duc-Quang Vu,
  • Ma T. Hong Thu

摘要

Data imbalance is one of the most important issues in the process of training deep learning networks. It directly affects the performance of the proposed models. Most recent methods have focused on improving the model’s performance by proposing new rebalance strategies, using more networks, transfer learning, etc. In particular, Balanced Meta-Softmax (BLMS) is a simple method that focuses on adjusting the softmax function to better fit imbalanced data via pre-calculating the number of samples of each parameter. However, this adjustment is constant throughout training. This seems inappropriate when applying data augmentation. In this paper, we proposed a simple yet robust approach to overcome the above drawback by calculating the number of samples that incorrectly predicted the label at epoch \(t-1\) to adjust the balance softmax function at epoch t. The experiment results on various long-tailed datasets have shown that the proposed framework significantly outperforms different methods.