Probabilistic Domain Adaptation for Biomedical Image Segmentation

Anwai Archit,Constantin Pape
2023-03-21
Abstract:Segmentation is a key analysis tasks in biomedical imaging. Given the many different experimental settings in this field, the lack of generalization limits the use of deep learning in practice. Domain adaptation is a promising remedy: it trains a model for a given task on a source dataset with labels and adapts it to a target dataset without additional labels. We introduce a probabilistic domain adaptation method, building on self-training approaches and the Probabilistic UNet. We use the latter to sample multiple segmentation hypothesis to implement better pseudo-label filtering. We further study joint and separate source-target training strategies and evaluate our method on three challenging domain adaptation tasks for biomedical segmentation.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The paper primarily addresses the issue of domain adaptation in the field of biomedical image segmentation, particularly for image datasets under different experimental conditions. Specifically, the paper proposes a probabilistic domain adaptation method aimed at improving the segmentation performance of deep learning models on unlabeled target datasets without the need for additional manual annotation work. The core contributions of the paper can be summarized as follows: 1. **Proposed a probabilistic domain adaptation method**: This method combines self-training techniques with Probabilistic U-Net (PUNet) to generate more reliable pseudo-labels and uses these pseudo-labels to filter and train the model. 2. **Improved pseudo-label filtering mechanism**: By utilizing PUNet to sample multiple segmentation hypotheses, better pseudo-label filtering is implemented, improving the accuracy of segmentation results. 3. **Studied different training strategies**: Including joint training and staged training strategies. Joint training is conducted simultaneously on both source and target data; staged training involves pre-training the model on source data and then fine-tuning it only on target data. 4. **Evaluated the effectiveness of the method**: The method was evaluated on three challenging domain adaptation tasks, including cell segmentation under live-cell microscopy, mitochondrial segmentation under electron microscopy, and lung segmentation under X-ray. Through these techniques, the paper aims to enhance the generalization ability of models across different datasets, reduce the need for large amounts of annotated data, and lower annotation costs. This approach is particularly useful in the field of biomedical imaging, which often faces diverse experimental setups and modalities, as well as limited annotation resources.