arXiv reaDer
ロングテール学習のための分布ロバストネス損失
Distributional Robustness Loss for Long-tail Learning
実世界のデータは、多くの場合、不均衡でロングテールですが、深いモデルは、頻繁なクラスが存在する場合、まれなクラスを認識するのに苦労します。不均衡なデータに対処するために、ほとんどの研究では、データ、損失、または分類子のバランスを取り、ヘッドクラスに対する分類バイアスを減らします。不均衡なデータで学習された潜在的な表現には、あまり注意が払われていません。ディープネットワークの特徴抽出部分がこのバイアスに大きく苦しんでいることを示します。ロバストネス理論に基づく新しい損失を提案します。これにより、モデルはヘッドクラスとテールクラスの両方で高品質の表現を学習することができます。ロバストネス損失の一般的な形式を計算するのは難しいかもしれませんが、効率的に最小化できる計算しやすい上限をさらに導き出します。この手順により、特徴空間のヘッドクラスに対する表現の偏りが減少し、CIFAR100-LT、ImageNet-LT、およびiNaturalistロングテールベンチマークで新しいSOTA結果が得られます。堅牢性を備えたトレーニングは、ヘッドクラスの精度を大幅に維持しながら、テールクラスの認識精度を向上させることがわかります。新しいロバストネス損失は、さまざまな分類器バランシング手法と組み合わせることができ、ディープモデルのいくつかのレイヤーでの表現に適用できます。
Real-world data is often unbalanced and long-tailed, but deep models struggle to recognize rare classes in the presence of frequent classes. To address unbalanced data, most studies try balancing the data, the loss, or the classifier to reduce classification bias towards head classes. Far less attention has been given to the latent representations learned with unbalanced data. We show that the feature extractor part of deep networks suffers greatly from this bias. We propose a new loss based on robustness theory, which encourages the model to learn high-quality representations for both head and tail classes. While the general form of the robustness loss may be hard to compute, we further derive an easy-to-compute upper bound that can be minimized efficiently. This procedure reduces representation bias towards head classes in the feature space and achieves new SOTA results on CIFAR100-LT, ImageNet-LT, and iNaturalist long-tail benchmarks. We find that training with robustness increases recognition accuracy of tail classes while largely maintaining the accuracy of head classes. The new robustness loss can be combined with various classifier balancing techniques and can be applied to representations at several layers of the deep model.
updated: Sat Oct 30 2021 22:31:51 GMT+0000 (UTC)
published: Wed Apr 07 2021 11:34:04 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アソシエイト