Tackling the Class Imbalance Problem of Deep Learning Based Head and Neck Organ Segmentation

1 code implementation • 5 Jan 2022 • Elias Tappeiner, Martin Welk, Rainer Schubert

Both the patch-size and the loss function are parameters with direct influence on the class imbalance and their optimization leads to a 3\% increase of the Dice score and 22% reduction of the 95% Hausdorff distance compared to the baseline, finally reaching $0. 8\pm0. 15$ and $3. 17\pm1. 7$ mm for the segmentation of seven HAN organs using a single and simple neural network.

Image Segmentation Medical Image Segmentation +3