Semi-supervised organ segmentation with Mask Propagation Refinement and Uncertainty Estimation for Data Generation

Minh-Khoi Pham, Thang-Long Nguyen-Ho, Tan-Cong Nguyen, Thao Thi Phuong Dao, Minh-triet Tran

Keywords: 2D semi-supervised segmentation, Mask Propagation, Uncertainty Estimation

TL;DR: Using a 2D two-staged pipeline with Mask Propagation for temporal refinement and Uncertainty Estimation for pseudo labeling

Abstract:

We present a novel two-staged method that employs various 2D-based techniques to deal with the 3D segmentation task. In most of
the previous challenges, it is unlikely for 2D CNNs to be comparable with other 3D CNNs since 2D models can hardly capture temporal information. In light of that, we propose using the recent state-of-the-art technique in video object segmentation, combining it with other semi-supervised training techniques to leverage the extensive unlabeled data. Moreover, we introduce a way to generate pseudo-labeled data that is both plausible and consistent for further retraining by using uncertainty estimation. Overall, our method achieves 0.7841 DSC on the validation set of FLARE22.