Semi-Supervised Gland Segmentation via Feature-Enhanced Contrastive Learning and Dual-Consistency Strategy.

Journal: IEEE journal of biomedical and health informatics
Published Date:

Abstract

In the field of gland segmentation in histopathology, deep-learning methods have made significant progress. However, most existing methods not only require a large amount of high-quality annotated data but also tend to confuse the internal of the gland with the background. To address this challenge, we propose a new semi-supervised method named DCCL-Seg for gland segmentation, which follows the teacher-student framework. Our approach can be divided into follows steps. First, we design a contrastive learning module to improve the ability of the student model's feature extractor to distinguish between gland and background features. Then, we introduce a Signed Distance Field (SDF) prediction task and employ dual-consistency strategy (across tasks and models) to better reinforce the learning of gland internal. Next, we proposed a pseudo label filtering and reweighting mechanism, which filters and reweights the pseudo labels generated by the teacher model based on confidence. However, even after reweighting, the pseudo labels may still be influenced by unreliable pixels. Finally, we further designed an assistant predictor to learn the reweighted pseudo labels, which do not interfere with the student model's predictor and ensure the reliability of the student model's predictions. Experimental results on the publicly available GlaS and CRAG datasets demonstrate that our method outperforms other semi-supervised medical image segmentation methods.

Authors

  • Jiejiang Yu
  • Bingbing Li
    Department of Pathology, Nanfang Hospital and Basic Medical College, Southern Medical University, Guangzhou 510515, Guangdong Province, China; Guangdong Province Key Laboratory of Molecular Tumor Pathology, Guangzhou 510515, Guangdong Province, China.
  • Xipeng Pan
    Department of Radiology, Guangdong Provincial People's Hospital, Guangdong Academy of Medical Sciences, Guangzhou, Guangdong 510080, China; Guangdong Cardiovascular Institute, Guangzhou, Guangdong 510080, China; Guangdong Provincial Key Laboratory of Artificial Intelligence in Medical Image Analysis and Application, Guangdong Provincial People's Hospital, Guangdong Academy of Medical Sciences, Guangzhou, Guangdong 510080, China.
  • Zhenwei Shi
    Department of Radiation Oncology (MAASTRO), GROW - School for Oncology and Development Biology, Maastricht University Medical Centre+, Maastricht, 6229 ET, The Netherlands.
  • Huadeng Wang
    School of Computer Science and Information Security, Guilin University of Electronic Technology, Guilin, 541004, China; Guangxi Key Laboratory of Image and Graphic Intelligent Processing, Guilin, 541004, China.
  • Rushi Lan
    Guangxi Colleges and Universities Key Laboratory of Intelligent Processing of Computer Image and Graphics, Guilin University of Electronic Technology, Guilin, Guangxi, China.
  • Xiaonan Luo