PedSemiSeg: Pedagogy-inspired semi-supervised polyp segmentation.

Journal: Computerized medical imaging and graphics : the official journal of the Computerized Medical Imaging Society
Published Date:

Abstract

Recent advancements in deep learning techniques have contributed to developing improved polyp segmentation methods, thereby aiding in the diagnosis of colorectal cancer and facilitating automated surgery like endoscopic submucosal dissection (ESD). However, the scarcity of well-annotated data poses challenges by increasing the annotation burden and diminishing the performance of fully-supervised learning approaches. Additionally, distribution shifts due to variations among patients and medical centers require the model to generalize well during testing. To address these concerns, we present PedSemiSeg, a pedagogy-inspired semi-supervised learning framework designed to enhance polyp segmentation performance with limited labeled training data. In particular, we take inspiration from the pedagogy used in real-world educational settings, where teacher feedback and peer tutoring are both crucial in influencing the overall learning outcome. Expanding upon this concept, our approach involves supervising the outputs of the strongly augmented input (the students) using the pseudo and complementary labels crafted from the output of the weakly augmented input (the teacher) in both positive and negative learning manners. Additionally, we introduce reciprocal peer tutoring among the students, guided by respective prediction entropy. With these holistic learning processes, we aim to achieve consistent predictions for various versions of the same input and maximize the utilization of the abundant unlabeled data. Experimental results on two public datasets demonstrate the superiority of our method in polyp segmentation across various labeled data ratios. Furthermore, our approach exhibits excellent generalization capabilities on external unseen multi-center datasets, highlighting its broader clinical significance in practical applications during deployment.

Authors

  • An Wang
    Department of Neonatal Surgery, Beijing Children's Hospital, Capital Medical University, National Center for Children's Health, Beijing, China.
  • Haoyu Ma
    University of California, Irvine, Irvine, CA 92697, USA. Electronic address: haoyum3@uci.edu.
  • Long Bai
    State Key Laboratory of Mechanical Transmission, Chongqing University, Chongqing 400044, China. bailong@cqu.edu.cn.
  • Yanan Wu
    School of Physics and Optoelectronic Engineering, Ludong University, Yantai, Shandong 264025, China.
  • Mengya Xu
  • Yang Zhang
    Innovative Institute of Chinese Medicine and Pharmacy, Academy for Interdiscipline, Chengdu University of Traditional Chinese Medicine, Chengdu, China.
  • Mobarakol Islam
  • Hongliang Ren
    Department of Biomedical Engineering, National University of Singapore, Singapore, Singapore. ren@nus.edu.sg.