arXiv reaDer
隠されたラベル-セグメンテーション損失のわずかなバイアス
The hidden label-marginal biases of segmentation losses
ほとんどのセグメンテーション損失は、間違いなくクロスエントロピー(CE)またはダイス損失の変形です。文献では、これらの損失のどれがより良い選択であるかについて明確なコンセンサスはなく、さまざまなベンチマークとアプリケーションにわたってそれぞれのパフォーマンスが異なります。これら2種類の損失を関連付け、それらの長所と短所を明らかにする理論的分析を開発します。まず、CEとダイスが以前に考えられていたよりもはるかに深い関係を共有していることを明示的に示します。CEは対数ダイス損失と線形ダイス損失の両方の上限です。さらに、隠れたラベルの限界バイアスを強調する情報理論的分析を提供します。ダイスには不均衡なソリューションに対する固有のバイアスがありますが、CEは暗黙的にグラウンドトゥルース領域の比率を推奨します。私たちの理論的結果は、医用画像の文献における幅広い実験的証拠を説明しており、それによってダイスの損失が不均衡なセグメンテーションの改善をもたらします。また、CEがさまざまなクラス比率で自然画像の問題を支配する理由についても説明します。この場合、Diceはさまざまなラベル周辺分布に適応するのが難しい可能性があります。私たちの理論的分析に基づいて、ラベルの限界バイアスを明示的に制御することを可能にする、原理的で単純なソリューションを提案します。私たちの損失は、CEを明示的なL_1正則化と統合します。これにより、ラベルの周辺分布がターゲットクラスの比率に一致するようになり、一般性を失うことなくクラスの不均衡が緩和されます。さまざまな損失とアプリケーションに関する包括的な実験とアブレーション研究により、理論的分析と、明示的なラベルマージナルレギュラライザーの有効性が検証されます。
Most segmentation losses are arguably variants of the Cross-Entropy (CE) or Dice loss. In the literature, there is no clear consensus as to which of these losses is a better choice, with varying performances for each across different benchmarks and applications. We develop a theoretical analysis that links these two types of losses, exposing their advantages and weaknesses. First, we explicitly demonstrate that CE and Dice share a much deeper connection than previously thought: CE is an upper bound on both logarithmic and linear Dice losses. Furthermore, we provide an information-theoretic analysis, which highlights hidden label-marginal biases : Dice has an intrinsic bias towards imbalanced solutions, whereas CE implicitly encourages the ground-truth region proportions. Our theoretical results explain the wide experimental evidence in the medical-imaging literature, whereby Dice losses bring improvements for imbalanced segmentation. It also explains why CE dominates natural-image problems with diverse class proportions, in which case Dice might have difficulty adapting to different label-marginal distributions. Based on our theoretical analysis, we propose a principled and simple solution, which enables to control explicitly the label-marginal bias. Our loss integrates CE with explicit L_1 regularization, which encourages label marginals to match target class proportions, thereby mitigating class imbalance but without losing generality. Comprehensive experiments and ablation studies over different losses and applications validate our theoretical analysis, as well as the effectiveness of our explicit label-marginal regularizers.
updated: Sun Apr 18 2021 04:59:39 GMT+0000 (UTC)
published: Sun Apr 18 2021 04:59:39 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アソシエイト