arXiv reaDer
隠されたラベル-セグメンテーション損失のわずかなバイアス
The hidden label-marginal biases of segmentation losses
ほとんどのセグメンテーション損失は、間違いなくクロスエントロピー(CE)またはダイス損失の変形です。豊富なセグメンテーションの文献では、これらの損失のどれがより良い選択であるかについて明確なコンセンサスはなく、さまざまなベンチマークとアプリケーションにわたってそれぞれのパフォーマンスが異なります。この作業では、これら2つのタイプの損失をリンクし、それらの長所と短所を明らかにする理論的分析を開発します。まず、CEとDiceが以前に考えられていたよりもはるかに深い関係を共有していることを示す、制約付き最適化の視点を提供します。これらは両方とも、ラベル限界ペナルティと密接に関連するグラウンドトゥルースマッチングペナルティに分解されます。次に、バインドされた関係と情報理論的分析を提供し、隠れたラベルの限界バイアスを明らかにします。ダイスには特定の極端に不均衡なソリューションに対する固有のバイアスがありますが、CEは暗黙的にグラウンドトゥルース領域の比率を推奨します。私たちの理論的結果は、医用画像の文献における幅広い実験的証拠を説明しており、それによってダイスの損失が不均衡なセグメンテーションの改善をもたらします。また、CEがさまざまなクラス比率で自然画像の問題を支配する理由についても説明します。この場合、Diceはさまざまなラベル周辺分布に適応するのが難しい可能性があります。私たちの理論的分析に基づいて、ラベルの限界バイアスを明示的に制御することを可能にする、原理的で単純なソリューションを提案します。私たちの損失は、CEを明示的なL_1正則化と統合します。これにより、ラベルの周辺分布がターゲットクラスの比率に一致するようになり、一般性を失うことなくクラスの不均衡が緩和されます。さまざまな損失とアプリケーションに関する包括的な実験とアブレーション研究により、理論的分析と、明示的なラベルマージナルレギュラライザーの有効性が検証されます。
Most segmentation losses are arguably variants of the Cross-Entropy (CE) or Dice losses. In the abundant segmentation 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. In this work, we develop a theoretical analysis that links these two types of losses, exposing their advantages and weaknesses. First, we provide a constrained-optimization perspective showing that CE and Dice share a much deeper connection than previously thought: They both decompose into label-marginal penalties and closely related ground-truth matching penalties. Then, we provide bound relationships and an information-theoretic analysis, which uncover hidden label-marginal biases: Dice has an intrinsic bias towards specific extremely 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: Wed Oct 06 2021 00:02:13 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アソシエイト