A Weighted Normalized Boundary Loss for Reducing the Hausdorff Distance in Medical Imaging Segmentation

02/08/2023
by   Adrian Celaya, et al.
0

Within medical imaging segmentation, the Dice coefficient and Hausdorff-based metrics are standard measures of success for deep learning models. However, modern loss functions for medical image segmentation often only consider the Dice coefficient or similar region-based metrics during training. As a result, segmentation architectures trained over such loss functions run the risk of achieving high accuracy for the Dice coefficient but low accuracy for Hausdorff-based metrics. Low accuracy on Hausdorff-based metrics can be problematic for applications such as tumor segmentation, where such benchmarks are crucial. For example, high Dice scores accompanied by significant Hausdorff errors could indicate that the predictions fail to detect small tumors. We propose the Weighted Normalized Boundary Loss, a novel loss function to minimize Hausdorff-based metrics with more desirable numerical properties than current methods and with weighting terms for class imbalance. Our loss function outperforms other losses when tested on the BraTS dataset using a standard 3D U-Net and the state-of-the-art nnUNet architectures. These results suggest we can improve segmentation accuracy with our novel loss function.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
02/08/2021

A Mixed Focal Loss Function for Handling Class Imbalanced Medical Image Segmentation

Automatic segmentation methods are an important advancement in medical i...
research
05/29/2020

Enhancing Foreground Boundaries for Medical Image Segmentation

Object segmentation plays an important role in the modern medical image ...
research
02/25/2021

Binary segmentation of medical images using implicit spline representations and deep learning

We propose a novel approach to image segmentation based on combining imp...
research
01/18/2022

A Deep Learning Approach for Semantic Segmentation of Unbalanced Data in Electron Tomography of Catalytic Materials

Heterogeneous catalysts possess complex surface and bulk structures, rel...
research
01/10/2022

3D Segmentation with Fully Trainable Gabor Kernels and Pearson's Correlation Coefficient

The convolutional layer and loss function are two fundamental components...
research
10/22/2019

Penalizing small errors using an Adaptive Logarithmic Loss

Loss functions are error metrics that quantify the difference between a ...

Please sign up or login with your details

Forgot password? Click here to reset