arXiv reaDer
生物医学的画像セグメンテーションのための損失関数への境界の不確実性の組み込み
Incorporating Boundary Uncertainty into loss functions for biomedical image segmentation
手動セグメンテーションは、自動画像セグメンテーションタスクでニューラルネットワークを評価するためのゴールドスタンダードとして使用されます。形状、色、テクスチャがかなり不均一であるため、生物医学画像ではオブジェクトの境界を区切ることが特に難しく、評価者間および評価者内で大きなばらつきが生じます。ソフトラベリングや距離ペナルティ項などのアプローチは、グラウンドトゥルースにグローバル変換を適用し、不確実性に関して損失関数を再定義します。ただし、グローバル操作は計算コストが高く、どちらのアプローチも手動注釈の根底にある不確実性を正確に反映していません。この論文では、形態学的操作を使用してソフトラベリングをオブジェクト境界に制限し、グラウンドトゥルースラベルの不確実性を適切に表現し、体系的な手動セグメンテーションエラーが存在するロバストなモデルトレーニングを可能にする境界不確実性を提案します。境界の不確実性とダイスの損失を組み込み、ソフトラベリングや距離加重ペナルティと比較して、3つの十分に検証された生物医学画像データセット全体で一貫して改善されたパフォーマンスを実現します。境界の不確実性は、セグメンテーションプロセスをより正確に反映するだけでなく、効率的で、セグメンテーションエラーに対して堅牢であり、より優れた一般化を示します。
Manual segmentation is used as the gold-standard for evaluating neural networks on automated image segmentation tasks. Due to considerable heterogeneity in shapes, colours and textures, demarcating object boundaries is particularly difficult in biomedical images, resulting in significant inter and intra-rater variability. Approaches, such as soft labelling and distance penalty term, apply a global transformation to the ground truth, redefining the loss function with respect to uncertainty. However, global operations are computationally expensive, and neither approach accurately reflects the uncertainty underlying manual annotation. In this paper, we propose the Boundary Uncertainty, which uses morphological operations to restrict soft labelling to object boundaries, providing an appropriate representation of uncertainty in ground truth labels, and may be adapted to enable robust model training where systematic manual segmentation errors are present. We incorporate Boundary Uncertainty with the Dice loss, achieving consistently improved performance across three well-validated biomedical imaging datasets compared to soft labelling and distance-weighted penalty. Boundary Uncertainty not only more accurately reflects the segmentation process, but it is also efficient, robust to segmentation errors and exhibits better generalisation.
updated: Sun Oct 31 2021 16:19:57 GMT+0000 (UTC)
published: Sun Oct 31 2021 16:19:57 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アソシエイト