CNNセグメンテーションで広く使用されている損失関数(ダイスやクロスエントロピーなど)は、セグメンテーション領域の積分に基づいています。残念ながら、非常に不均衡なセグメンテーションの場合、このような地域の合計の値はクラス間で数桁異なり、トレーニングのパフォーマンスと安定性に影響します。境界損失を提案します。これは、領域ではなく、輪郭の空間での距離メトリックの形式を取ります。これにより、領域全体の不均衡な積分ではなく、領域間のインターフェイス上の積分が使用されるため、高度に不均衡な問題の問題を軽減できます。さらに、境界損失は地域情報を補完します。アクティブな等高線フローを計算するためのグラフベースの最適化手法に着想を得て、等高線の空間上の非対称L_2距離を領域積分として表現します。これにより、等高線点を含む完全に局所的な微分計算が回避されます。これにより、ネットワークの地域ソフトマックス確率出力で表される境界損失が生成されます。これは、標準の地域損失と簡単に組み合わせることができ、NDセグメンテーション用の既存のディープネットワークアーキテクチャで実装できます。さまざまな不均衡な問題に関する包括的な評価と比較を報告し、境界損失がトレーニングの安定性を向上させながらパフォーマンスを大幅に向上させることができることを示しています。私たちのコードは公開されています:https://github.com/LIVIAETS/surface-loss。
Widely used loss functions for CNN segmentation, e.g., Dice or cross-entropy, are based on integrals over the segmentation regions. Unfortunately, for highly unbalanced segmentations, such regional summations have values that differ by several orders of magnitude across classes, which affects training performance and stability. We propose a boundary loss, which takes the form of a distance metric on the space of contours, not regions. This can mitigate the difficulties of highly unbalanced problems because it uses integrals over the interface between regions instead of unbalanced integrals over the regions. Furthermore, a boundary loss complements regional information. Inspired by graph-based optimization techniques for computing active-contour flows, we express a non-symmetric L_2 distance on the space of contours as a regional integral, which avoids completely local differential computations involving contour points. This yields a boundary loss expressed with the regional softmax probability outputs of the network, which can be easily combined with standard regional losses and implemented with any existing deep network architecture for N-D segmentation. We report comprehensive evaluations and comparisons on different unbalanced problems, showing that our boundary loss can yield significant increases in performances while improving training stability. Our code is publicly available: https://github.com/LIVIAETS/surface-loss .