arXiv reaDer
補数クロスエントロピーによる不均衡な画像分類
Imbalanced Image Classification with Complement Cross Entropy
最近、深層学習モデルは、大規模なクラスバランスの取れたデータセットに依存して、コンピュータービジョンアプリケーションで大きな成功を収めています。ただし、不均衡なクラス分布は、パフォーマンスの低下により、これらのモデルの幅広い適用性を制限します。この問題を解決するために、この論文では、誤ったクラスの出力スコアをほとんど無視するクロスエントロピーの研究に焦点を当てます。この作業は、誤ったクラスで予測された確率を中和することで、不均衡な画像分類の予測精度が向上することを発見しました。この論文は、この発見に基づいて、補数交差エントロピーと名付けられた単純だが効果的な損失を提案している。提案された損失は、追加のトレーニング手順なしで、誤ったクラスの確率を中和することにより、ソフトマックス確率の点でグラウンドトゥルースクラスを他のクラスに圧倒させます。それに加えて、この損失により、モデルは特にマイノリティクラスのサンプルから重要な情報を学習しやすくなります。これにより、不均衡な分布に対してより正確で堅牢な分類結果が保証されます。不均衡なデータセットに関する広範な実験は、提案された方法の有効性を示しています。
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.
updated: Wed Aug 04 2021 10:52:44 GMT+0000 (UTC)
published: Fri Sep 04 2020 13:46:24 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト