Data variation-aware medical image segmentation

TL;DR

This paper introduces a multi-path neural network approach that captures data variability in medical image segmentation, improving accuracy and clinical acceptance by generating multiple segmentation options tailored to dataset variations.

Contribution

It proposes a novel multi-path Unet architecture with automatic data partitioning via evolutionary optimization to enhance segmentation performance on heterogeneous clinical data.

Findings

Improved Dice scores by several percentage points over baseline methods.

Largest improvements observed in challenging prostate regions.

Effective handling of inter-observer segmentation variability.

Abstract

Deep learning algorithms have become the golden standard for segmentation of medical imaging data. In most works, the variability and heterogeneity of real clinical data is acknowledged to still be a problem. One way to automatically overcome this is to capture and exploit this variation explicitly. Here, we propose an approach that improves on our previous work in this area and explain how it potentially can improve clinical acceptance of (semi-)automatic segmentation methods. In contrast to a standard neural network that produces one segmentation, we propose to use a multi-pathUnet network that produces multiple segmentation variants, presumably corresponding to the variations that reside in the dataset. Different paths of the network are trained on disjoint data subsets. Because a priori it may be unclear what variations exist in the data, the subsets should be automatically determined. This is achieved by searching for the best data partitioning with an evolutionary optimization algorithm. Because each network path can become more specialized when trained on a more homogeneous data subset, better segmentation quality can be achieved. In practical usage, various automatically produced segmentations can be presented to a medical expert, from which the preferred segmentation can be selected. In experiments with a real clinical dataset of CT scans with prostate segmentations, our approach provides an improvement of several percentage points in terms of Dice and surface Dice coefficients compared to when all network paths are trained on all training data. Noticeably, the largest improvement occurs in the upper part of the prostate that is known to be most prone to inter-observer segmentation variation.