Boundary-aware Contrastive Learning for Semi-supervised Nuclei Instance Segmentation

Ye Zhang,Ziyue Wang,Yifeng Wang,Hao Bian,Linghan Cai,Hengrui Li,Lingbo Zhang,Yongbing Zhang

2024-02-07

Abstract:Semi-supervised segmentation methods have demonstrated promising results in natural scenarios, providing a solution to reduce dependency on manual annotation. However, these methods face significant challenges when directly applied to pathological images due to the subtle color differences between nuclei and tissues, as well as the significant morphological variations among nuclei. Consequently, the generated pseudo-labels often contain much noise, especially at the nuclei boundaries. To address the above problem, this paper proposes a boundary-aware contrastive learning network to denoise the boundary noise in a semi-supervised nuclei segmentation task. The model has two key designs: a low-resolution denoising (LRD) module and a cross-RoI contrastive learning (CRC) module. The LRD improves the smoothness of the nuclei boundary by pseudo-labels denoising, and the CRC enhances the discrimination between foreground and background by boundary feature contrastive learning. We conduct extensive experiments to demonstrate the superiority of our proposed method over existing semi-supervised instance segmentation methods.

Computer Vision and Pattern Recognition

What problem does this paper attempt to address?

### Problems the Paper Aims to Solve

This paper aims to address the issue of semi-supervised nuclear instance segmentation in pathological images, particularly the problem of high boundary noise during pseudo-label generation. Specifically:

- **Limitations of Existing Methods**: Existing semi-supervised segmentation methods face challenges when applied to pathological images due to the low color contrast between nuclei and tissue, and the significant morphological variations of nuclei. This results in pseudo-labels containing a large amount of noise, especially at the nuclear boundaries.
- **Proposed Solution**: To address this, the paper proposes a boundary-aware contrastive learning network (BASS). This model includes two key modules: the Low-Resolution Denoising (LRD) module and the Cross-RoI Contrastive Learning (CRC) module. The LRD module improves the smoothness of nuclear boundaries by denoising pseudo-labels, while the CRC module enhances the distinction between foreground and background through boundary feature contrastive learning.

Experimental results show that this method outperforms existing semi-supervised instance segmentation methods on two public datasets.