arXiv reaDer
Adaptive Class Weight based Dual Focal Loss for Improved Semantic Segmentation
この論文では、データセット内の不均衡なクラスのより良い処理を実現するために、標準のクロスエントロピー(CE)関数の代わりとして、デュアルフォーカルロス(DFL)関数を提案します。私たちのDFL法は、最近報告されたFocal Loss(FL)クロスエントロピー関数を改良したもので、簡単な例よりも分類が難しい例に重点を置くスケーリング方法を提案しています。ただし、FLのスケーリングパラメータは経験的に設定されており、問題に依存します。さらに、他のCEバリアントと同様に、FLは真のクラスの喪失にのみ焦点を当てています。したがって、偽のクラスから損失フィードバックは得られません。ソフトマックス関数の性質上、真の例のみに焦点を当てると真のクラスの確率が高まり、それに応じて偽のクラスの確率が低下しますが、偽のクラスの損失を回避するため、最良の収束は達成されません。私たちのDFL法は、2つの方法で単純なFLを改善します。まず、FLの考え方は、簡単な例よりも難しい例に焦点を当てる必要がありますが、真のクラスと負のクラスの両方の損失を同じ重要性で評価します。次に、DFLのスケーリングパラメータが学習可能になり、手動チューニングに依存するのではなく、バックプロパゲーションによって自身をチューニングできるようになりました。このように、提案されたDFLメソッドは、クラスの不均衡効果を減らすだけでなく、真の難しい例と否定的な簡単な例の両方に焦点を当てることができる自動調整可能な損失関数を提供します。
In this paper, we propose a Dual Focal Loss (DFL) function, as a replacement for the standard cross entropy (CE) function to achieve a better treatment of the unbalanced classes in a dataset. Our DFL method is an improvement on the recently reported Focal Loss (FL) cross-entropy function, which proposes a scaling method that puts more weight on the examples that are difficult to classify over those that are easy. However, the scaling parameter of FL is empirically set, which is problem-dependent. In addition, like other CE variants, FL only focuses on the loss of true classes. Therefore, no loss feedback is gained from the false classes. Although focusing only on true examples increases probability on true classes and correspondingly reduces probability on false classes due to the nature of the softmax function, it does not achieve the best convergence due to avoidance of the loss on false classes. Our DFL method improves on the simple FL in two ways. Firstly, it takes the idea of FL to focus more on difficult examples than the easy ones, but evaluates loss on both true and negative classes with equal importance. Secondly, the scaling parameter of DFL has been made learnable so that it can tune itself by backpropagation rather than being dependent on manual tuning. In this way, our proposed DFL method offers an auto-tunable loss function that can reduce the class imbalance effect as well as put more focus on both true difficult examples and negative easy examples.
updated: Thu Nov 26 2020 05:20:15 GMT+0000 (UTC)
published: Thu Sep 26 2019 06:36:21 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト