arXiv reaDer
ロングテール認識にはエンドツーエンドのトレーニングのみが必要です
You Only Need End-to-End Training for Long-Tailed Recognition
ロングテールデータセットの一般化のギャップは、主に、ほとんどのカテゴリが少数のトレーニングサンプルしか占有していないためです。分離されたトレーニングは、バックボーンと分類器を別々にトレーニングすることにより、パフォーマンスを向上させます。エンドツーエンドのモデルトレーニング(ロジットマージンベースの方法など)のパフォーマンスが低下する原因は何ですか?この作業では、分類器の学習に影響を与える重要な要素を特定します。分類器に入力する前に、エントロピーが低いチャネル相関特徴です。情報理論の観点から、クロスエントロピー損失が不均衡なデータに高度に相関する特徴を生成する傾向がある理由を分析します。さらに、分類器の重みの勾配、ヘッセ行列の条件数、およびロジットのマージンベースのアプローチへの影響を理論的に分析して証明します。したがって、最初に、チャネルホワイトニングを使用して、分類器の入力を非相関化(「分散」)して、重みの更新を分離し、偏った決定境界を再形成することを提案します。これにより、ロジットマージンベースの方法と組み合わせて満足のいく結果が得られます。ただし、マイナークラスの数が多い場合、バッチの不均衡とトレーニングへの参加の増加により、メジャークラスの過剰適合が発生します。また、上記の問題を解決するために、ブロックベースの相対バランスバッチサンプラー(B3RS)とバッチ埋め込みトレーニング(BET)の2つの新しいモジュールを提案します。これにより、エンドツーエンドのトレーニングは、分離されたトレーニングよりも優れたパフォーマンスを実現します。ロングテール分類ベンチマークであるCIFAR-LTとImageNet-LTの実験結果は、私たちの方法の有効性を示しています。
The generalization gap on the long-tailed data sets is largely owing to most categories only occupying a few training samples. Decoupled training achieves better performance by training backbone and classifier separately. What causes the poorer performance of end-to-end model training (e.g., logits margin-based methods)? In this work, we identify a key factor that affects the learning of the classifier: the channel-correlated features with low entropy before inputting into the classifier. From the perspective of information theory, we analyze why cross-entropy loss tends to produce highly correlated features on the imbalanced data. In addition, we theoretically analyze and prove its impacts on the gradients of classifier weights, the condition number of Hessian, and logits margin-based approach. Therefore, we firstly propose to use Channel Whitening to decorrelate ("scatter") the classifier's inputs for decoupling the weight update and reshaping the skewed decision boundary, which achieves satisfactory results combined with logits margin-based method. However, when the number of minor classes are large, batch imbalance and more participation in training cause over-fitting of the major classes. We also propose two novel modules, Block-based Relatively Balanced Batch Sampler (B3RS) and Batch Embedded Training (BET) to solve the above problems, which makes the end-to-end training achieve even better performance than decoupled training. Experimental results on the long-tailed classification benchmarks, CIFAR-LT and ImageNet-LT, demonstrate the effectiveness of our method.
updated: Wed Dec 15 2021 11:40:54 GMT+0000 (UTC)
published: Sat Dec 11 2021 11:44:09 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アソシエイト