arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2009.02189cs.CV
收藏

基于互补交叉熵的不平衡图像分类

Imbalanced Image Classification with Complement Cross Entropy

Gwangju Institute of Science and Technology (GIST)(光州科学技术院)

Yechan Kim, Younkwan Lee, Moongu Jeon

摘要:

近年来,深度学习模型在计算机视觉应用中取得了巨大的成功,这依赖于大规模的类平衡数据集。然而,不平衡的类分布仍然限制了这些模型的广泛适用性,由于性能下降。为了解决这个问题,本文重点研究了交叉熵,它大多忽略了错误类的输出分数。这项工作发现,中和不正确的类的预测概率,提高了不平衡的图像分类的预测精度。基于这一发现,本文提出了一种简单而有效的损失--互补交叉熵。所提出的损失使得地面真值类在softmax概率方面压倒其他类,通过中和不正确类的概率,而无需额外的训练过程。与此同时,这种损失有助于模型学习关键信息,特别是从少数类的样本中学习。它确保了在不平衡分布上更准确和鲁棒的分类结果。在不平衡数据集上的实验证明了该方法的有效性。

英文摘要:

Recently, deep learning models have achieved great success in computer vision applications, relying on large-scale class-balanced datasets. However, imbalanced class distributions still limit the wide applicability of these models due to degradation in performance. To solve this problem, in this paper, we concentrate on the study of cross entropy which mostly ignores output scores on incorrect classes. This work discovers that neutralizing predicted probabilities on incorrect classes improves the prediction accuracy for imbalanced image classification. This paper proposes a simple but effective loss named complement cross entropy based on this finding. The proposed loss makes the ground truth class overwhelm the other classes in terms of softmax probability, by neutralizing probabilities of incorrect classes, without additional training procedures. Along with it, this loss facilitates the models to learn key information especially from samples on minority classes. It ensures more accurate and robust classification results on imbalanced distributions. Extensive experiments on imbalanced datasets demonstrate the effectiveness of the proposed method.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑