arXiv reaDer
FocusNet:紛らわしいクラスに焦点を当てることでより適切に分類する
FocusNet: Classifying Better by Focusing on Confusing Classes
現在、ほとんどの分類ネットワークは、その単純さから、ワンホットエンコーディングを使用してカテゴリデータを表現しています。ただし、ワンホットエンコーディングはクラス間の相関を無視するため、一般化機能に影響を与える可能性があります。ワンホットラベルでトレーニングされたニューラルネットワークが誤った予測を生成する場合でも、ターゲット画像領域に注意を払い、どのクラスがネットワークを混乱させるかを明らかにします。この観察に触発されて、クラスの混乱の問題に対処するための混乱に焦点を当てたメカニズムを提案します。私たちの混乱に焦点を当てたメカニズムは、2ブランチネットワークアーキテクチャによって実装されています。そのベースラインブランチは紛らわしいクラスを生成し、アーキテクチャが柔軟なそのFocusNetブランチは、これらの紛らわしいクラスから正しいラベルを区別します。また、FocusNetに最も混乱しやすいクラスに焦点を当てるように促すことで、分類の精度を向上させる新しいフォーカスピッキング損失関数を紹介します。実験結果は、FocusNetが一般的なデータセットの画像分類に効果的であり、フォーカスピッキング損失関数が現在のニューラルネットワークの分類精度の向上にも役立つことを検証しています。
Nowadays, most classification networks use one-hot encoding to represent categorical data because of its simplicity. However, one-hot encoding may affect the generalization ability as it neglects inter-class correlations. We observe that, even when a neural network trained with one-hot labels produces incorrect predictions, it still pays attention to the target image region and reveals which classes confuse the network. Inspired by this observation, we propose a confusion-focusing mechanism to address the class-confusion issue. Our confusion-focusing mechanism is implemented by a two-branch network architecture. Its baseline branch generates confusing classes, and its FocusNet branch, whose architecture is flexible, discriminates correct labels from these confusing classes. We also introduce a novel focus-picking loss function to improve classification accuracy by encouraging FocusNet to focus on the most confusing classes. The experimental results validate that our FocusNet is effective for image classification on common datasets, and that our focus-picking loss function can also benefit the current neural networks in improving their classification accuracy.
updated: Wed Apr 20 2022 02:08:08 GMT+0000 (UTC)
published: Thu Oct 14 2021 12:22:59 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アソシエイト