arXiv reaDer
不均衡な深層学習のためのプロクラステストレーニング
Procrustean Training for Imbalanced Deep Learning
クラスの不均衡なデータでトレーニングされたニューラルネットワークは、不足しているトレーニングデータのマイナークラスではパフォーマンスが低下することが知られています。最近のいくつかの研究では、これはマイナークラスへの過剰適合に起因するとされています。この論文では、この問題の新しい説明を提供します。ニューラルネットワークは、初期のトレーニングエポックでほとんどのデータをメジャークラスに分類することにより、最初にマイナークラスに適合しない傾向があることがわかりました。これらの誤った予測を修正するには、ニューラルネットワークは、メジャークラスとマイナークラスの間の決定境界を越えてマイナークラスデータの特徴をプッシュすることに焦点を当てる必要があり、マイナークラスの特徴の勾配がはるかに大きくなります。このような適合不足フェーズは、メジャークラスとマイナークラス間の競争を過度に強調し、ニューラルネットワークがテストデータに一般化できる識別知識を学習するのを妨げ、最終的には過剰適合になると主張します。この問題に対処するために、クラス間でトレーニングの進捗状況を均等化するための新しい学習戦略を提案します。主要なクラスのデータの特徴を他のデータの特徴とミニバッチで混合し、それらの特徴を意図的に弱めて、ニューラルネットワークが最初にそれらを適合させないようにします。この戦略により、クラス間でトレーニングの精度と機能の勾配のバランスを大幅にとることができ、マイナークラスデータの過適合と過適合の問題を効果的に軽減できることを示します。いくつかのベンチマークデータセットで、私たちのアプローチは、特に困難なステップ不均衡の場合に、最先端の精度を達成します。
Neural networks trained with class-imbalanced data are known to perform poorly on minor classes of scarce training data. Several recent works attribute this to over-fitting to minor classes. In this paper, we provide a novel explanation of this issue. We found that a neural network tends to first under-fit the minor classes by classifying most of their data into the major classes in early training epochs. To correct these wrong predictions, the neural network then must focus on pushing features of minor class data across the decision boundaries between major and minor classes, leading to much larger gradients for features of minor classes. We argue that such an under-fitting phase over-emphasizes the competition between major and minor classes, hinders the neural network from learning the discriminative knowledge that can be generalized to test data, and eventually results in over-fitting. To address this issue, we propose a novel learning strategy to equalize the training progress across classes. We mix features of the major class data with those of other data in a mini-batch, intentionally weakening their features to prevent a neural network from fitting them first. We show that this strategy can largely balance the training accuracy and feature gradients across classes, effectively mitigating the under-fitting then over-fitting problem for minor class data. On several benchmark datasets, our approach achieves the state-of-the-art accuracy, especially for the challenging step-imbalanced cases.
updated: Sun Oct 10 2021 17:53:22 GMT+0000 (UTC)
published: Mon Apr 05 2021 04:44:01 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アソシエイト