生物医学画像セグメンテーションのための地域ごとの(RW)損失を提案します。領域ごとの損失は用途が広く、クラスの不均衡とピクセルの重要性を同時に説明でき、softmax出力とRWマップの間のピクセルごとの乗算として簡単に実装できます。提案されたRW損失フレームワークの下で、アクティブコンターや境界損失などの特定の損失関数を適切なRWマップで同様に再定式化できることを示します。これにより、これらの損失関数を理解するための基本的な類似点と新しい視点が明らかになります。境界損失距離マップなどの特定のRWマップによって引き起こされる観測された最適化の不安定性を調査し、そのような不安定性を回避するために数学的に根拠のある原理を導入します。この原則は、あらゆるデータセットへの優れた適応性を提供し、余分な正則化項や最適化のトリックなしで実質的に収束を保証します。この原則に従って、修正されたリージョンワイズ(RRW)マップと呼ばれる境界距離マップの単純なバージョンを提案します。これは、実験で示すように、同等以上のダイス係数とハウスドルフ距離で最先端のパフォーマンスを実現します。 3つの異なるセグメンテーションタスクにおけるダイス、フォーカル、加重クロスエントロピー、および境界損失。境界損失距離マップによって提供される最適化の不安定性を定量化し、RRWマップが最適化に対して安定していることを経験的に示します。すべての実験を実行するためのコードは、https://github.com/jmlipman/RegionWiseLossで公開されています。
We propose Region-wise (RW) loss for biomedical image segmentation. Region-wise loss is versatile, can simultaneously account for class imbalance and pixel importance, and it can be easily implemented as the pixel-wise multiplication between the softmax output and a RW map. We show that, under the proposed RW loss framework, certain loss functions, such as Active Contour and Boundary loss, can be reformulated similarly with appropriate RW maps, thus revealing their underlying similarities and a new perspective to understand these loss functions. We investigate the observed optimization instability caused by certain RW maps, such as Boundary loss distance maps, and we introduce a mathematically-grounded principle to avoid such instability. This principle provides excellent adaptability to any dataset and practically ensures convergence without extra regularization terms or optimization tricks. Following this principle, we propose a simple version of boundary distance maps called rectified Region-wise (RRW) maps that, as we demonstrate in our experiments, achieve state-of-the-art performance with similar or better Dice coefficients and Hausdorff distances than Dice, Focal, weighted Cross entropy, and Boundary losses in three distinct segmentation tasks. We quantify the optimization instability provided by Boundary loss distance maps, and we empirically show that our RRW maps are stable to optimize. The code to run all our experiments is publicly available at: https://github.com/jmlipman/RegionWiseLoss.