Contrast Learning Based Robust Framework for Weakly Supervised Medical Image Segmentation with Coarse Bounding Box Annotations

Ziqi Zhu,Jun Shi,Minfan Zhao,Zhaohui Wang,Liang Qiao,Hong An
DOI: https://doi.org/10.1007/978-3-031-45087-7_12
2023-01-01
Abstract:The shortage of data due to high annotation costs has limited the development of supervised medical image segmentation methods that rely on tight pixel-level annotations. Recently, weakly supervised methods based on multiple instance learning have been proposed to reduce the annotation cost by using bounding box annotations and achieve competitive performance. However, most existing methods require accurate bounding box annotations to generate positive and negative sample bags, which is difficult to realize due to the inevitable errors associated with manually annotated bounding boxes. In this study, we propose a robust framework based on contrast learning for weakly supervised medical image segmentation. Specifically, our method involves a Fine-grained Semantic Representation Module (FSRM), which is used to distinguish foreground and background pixels inside a coarse bounding box. Positive and negative sample bags are generated for multiple instance learning based on the obtained foreground results instead of bounding box constraints. Therefore, our proposed method can ensure the performance under coarse labeling by automatically extracting the boundaries of foreground and background. Our method achieves state-of-the-art results on two publicly available datasets, and extensive experiments validate the robustness of our method under noisy annotations. The source code will be available at https://github.com/ta1ly/wsis-contrastlearning.
What problem does this paper attempt to address?